Update Summarization/main.py

Summarization/main.py

@@ -4,7 +4,9 @@ from fastapi.templating import Jinja2Templates
 from fastapi.middleware.cors import CORSMiddleware
 import tempfile, os
 from pathlib import Path
-import mimetypes
+
+from .app import summarize_document
+from .appImage import caption_image
 
 app = FastAPI()
 
@@ -16,14 +18,16 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+# Dynamic template path
 current_dir = Path(__file__).parent
 templates = Jinja2Templates(directory=current_dir.parent / "templates")
 
-@app.get("/", response_class=HTMLResponse) 
+# Note: No leading slashes in routes
+@app.get("", response_class=HTMLResponse)  # Handles /summarization/
 async def serve_home(request: Request):
     return templates.TemplateResponse("HomeS.html", {"request": request})
 
-@app.post("/summarize/")
+@app.post("summarize/")  # Handles /summarization/summarize/
 async def summarize_api(file: UploadFile = File(...), length: str = Form("medium")):
     try:
         result = await summarize_document(file, length)
@@ -31,7 +35,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/")
+@app.post("imagecaption/")  # Handles /summarization/imagecaption/
 async def caption(file: UploadFile = File(...)):
     try:
         result = await caption_image(file)
@@ -39,20 +43,9 @@ async def caption(file: UploadFile = File(...)):
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
-@app.get("/files/{filename}")
-async def serve_file(filename: str, request: Request):
+@app.get("files/{filename}")  #for download resume as pdf 
+async def serve_file(filename: str):
     filepath = os.path.join(tempfile.gettempdir(), filename)
-    
-    if not os.path.exists(filepath):
-        return JSONResponse({"error": "File not found"}, status_code=404)
-    
-    media_type, _ = mimetypes.guess_type(filepath)
-    if media_type is None:
-        media_type = "application/octet-stream"
-    
-    return FileResponse(
-        filepath,
-        media_type=media_type,
-        filename=filename,
-        headers={"Content-Disposition": f"attachment; filename={filename}"}
-    )
+    if os.path.exists(filepath):
+        return FileResponse(filepath)
+    return JSONResponse({"error": "File not found"}, status_code=404)