Update app.py

app.py

@@ -1,12 +1,21 @@
 import os
 import io
-from flask import Flask, request, jsonify
-from flask_cors import CORS  # Add CORS support
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
 from PIL import Image
 import google.generativeai as genai
+import uvicorn
 
-app = Flask(__name__)
-CORS(app)  # Enable CORS for all routes
+app = FastAPI()
+
+# CORS Middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # Allows all origins
+    allow_credentials=True,
+    allow_methods=["*"],  # Allows all methods
+    allow_headers=["*"],  # Allows all headers
+)
 
 # Configure Gemini API
 genai.configure(api_key=os.getenv("GEMINI_API_KEY"))
@@ -37,23 +46,23 @@ def extract_mixed_text(image):
         return filtered_text.strip() if filtered_text.strip() else "No valid text detected."
     
     except Exception as e:
-        return f"An error occurred: {str(e)}"
+        raise HTTPException(status_code=500, detail=f"Error extracting text: {str(e)}")
 
-@app.route("/extract_text", methods=["POST"])
-def extract_text():
+@app.post("/extract_text")
+async def extract_text(image: UploadFile = File(...)):
     """API Endpoint for text extraction."""
     try:
-        if "image" not in request.files:
-            return jsonify({"error": "No image uploaded"}), 400
+        # Read the image file
+        image_contents = await image.read()
+        pil_image = Image.open(io.BytesIO(image_contents))
         
-        image_file = request.files["image"]
-        image = Image.open(io.BytesIO(image_file.read()))
+        # Extract text
+        extracted_text = extract_mixed_text(pil_image)
         
-        extracted_text = extract_mixed_text(image)
-        return jsonify({"extracted_text": extracted_text})
+        return {"extracted_text": extracted_text}
     
     except Exception as e:
-        return jsonify({"error": str(e)}), 500
+        raise HTTPException(status_code=400, detail=str(e))
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    uvicorn.run(app, host="0.0.0.0", port=7860)