Update app.py

app.py

@@ -1,11 +1,10 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
-from fastapi.responses import JSONResponse
+from fastapi.responses import FileResponse
 from pydantic import BaseModel
 from PIL import Image
 import io
 import os
 import google.generativeai as genai
-from fastapi.responses import FileResponse
 from pdf_generator import create_receipt
 
 
@@ -16,9 +15,9 @@ app = FastAPI()
 class OCRResponse(BaseModel):
     extracted_text: str
 
+
 @app.post("/ocr-image/", response_model=OCRResponse)
 async def ocr_image(file: UploadFile = File(...)):
-    
     if not file.content_type.startswith("image/"):
         raise HTTPException(status_code=400, detail="Uploaded file must be an image.")
 
@@ -26,20 +25,24 @@ async def ocr_image(file: UploadFile = File(...)):
         image_data = await file.read()
         image = Image.open(io.BytesIO(image_data))
 
-    
-        model = genai.GenerativeModel('gemini-2.0-flash')
+        model = genai.GenerativeModel("gemini-2.0-flash")
 
-        prompt = "Extraire le texte de cette image et générer un reçu à propos les marchandises. sans aucune information supplémentaire."
-        contents = [prompt, image]
+        prompt = (
+            "Extraire le texte de cette image et générer un reçu à propos les marchandises, "
+            "sans aucune information supplémentaire."
+        )
 
+        contents = [prompt, image]
         response = model.generate_content(contents)
 
         extracted_text = response.text
 
-        #create_receipt(extracted_text, file_name="result/receipt.pdf")
-        #return FileResponse(path="result/receipt.pdf", filename="receipt.pdf", media_type='application/pdf')
-        return extracted_text
+        # If PDF generation is desired:
+        # create_receipt(extracted_text, file_name="result/receipt.pdf")
+        # return FileResponse("result/receipt.pdf", filename="receipt.pdf", media_type="application/pdf")
+
+        # FIX: return correct JSON matching OCRResponse
+        return {"extracted_text": extracted_text}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=f"OCR failed: {str(e)}")
-