?

Dockerfile

@@ -29,20 +29,20 @@
 
 FROM python:3.10-slim
 
-# System deps for OpenCV headless (no GUI) – keep these
+# System deps for OpenCV headless
 RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 \
     libglib2.0-0 \
-  && rm -rf /var/lib/apt/lists/*
+ && rm -rf /var/lib/apt/lists/*
 
-# Non-root user (HF best practice)
+# Non-root user
 RUN useradd -m -u 1000 user
 USER user
 ENV PATH="/home/user/.local/bin:$PATH"
 
 WORKDIR /app
 
-# Install Python deps
+# Python deps
 COPY --chown=user requirements.txt .
 RUN pip install --no-cache-dir --upgrade pip \
  && pip install --no-cache-dir -r requirements.txt
@@ -50,17 +50,20 @@ RUN pip install --no-cache-dir --upgrade pip \
 # App code
 COPY --chown=user . /app
 
-# Perf hints (safe defaults; can override via Space Secrets)
+# Perf + Ultralytics config dir
 ENV OMP_NUM_THREADS=1 \
     OPENBLAS_NUM_THREADS=1 \
     UVICORN_WORKERS=2 \
     YOLO_CONCURRENCY=1 \
     HTTP_MAX_CONNECTIONS=50 \
     DB_POOL_MAX=10 \
-    AUTH_CACHE_TTL_SECONDS=45
+    AUTH_CACHE_TTL_SECONDS=45 \
+    YOLO_CONFIG_DIR=/tmp/Ultralytics
+
+RUN mkdir -p /tmp/Ultralytics
 
 EXPOSE 7860
 
-# Run the app. We call python so app.py reads env (workers, etc.)
-CMD ["python", "app.py"]
+# Run via import string so multi-workers work cleanly
+CMD ["bash", "-lc", "uvicorn app:app --host 0.0.0.0 --port 7860 --workers ${UVICORN_WORKERS:-2}"]
 

app.py

@@ -775,26 +775,61 @@ def category_to_slug(category: str | None) -> str:
         .replace("-", "_")
     )
 
+# def parse_extracted_json(raw_text: str) -> dict:
+#     if not raw_text:
+#         return {}
+#     text = str(raw_text).strip()
+#     s, e = text.find("{"), text.rfind("}")
+#     if s != -1 and e != -1 and e > s:
+#         try:
+#             return json.loads(text[s:e+1])
+#         except Exception:
+#             pass
+#     if text.startswith("```") and text.endswith("```"):
+#         lines = text.splitlines()
+#         if lines and lines[0].startswith("```"): lines = lines[1:]
+#         if lines and lines[-1].startswith("```"): lines = lines[:-1]
+#         try:
+#             return json.loads("\n".join(lines).strip())
+#         except Exception:
+#             return {}
+#     return {}
 def parse_extracted_json(raw_text: str) -> dict:
     if not raw_text:
         return {}
     text = str(raw_text).strip()
+
+    # 1) Try largest JSON object between first '{' and last '}'
     s, e = text.find("{"), text.rfind("}")
     if s != -1 and e != -1 and e > s:
+        candidate = text[s:e + 1]
         try:
-            return json.loads(text[s:e+1])
+            return json.loads(candidate)
         except Exception:
             pass
+
+    # 2) Handle fenced code blocks ```json ... ```
     if text.startswith("```") and text.endswith("```"):
         lines = text.splitlines()
-        if lines and lines[0].startswith("```"): lines = lines[1:]
-        if lines and lines[-1].startswith("```"): lines = lines[:-1]
+        if lines and lines[0].startswith("```"):
+            lines = lines[1:]
+        if lines and lines[-1].startswith("```"):
+            lines = lines[:-1]
         try:
             return json.loads("\n".join(lines).strip())
         except Exception:
+            # last-chance bracket slice inside the fence
+            inner = "\n".join(lines).strip()
+            s2, e2 = inner.find("{"), inner.rfind("}")
+            if s2 != -1 and e2 != -1 and e2 > s2:
+                try:
+                    return json.loads(inner[s2:e2 + 1])
+                except Exception:
+                    return {}
             return {}
     return {}
 
+
 def parse_total(value: str | float | int | None) -> float:
     if value is None:
         return 0.0