Update main.py

main.py

@@ -1,4 +1,4 @@
-from fastapi import FastAPI, UploadFile, Form, Request
+from fastapi import FastAPI, UploadFile, Form, Request, File
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.responses import HTMLResponse, JSONResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
@@ -34,22 +34,32 @@ async def summarize_document(file: UploadFile = File(...), length: str = Form("m
     try:
         from app import summarize_api
         return await summarize_api(file, length)
-    except ImportError:
+    except ImportError as e:
         return JSONResponse(
-            {"error": "Summarization module not available"}, 
+            {"error": f"Summarization module not available: {str(e)}"}, 
             status_code=501
         )
+    except Exception as e:
+        return JSONResponse(
+            {"error": f"Summarization failed: {str(e)}"},
+            status_code=500
+        )
 
 @app.post("/imagecaption/")
-async def caption_image(file: UploadFile):
+async def caption_image(file: UploadFile = File(...)):
     try:
         from appImage import caption_from_frontend
         return await caption_from_frontend(file)
-    except ImportError:
+    except ImportError as e:
         return JSONResponse(
-            {"error": "Image captioning module not available"}, 
+            {"error": f"Image captioning module not available: {str(e)}"}, 
             status_code=501
         )
+    except Exception as e:
+        return JSONResponse(
+            {"error": f"Image captioning failed: {str(e)}"},
+            status_code=500
+        )
 
 @app.get("/files/{filename}")
 async def serve_file(filename: str):
@@ -58,20 +68,25 @@ async def serve_file(filename: str):
         return FileResponse(file_path)
     return JSONResponse({"error": "File not found"}, status_code=404)
 
-# Optional: Fallback endpoint if you want a single endpoint
+# Optional unified endpoint
 @app.post("/predict")
 async def predict(
-    file: UploadFile,
+    file: UploadFile = File(...),
     option: str = Form(...),  # "Summarize" or "Captioning"
     length: Optional[str] = Form(None)  # Only for summarize
 ):
     try:
         if option == "Summarize":
             return await summarize_document(file, length or "medium")
-        else:
+        elif option == "Captioning":
             return await caption_image(file)
+        else:
+            return JSONResponse(
+                {"error": "Invalid option - must be 'Summarize' or 'Captioning'"},
+                status_code=400
+            )
     except Exception as e:
         return JSONResponse(
-            {"error": str(e)},
+            {"error": f"Prediction failed: {str(e)}"},
             status_code=500
         )