Update app.py

app.py

@@ -17,6 +17,7 @@ from huggingface_hub import hf_hub_download
 from fastapi import FastAPI, UploadFile, File, HTTPException, Response, Depends, Security, Query
 from fastapi.responses import RedirectResponse
 from fastapi.security import HTTPBearer, HTTPAuthorizationCredentials
+from fastapi.concurrency import run_in_threadpool
 from pydantic import BaseModel
 from motor.motor_asyncio import AsyncIOMotorClient
 
@@ -235,9 +236,62 @@ def download_from_spaces(key):
 def root():
     return RedirectResponse("/gradio")
 
+# @fastapi_app.get("/health")
+# async def health():
+#     return {"status": "healthy
 @fastapi_app.get("/health")
 async def health():
-    return {"status": "healthy"}
+    mongo_ok = False
+    spaces_ok = False
+    mongo_error = None
+    spaces_error = None
+
+    # ---- MongoDB Check (ASYNC SAFE) ----
+    try:
+        if database is None:
+            raise Exception("MongoDB client not initialized")
+
+        await database.command("ping")
+        mongo_ok = True
+
+    except Exception as e:
+        mongo_error = str(e)
+
+    # ---- DigitalOcean Spaces Check (RUN IN THREADPOOL) ----
+    try:
+        def check_spaces():
+            client = get_spaces_client()
+            client.list_objects_v2(
+                Bucket=DO_SPACES_BUCKET,
+                MaxKeys=1
+            )
+
+        await run_in_threadpool(check_spaces)
+        spaces_ok = True
+
+    except Exception as e:
+        spaces_error = str(e)
+
+    # ---- FINAL RESPONSE ----
+    if mongo_ok and spaces_ok:
+        return {
+            "status": "healthy",
+            "mongo": "connected",
+            "spaces": "connected",
+            "timestamp": datetime.utcnow().isoformat()
+        }
+
+    raise HTTPException(
+        status_code=503,
+        detail={
+            "status": "unhealthy",
+            "mongo": "connected" if mongo_ok else "failed",
+            "spaces": "connected" if spaces_ok else "failed",
+            "mongo_error": mongo_error,
+            "spaces_error": spaces_error,
+            "timestamp": datetime.utcnow().isoformat()
+        }
+    )
 
 from fastapi import Form
 import requests