Add health server so HF marks worker healthy

Dockerfile

@@ -8,4 +8,4 @@ COPY requirements.txt /app/requirements.txt
 RUN pip install --no-cache-dir -r /app/requirements.txt
 
 COPY . /app
-CMD ["python", "-u", "app.py"]
+CMD ["sh","-lc","set -e; python -u backend/worker/worker.py & exec uvicorn health_api:app --host 0.0.0.0 --port "]

health_api.py

@@ -0,0 +1,11 @@
+from fastapi import FastAPI
+
+app = FastAPI()
+
+@app.get("/")
+def root():
+    return {"ok": True}
+
+@app.get("/health")
+def health():
+    return {"ok": True}

requirements.txt

@@ -10,3 +10,5 @@ Pillow>=10.0.0
 openai>=1.0.0
 
 PyMuPDF>=1.23.0
+fastapi==0.128.0
+uvicorn==0.40.0

start.sh

@@ -0,0 +1,4 @@
+#!/bin/sh
+set -e
+python -u backend/worker/worker.py &
+exec uvicorn health_api:app --host 0.0.0.0 --port "${PORT:-7860}"