Update app.py

app.py

@@ -3,73 +3,82 @@ import uvicorn
 import numpy as np
 import cv2
 from fastapi import FastAPI, UploadFile, File, HTTPException
-from fastapi.responses import JSONResponse
 from bson import ObjectId
 from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorGridFSBucket
-from PIL import Image
 from rapidocr_onnxruntime import RapidOCR
 
-# -------------------------------------------------------------------
+# ------------------------------
 # CONFIG
-# -------------------------------------------------------------------
-MONGO_URI = "mongodb+srv://harilogicgo_db_user:jFhyDM4oA4dklUsp@api-logs.i7rqf9p.mongodb.net/?appName=API-LOGS"
+# ------------------------------
+MONGO_URI = (
+    "mongodb+srv://harilogicgo_db_user:jFhyDM4oA4dklUsp"
+    "@api-logs.i7rqf9p.mongodb.net/ocr_fastapi"
+    "?retryWrites=true&w=majority&appName=API-LOGS"
+)
+
 DB_NAME = "ocr_fastapi"
 BUCKET_NAME = "ocr_images"
 
-app = FastAPI(title="RapidOCR FastAPI", version="1.0")
+app = FastAPI()
 
 # DB INIT
 _client = AsyncIOMotorClient(MONGO_URI)
 _db = _client[DB_NAME]
 
+def gridfs():
+    return AsyncIOMotorGridFSBucket(_db, bucket_name=BUCKET_NAME)
+
 ocr_engine = RapidOCR()
 OCR_RESULTS = {}
 
-
+# ------------------------------
 @app.get("/health")
 async def health():
     return {"status": "ok"}
 
-
-# -------------------------------------------------------------------
-# UPLOAD
-# -------------------------------------------------------------------
+# ------------------------------
 @app.post("/upload")
 async def upload_image(file: UploadFile = File(...)):
     try:
         data = await file.read()
 
-        fs = AsyncIOMotorGridFSBucket(_db, bucket_name=BUCKET_NAME)
+        fs = gridfs()
 
         oid = await fs.upload_from_stream(
             file.filename,
             data,
             metadata={"contentType": file.content_type},
         )
+
+        print("📌 Uploaded ID:", oid)
+
         return {"image_id": str(oid)}
+
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
 
-
-# -------------------------------------------------------------------
-# LOAD IMAGE
-# -------------------------------------------------------------------
+# ------------------------------
 async def load_image_from_gridfs(image_id: str):
     try:
+        print("🔍 Fetching ID:", image_id)
+
         oid = ObjectId(image_id)
-        fs = AsyncIOMotorGridFSBucket(_db, bucket_name=BUCKET_NAME)
+        fs = gridfs()
 
         stream = await fs.open_download_stream(oid)
         data = await stream.read()
         await stream.close()
+
+        print("✅ Found image in GridFS")
+
         return data
+
     except Exception as e:
+        print("❌ GridFS error:", e)
         raise HTTPException(status_code=404, detail="Image not found")
 
 
-# -------------------------------------------------------------------
-# GENERATE OCR
-# -------------------------------------------------------------------
+# ------------------------------
 @app.post("/generate/{image_id}")
 async def generate(image_id: str):
     raw_bytes = await load_image_from_gridfs(image_id)
@@ -96,14 +105,13 @@ async def generate(image_id: str):
 
     return {"image_id": image_id, "text": full_text}
 
-
+# ------------------------------
 @app.get("/view/{image_id}")
 async def view_details(image_id: str):
     if image_id not in OCR_RESULTS:
         raise HTTPException(status_code=404, detail="No OCR result found")
-
     return OCR_RESULTS[image_id]
 
 
 if __name__ == "__main__":
-    uvicorn.run("app:app", host="0.0.0.0", port=8000, reload=True)
+    uvicorn.run("app:app", host="0.0.0.0", port=8000)