Update Summarization/main.py

Summarization/main.py

@@ -4,9 +4,7 @@ from fastapi.templating import Jinja2Templates
 from fastapi.middleware.cors import CORSMiddleware
 import tempfile, os
 from pathlib import Path
-
-from .app import summarize_document
-from .appImage import caption_image
+import mimetypes
 
 app = FastAPI()
 
@@ -21,7 +19,7 @@ app.add_middleware(
 current_dir = Path(__file__).parent
 templates = Jinja2Templates(directory=current_dir.parent / "templates")
 
-@app.get("/", response_class=HTMLResponse)  # ✅ Corrected
+@app.get("/", response_class=HTMLResponse)
 async def serve_home(request: Request):
     return templates.TemplateResponse("HomeS.html", {"request": request})
 
@@ -41,9 +39,21 @@ 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):
+@app.get("/files/{filename}")
+async def serve_file(filename: str, request: Request):
     filepath = os.path.join(tempfile.gettempdir(), filename)
-    if os.path.exists(filepath):
-        return FileResponse(filepath)
-    return JSONResponse({"error": "File not found"}, status_code=404)
+    
+    if not os.path.exists(filepath):
+        return JSONResponse({"error": "File not found"}, status_code=404)
+    
+    # Determine correct media type and set attachment disposition
+    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}"}
+    )