Update main.py

main.py

@@ -1,25 +1,19 @@
-from fastapi import FastAPI, File, UploadFile, Form
+from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import HTMLResponse
-from fastapi.staticfiles import StaticFiles
-import cv2
+from io import BytesIO
 import pytesseract
-import os
+from PIL import Image
+import cv2
+import numpy as np
 
 # Initialize FastAPI app
 app = FastAPI()
 
-# Define the upload folder path
-UPLOAD_FOLDER = "static/uploads/"
-
-# Ensure the 'uploads' folder exists and has correct permissions
-if not os.path.exists(UPLOAD_FOLDER):
-    os.makedirs(UPLOAD_FOLDER, mode=0o777)  # Create folder with read, write, execute permissions for everyone
-
 # Inform pytesseract where Tesseract is installed
 pytesseract.pytesseract.tesseract_cmd = "/usr/local/bin/tesseract"
 
 # Serve static files (images, etc.)
-app.mount("/static", StaticFiles(directory="static"), name="static")
+# app.mount("/static", StaticFiles(directory="static"), name="static")  # Not needed if you're not saving the images
 
 # Home route
 @app.get("/", response_class=HTMLResponse)
@@ -68,24 +62,17 @@ async def home():
     """
     return HTMLResponse(content=html_content)
 
-# Upload image route
+# Upload image route (image is processed directly in memory)
 @app.post("/upload_image/")
 async def upload_image(image: UploadFile = File(...)):
-    file_location = f"{UPLOAD_FOLDER}/{image.filename}"
-    with open(file_location, "wb") as file:
-        file.write(await image.read())
-
-    return {"image_url": f"/static/uploads/{image.filename}"}
-
-# Process image and extract text
-@app.post("/process_image/")
-async def process_image(image_path: str = Form(...)):
-    img = cv2.imread(image_path)
+    # Read the image file directly into memory
+    image_bytes = await image.read()
+    image_stream = BytesIO(image_bytes)
 
-    if img is None:
-        return {"error": "Unable to read the uploaded image. Please upload a valid image."}
+    # Open the image with PIL (Pillow)
+    img = Image.open(image_stream)
 
-    # Extract text from the image
+    # Process the image and extract text
     extracted_text = pytesseract.image_to_string(img)
 
     return {"extracted_text": extracted_text}