Update Summarization/main.py

Summarization/main.py

@@ -43,9 +43,17 @@ async def caption(file: UploadFile = File(...)):
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
-@app.get("/files/{filename}")  #for download resume as pdf 
+@app.get("/files/{filename}")
 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)
+    
+    if not os.path.exists(filepath):
+        return JSONResponse({"error": "File not found"}, status_code=404)
+    
+    # Determine media type for proper browser handling
+    media_type, _ = mimetypes.guess_type(filename)
+    return FileResponse(
+        filepath,
+        media_type=media_type or "application/octet-stream",
+        filename=filename 
+    )