Update app.py

app.py

@@ -27,7 +27,8 @@ from collections import defaultdict
 import facexlib.utils.misc as facex_misc
 from basicsr.utils.realesrganer import RealESRGANer
 from facexlib.utils.misc import download_from_url
-
+from fastapi.concurrency import run_in_threadpool
+import shutil
 from utils.dataops import auto_split_upscale
 from typing import List, Optional
 from datetime import datetime
@@ -928,10 +929,62 @@ DEFAULT_WITH_MODEL_NAME = True
 # Singleton Upscale instance for API
 _api_upscale = Upscale()
 
+# @fastapi_app.get("/health")
+# def health(_: bool = Depends(_verify_bearer_token)):
+#     return {"status": "ok"}
 @fastapi_app.get("/health")
 def health(_: bool = Depends(_verify_bearer_token)):
-    return {"status": "ok"}
+    checks = {
+        "api": "ok",
+        "gpu": "unknown",
+        "torch": "unknown",
+        "disk": "unknown",
+        "mongo": "skipped",
+        "weights_dir": "unknown",
+        "timestamp": datetime.utcnow().isoformat()
+    }
+
+    errors = {}
 
+    # -------- 1. TORCH & CUDA CHECK --------
+    try:
+        checks["torch"] = torch.__version__
+        checks["gpu"] = "available" if torch.cuda.is_available() else "cpu-only"
+        if torch.cuda.is_available():
+            checks["gpu_name"] = torch.cuda.get_device_name(0)
+            checks["gpu_mem_allocated_mb"] = int(torch.cuda.memory_allocated(0) / 1024 / 1024)
+    except Exception as e:
+        checks["gpu"] = "failed"
+        errors["gpu"] = str(e)
+    # -------- 2. MONGODB CHECK (OPTIONAL) --------
+    if _mongo_client is not None:
+        try:
+            def mongo_ping():
+                _mongo_client.admin.command("ping")
+
+            await run_in_threadpool(mongo_ping)
+            checks["mongo"] = "connected"
+        except Exception as e:
+            checks["mongo"] = "failed"
+            errors["mongo"] = str(e)
+    else:
+        checks["mongo"] = "not_configured"
+
+    # -------- FINAL STATUS --------
+    is_healthy = len(errors) == 0
+
+    if is_healthy:
+        checks["status"] = "healthy"
+        return checks
+
+    raise HTTPException(
+        status_code=503,
+        detail={
+            "status": "unhealthy",
+            "checks": checks,
+            "errors": errors
+        }
+    )
 @fastapi_app.get("/source")
 def source(_: bool = Depends(_verify_bearer_token)):
     return {