Update Summarization/main.py

Summarization/main.py

@@ -18,16 +18,14 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Dynamic template path
 current_dir = Path(__file__).parent
 templates = Jinja2Templates(directory=current_dir.parent / "templates")
 
-# Note: No leading slashes in routes
-@app.get("", response_class=HTMLResponse)  # Handles /summarization/
+@app.get("/", response_class=HTMLResponse)  # ✅ Corrected
 async def serve_home(request: Request):
     return templates.TemplateResponse("HomeS.html", {"request": request})
 
-@app.post("/summarize/")  # Handles /summarization/summarize/
+@app.post("/summarize/")
 async def summarize_api(file: UploadFile = File(...), length: str = Form("medium")):
     try:
         result = await summarize_document(file, length)
@@ -35,7 +33,7 @@ async def summarize_api(file: UploadFile = File(...), length: str = Form("medium
     except Exception as e:
         return JSONResponse({"detail": str(e)}, status_code=500)
 
-@app.post("/imagecaption/")  # Handles /summarization/imagecaption/
+@app.post("/imagecaption/")
 async def caption(file: UploadFile = File(...)):
     try:
         result = await caption_image(file)
@@ -43,9 +41,9 @@ async def caption(file: UploadFile = File(...)):
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
-@app.get("files/{filename}")  # Handles /summarization/files/{filename}
+@app.get("/files/{filename}")  # ✅ Corrected
 async def serve_file(filename: str):
     filepath = os.path.join(tempfile.gettempdir(), filename)
     if os.path.exists(filepath):
         return FileResponse(filepath)
-    return JSONResponse({"error": "File not found"}, status_code=404)
+    return JSONResponse({"error": "File not found"}, status_code=404)