Update app.py

app.py

@@ -7,21 +7,25 @@ import json
 import requests
 from process_interview import process_interview
 from fastapi.staticfiles import StaticFiles
+from fastapi.responses import FileResponse
 
 app = FastAPI()
 
-app.mount("/static", StaticFiles(directory="static"), name="static")
-
 TEMP_DIR = "./temp_files"
 OUTPUT_DIR = "./static/outputs"
+JSON_DIR = os.path.join(OUTPUT_DIR, "json")
+PDF_DIR = os.path.join(OUTPUT_DIR, "pdf")
 
 os.makedirs(TEMP_DIR, exist_ok=True)
-os.makedirs(OUTPUT_DIR, exist_ok=True)
+os.makedirs(JSON_DIR, exist_ok=True)
+os.makedirs(PDF_DIR, exist_ok=True)
+
+app.mount("/static", StaticFiles(directory="static"), name="static")
 
 VALID_EXTENSIONS = ('.wav', '.mp3', '.m4a', '.flac')
 MAX_FILE_SIZE_MB = 300
 
-BASE_URL = os.getenv("BASE_URL", "http://localhost:7860/static/outputs")
+BASE_URL = os.getenv("BASE_URL", "https://evalbot-audio-evalbot.hf.space")
 
 class ProcessResponse(BaseModel):
     summary: str
@@ -63,19 +67,16 @@ async def process_audio(request: ProcessAudioRequest = Body(...)):
             os.remove(local_path)
             raise HTTPException(status_code=500, detail="Processing failed")
 
-        json_src = result['json_path']
-        pdf_src = result['pdf_path']
-
         json_dest_name = f"{user_id}_{uuid.uuid4().hex}.json"
         pdf_dest_name = f"{user_id}_{uuid.uuid4().hex}.pdf"
 
-        json_dest = os.path.join(OUTPUT_DIR, json_dest_name)
-        pdf_dest = os.path.join(OUTPUT_DIR, pdf_dest_name)
+        json_dest = os.path.join(JSON_DIR, json_dest_name)
+        pdf_dest = os.path.join(PDF_DIR, pdf_dest_name)
 
-        shutil.copyfile(json_src, json_dest)
-        shutil.copyfile(pdf_src, pdf_dest)
+        shutil.copyfile(result['json_path'], json_dest)
+        shutil.copyfile(result['pdf_path'], pdf_dest)
 
-        with open(json_src, "r") as jf:
+        with open(result['json_path'], "r") as jf:
             analysis_data = json.load(jf)
 
         voice = analysis_data.get('voice_analysis', {}).get('interpretation', {})
@@ -90,8 +91,8 @@ async def process_audio(request: ProcessAudioRequest = Body(...)):
             f"Anxiety: {voice.get('anxiety_level', 'N/A')}"
         )
 
-        json_url = f"{BASE_URL}/{json_dest_name}"
-        pdf_url = f"{BASE_URL}/{pdf_dest_name}"
+        json_url = f"{BASE_URL}/outputs/json/{json_dest_name}"
+        pdf_url = f"{BASE_URL}/outputs/pdf/{pdf_dest_name}"
 
         os.remove(local_path)
 
@@ -99,3 +100,18 @@ async def process_audio(request: ProcessAudioRequest = Body(...)):
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
+
+
+@app.get("/outputs/json/{filename}")
+async def get_json_file(filename: str):
+    file_path = os.path.join(JSON_DIR, filename)
+    if not os.path.exists(file_path):
+        raise HTTPException(status_code=404, detail="JSON file not found")
+    return FileResponse(file_path, media_type="application/json", filename=filename)
+
+@app.get("/outputs/pdf/{filename}")
+async def get_pdf_file(filename: str):
+    file_path = os.path.join(PDF_DIR, filename)
+    if not os.path.exists(file_path):
+        raise HTTPException(status_code=404, detail="PDF file not found")
+    return FileResponse(file_path, media_type="application/pdf", filename=filename)