Update Summarization/main.py

Summarization/main.py

@@ -43,9 +43,12 @@ async def caption(file: UploadFile = File(...)):
 
 @app.get("/files/{filename}")
 async def serve_file(filename: str):
-    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)
+    filepath = os.path.join(tempfile.gettempdir(), filename)
+        altpath = os.path.join(tempfile.gettempdir(), "aidan_files", filename)
+
+    if os.path.exists(filepath):
+        return FileResponse(filepath)
+    elif os.path.exists(altpath):
+        return FileResponse(altpath)
 
+    return JSONResponse({"error": "File not found"}, status_code=404)