Update Summarization/main.py

Summarization/main.py

@@ -41,9 +41,11 @@ async def caption(file: UploadFile = File(...)):
     except Exception as e:
         return JSONResponse({"error": str(e)}, status_code=500)
 
-@app.get("/files/{filename}")  # ✅ Corrected
+@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)
+    file_dir = os.path.join(tempfile.gettempdir(), "aidan_files")
+    file_path = os.path.join(file_dir, filename)
+    if os.path.exists(file_path):
+        return FileResponse(file_path)
     return JSONResponse({"error": "File not found"}, status_code=404)
+