Fix runtime timeout error: implement lazy client initialization with retry logic

.gitignore

@@ -0,0 +1,12 @@
+.history/
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.Python
+env/
+venv/
+.env
+static/
+*.log
+

app.py

@@ -14,8 +14,35 @@ load_dotenv()
 # Initialize FastAPI
 app = FastAPI()
 
-# Initialize Hugging Face Client (background remover Space)
-client = Client("LogicGoInfotechSpaces/background-remover")
+# Initialize Hugging Face Client (background remover Space) - lazy initialization
+client = None
+
+def get_client():
+    """Lazy initialization of Gradio Client with timeout handling"""
+    global client
+    if client is None:
+        max_retries = 3
+        retry_delay = 2  # seconds
+        
+        for attempt in range(max_retries):
+            try:
+                # Initialize with longer timeout
+                # Note: timeout parameter may vary by gradio_client version
+                client = Client(
+                    "LogicGoInfotechSpaces/background-remover",
+                    timeout=60.0  # 60 second timeout
+                )
+                print(f"Gradio Client initialized successfully (attempt {attempt + 1})")
+                break
+            except Exception as e:
+                if attempt < max_retries - 1:
+                    print(f"Error initializing Gradio Client (attempt {attempt + 1}/{max_retries}): {e}")
+                    print(f"Retrying in {retry_delay} seconds...")
+                    time.sleep(retry_delay)
+                else:
+                    print(f"Failed to initialize Gradio Client after {max_retries} attempts: {e}")
+                    raise
+    return client
 
 # Create static folder to store files
 os.makedirs("static", exist_ok=True)
@@ -64,10 +91,20 @@ async def bg_remove(
         f.write(await file.read())
 
     # Call Hugging Face Space via gradio_client (returns transparent PNG)
-    result_path = client.predict(
-        f=handle_file(input_path),
-        api_name="/png"
-    )
+    try:
+        hf_client = get_client()
+        result_path = hf_client.predict(
+            f=handle_file(input_path),
+            api_name="/png"
+        )
+    except Exception as e:
+        # Clean up input file on error
+        if os.path.exists(input_path):
+            os.remove(input_path)
+        raise HTTPException(
+            status_code=503,
+            detail=f"Background removal service unavailable: {str(e)}"
+        )
 
     # Ensure output has .png extension
     output_id = f"bgremoved_{os.path.splitext(file_id)[0]}.png"