Update main.py

main.py

@@ -1,4 +1,4 @@
-# import io
+import io
 import os
 
 from fastapi import FastAPI, File, HTTPException, UploadFile
@@ -6,10 +6,10 @@ from fastapi.responses import JSONResponse
 from starlette.responses import FileResponse
 from starlette.middleware.cors import CORSMiddleware
 
-# From PIL import Image
+from PIL import Image
 from pdftoword import convertPDFtoWORD
 
-# from model import inference
+from model import inference
 
 
 app = FastAPI()
@@ -27,27 +27,27 @@ allow_headers=["*"], # Allows all headers
 
 @app.post("/upload")
 async def extract_table_data(image: UploadFile = File(...)):
-    return f"table ocr is disabled 😔"
-    # try:
-    #     # Read image data
-    #     image_data = await image.read()
-
-    #     # Open image in memory
-    #     image = Image.open(io.BytesIO(image_data))
-    #     rgb_img = image.convert("RGB")
-    #     rgb_img.save('output.jpg')
-    #     image = Image.open('output.jpg')
-
-    #     table_fram= inference(image)
-    #     if table_fram.empty:
-    #         return "<h2 style=\"color: darkslategrey;\">💡 the image has no tables 💡</h2>"
-
-    #     return table_fram.to_html(escape=True,border=1,index=False).replace('\n', '')
-
-    # except Exception as e:
-    #     # Handle and log exceptions appropriately
-    #     print(f"Error processing image: {e}")
-    #     raise HTTPException(status_code=500, detail="Internal server error")
+    # return f"table ocr is disabled 😔"
+    try:
+        # Read image data
+        image_data = await image.read()
+
+        # Open image in memory
+        image = Image.open(io.BytesIO(image_data))
+        rgb_img = image.convert("RGB")
+        rgb_img.save('output.jpg')
+        image = Image.open('output.jpg')
+
+        table_fram= inference(image)
+        if table_fram.empty:
+            return "<h2 style=\"color: darkslategrey;\">💡 the image has no tables 💡</h2>"
+
+        return table_fram.to_html(escape=True,border=1,index=False).replace('\n', '')
+
+    except Exception as e:
+        # Handle and log exceptions appropriately
+        print(f"Error processing image: {e}")
+        raise HTTPException(status_code=500, detail="Internal server error")