Update app.py

app.py

@@ -1,5 +1,7 @@
 import logging
+import os
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse # --- NEW IMPORT ---
 from pydantic import BaseModel, HttpUrl
 from typing import List, Dict
 from concurrent.futures import ThreadPoolExecutor
@@ -16,20 +18,27 @@ app = FastAPI(
     version="1.0.0"
 )
 
+# --- The directory where output files are stored ---
+OUTPUT_DIR = "./processed_audio"
+
 
 class AudioItem(BaseModel):
     url: HttpUrl
     user_id: str
 
-
 class AnalysisRequest(BaseModel):
     audio_items: List[AudioItem]
 
-
 def process_item_wrapper(item: AudioItem) -> Dict:
     try:
         logger.info(f"Starting analysis for user '{item.user_id}' with URL: {item.url}")
         result = process_interview(str(item.url))
+        # --- MODIFICATION: Return just the filename, not the full path ---
+        if result and "pdf_path" in result:
+            result["pdf_filename"] = os.path.basename(result["pdf_path"])
+        if result and "json_path" in result:
+            result["json_filename"] = os.path.basename(result["json_path"])
+        
         return {
             "user_id": item.user_id,
             "url": str(item.url),
@@ -45,25 +54,37 @@ def process_item_wrapper(item: AudioItem) -> Dict:
             "detail": str(e)
         }
 
-
 @app.get("/health", summary="Check API Health")
 def health_check():
     return {"status": "ok"}
 
-
 @app.post("/analyze", summary="Analyze a list of audio URLs")
 def analyze_audios(request: AnalysisRequest):
     if not request.audio_items:
         raise HTTPException(status_code=400, detail="No audio items provided.")
-
     logger.info(f"Received request to analyze {len(request.audio_items)} audio items.")
     results = []
-
     with ThreadPoolExecutor(max_workers=5) as executor:
         futures = [executor.submit(process_item_wrapper, item) for item in request.audio_items]
-
         for future in futures:
             results.append(future.result())
-
     logger.info("Finished processing all items.")
-    return {"analysis_results": results}
+    return {"analysis_results": results}
+
+
+# --- NEW ENDPOINT TO DOWNLOAD OUTPUT FILES ---
+@app.get("/outputs/{file_name}", summary="Download an output file")
+def get_output_file(file_name: str):
+    """
+    Downloads a generated report (PDF or JSON) by its filename.
+    """
+    file_path = os.path.join(OUTPUT_DIR, file_name)
+
+    # Security check to prevent accessing files outside the output directory
+    if not os.path.abspath(file_path).startswith(os.path.abspath(OUTPUT_DIR)):
+        raise HTTPException(status_code=403, detail="Forbidden: Access is denied.")
+
+    if os.path.exists(file_path):
+        return FileResponse(path=file_path, media_type='application/octet-stream', filename=file_name)
+    else:
+        raise HTTPException(status_code=404, detail="File not found.")