Update app.py

app.py

@@ -13,7 +13,8 @@ from fastapi import FastAPI, UploadFile, File, HTTPException, Header
 from rapidocr_onnxruntime import RapidOCR
 from openai import OpenAI
 from pymongo import MongoClient
-
+from pymongo.errors import PyMongoError
+from botocore.exceptions import BotoCoreError, ClientError
 # ---------------- ENV CONFIG ----------------
 DO_KEY_ID = os.getenv("DO_SPACES_KEY_ID")
 DO_SECRET_KEY = os.getenv("DO_SPACES_SECRET_KEY")
@@ -84,7 +85,50 @@ def log_api_event(
 # ---------------- ROUTES ----------------
 @app.get("/health")
 async def health():
-    return {"status": "ok"}
+    health_report = {
+        "service": "Receipt Scanner API",
+        "status": "healthy",
+        "checks": {}
+    }
+
+    # ---------------- MongoDB ----------------
+    try:
+        mongo_client.admin.command("ping")
+        health_report["checks"]["mongodb"] = "ok"
+    except PyMongoError as e:
+        health_report["checks"]["mongodb"] = f"fail: {str(e)}"
+        health_report["status"] = "degraded"
+
+    # ---------------- OpenAI ----------------
+    try:
+        # very light call, does not consume tokens
+        client.models.list()
+        health_report["checks"]["openai"] = "ok"
+    except Exception as e:
+        health_report["checks"]["openai"] = f"fail: {str(e)}"
+        health_report["status"] = "degraded"
+
+    # ---------------- DO Spaces / S3 ----------------
+    try:
+        s3.head_bucket(Bucket=DO_BUCKET)
+        health_report["checks"]["object_storage"] = "ok"
+    except (BotoCoreError, ClientError) as e:
+        health_report["checks"]["object_storage"] = f"fail: {str(e)}"
+        health_report["status"] = "degraded"
+
+    # ---------------- OCR Engine ----------------
+    try:
+        if ocr_engine is None:
+            raise RuntimeError("OCR engine not initialized")
+        health_report["checks"]["ocr"] = "ok"
+    except Exception as e:
+        health_report["checks"]["ocr"] = f"fail: {str(e)}"
+        health_report["status"] = "degraded"
+
+    # ---------------- Overall ----------------
+    health_report["timestamp"] = datetime.utcnow().isoformat()
+
+    return health_report
 
 @app.post("/upload")
 async def upload_image(file: UploadFile = File(...)):