Update main.py

main.py

@@ -2,7 +2,7 @@ import os
 import io
 import time
 import PIL.Image
-from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi import FastAPI, File, UploadFile, HTTPException, Request
 from fastapi.responses import JSONResponse
 from pdf2image import convert_from_bytes
 from google import genai
@@ -10,6 +10,14 @@ from google.genai.errors import ClientError
 
 app = FastAPI(title="PDF/Image Text Extraction API")
 
+# Global exception handler to always return JSON responses
+@app.exception_handler(Exception)
+async def global_exception_handler(request: Request, exc: Exception):
+    return JSONResponse(
+        status_code=500,
+        content={"detail": str(exc)}
+    )
+
 # Retrieve the API key from an environment variable.
 API_KEY = os.getenv("API_KEY")
 if not API_KEY: