Upload 277 files

.gitignore

@@ -22,3 +22,21 @@ Thumbs.db
 # Editor directories
 .vscode/
 .idea/
+
+# Runtime data — never commit
+backend/uploads/
+backend/logs/
+backend/masks/
+
+# ML model weights
+gradio-space/models/
+**/*.pt
+**/*.pth
+**/*.bin
+**/*.safetensors
+**/*.gguf
+
+# Python cache
+__pycache__/
+*.pyc
+*.pyo

backend/services/gradio_client_service.py

@@ -29,7 +29,8 @@ def _call_gradio_sync(image_path: Path) -> tuple[np.ndarray, int]:
     """
     from gradio_client import Client, file  # type: ignore
 
-    client = Client(GRADIO_SPACE_URL)
+    # 300s timeout: ZeroGPU cold start + SAM2+DINO inference can take 60-120s
+    client = Client(GRADIO_SPACE_URL, httpx_kwargs={"timeout": 300.0})
 
     # segment_for_backend returns (overlay_image, combined_json_str)
     _overlay_file, combined_json_str = client.predict(
@@ -65,18 +66,18 @@ def _call_gradio_sync(image_path: Path) -> tuple[np.ndarray, int]:
     return label_map, mask_count
 
 
-def segment_via_gradio_sync(image_path: Path) -> tuple[np.ndarray, int] | None:
+def segment_via_gradio_sync(image_path: Path) -> tuple[np.ndarray, int]:
     """
     Blocking call to the Gradio Space from a sync context (background task thread).
-    Returns (label_map, mask_count) or None on failure / not configured.
+    Returns (label_map, mask_count). Raises on failure.
     """
     if not is_gradio_enabled():
-        return None
+        raise RuntimeError("GRADIO_SPACE_URL is not configured")
     try:
         return _call_gradio_sync(image_path)
     except Exception as exc:
-        logger.warning("Gradio Space call failed, falling back to local SAM2: %s", exc)
-        return None
+        logger.error("Gradio Space call failed: %s", exc, exc_info=True)
+        raise RuntimeError(f"Gradio Space error: {exc}") from exc
 
 
 async def segment_via_gradio(image_path: Path) -> tuple[np.ndarray, int] | None:

backend/services/image_service.py

@@ -134,13 +134,7 @@ def run_upload_job(job_id: str, content: bytes, original_name: str) -> None:
 
         image_path = UPLOAD_DIR / safe_name
         if is_gradio_enabled():
-            gradio_result = segment_via_gradio_sync(image_path)
-            if gradio_result is None:
-                raise RuntimeError(
-                    "El Space de Gradio no respondió. "
-                    "Verifica que esté activo y que GRADIO_SPACE_URL sea correcta."
-                )
-            label_map, mask_count = gradio_result
+            label_map, mask_count = segment_via_gradio_sync(image_path)
         else:
             generate_label_map = _get_generate_label_map()
             label_map, mask_count = generate_label_map(image_rgb)