Update app.py

app.py

@@ -6,48 +6,78 @@ from PIL import Image
 from io import BytesIO
 import requests
 
-app = FastAPI(title="PaddleOCR API", description="OCR via URL or File Upload")
+app = FastAPI(
+    title="PaddleOCR API",
+    description="OCR via URL or File Upload",
+    version="1.0.0"
+)
 
-ocr = PaddleOCR(use_angle_cls=True, lang='en')
+ocr = PaddleOCR(
+    use_angle_cls=True,
+    lang="en"
+)
 
-def run_ocr(image_bytes):
+def run_ocr(image_bytes: bytes):
     img = Image.open(BytesIO(image_bytes)).convert("RGB")
-    result = ocr.ocr(np.array(img), cls=True)
+    img_np = np.array(img)
+
+    result = ocr.predict(img_np)
 
-    parsed_texts = []
     outputs = []
+    parsed_texts = []
 
-    for line in result:
-        for box, (text, conf) in line:
+    for res in result:
+        boxes = res.get("rec_boxes", [])
+        texts = res.get("rec_texts", [])
+        scores = res.get("rec_scores", [])
+
+        for box, text, score in zip(boxes, texts, scores):
             parsed_texts.append(text)
             outputs.append({
                 "text": text,
-                "confidence": float(conf),
-                "bounding_box": [[float(x), float(y)] for x, y in box]
+                "confidence": float(score),
+                "bounding_box": box.tolist()
             })
 
-    final_text = " ".join(parsed_texts)
-
     return {
         "results": outputs,
         "parsedText": parsed_texts,
-        "finalText": final_text
+        "finalText": " ".join(parsed_texts)
     }
 
 @app.get("/ocr")
-async def ocr_url(url: str):
+async def ocr_from_url(url: str):
     try:
-        img_data = requests.get(url, timeout=10).content
-        data = run_ocr(img_data)
-        return JSONResponse({"success": True, **data})
+        response = requests.get(url, timeout=15)
+        response.raise_for_status()
+        data = run_ocr(response.content)
+        return JSONResponse({
+            "success": True,
+            **data
+        })
     except Exception as e:
-        return JSONResponse({"success": False, "error": str(e)}, status_code=400)
+        return JSONResponse(
+            status_code=400,
+            content={
+                "success": False,
+                "error": str(e)
+            }
+        )
 
 @app.post("/ocr")
-async def ocr_buffer(file: UploadFile = File(...)):
+async def ocr_from_file(file: UploadFile = File(...)):
     try:
         image_bytes = await file.read()
         data = run_ocr(image_bytes)
-        return JSONResponse({"success": True, **data})
+        return JSONResponse({
+            "success": True,
+            **data
+        })
     except Exception as e:
-        return JSONResponse({"success": False, "error": str(e)}, status_code=400)
+        return JSONResponse(
+            status_code=400,
+            content={
+                "success": False,
+                "error": str(e)
+            }
+        )