app.py

import logging
import os
from fastapi import FastAPI, HTTPException
from fastapi.responses import FileResponse
from pydantic import BaseModel, HttpUrl
from typing import Dict
from process_interview import process_interview

logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

app = FastAPI(
    title="EvalBot API",
    description="API to analyze a single audio interview URL.",
    version="1.0.0"
)

# Directory where output files are stored
OUTPUT_DIR = "./processed_audio"
os.makedirs(OUTPUT_DIR, exist_ok=True)

class AudioItem(BaseModel):
    url: HttpUrl
    user_id: str

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))
        # Return just the filename, not the full path
        if result and "pdf_path" in result and result["pdf_path"]:
            result["pdf_filename"] = os.path.basename(result["pdf_path"])
        if result and "json_path" in result and result["json_path"]:
            result["json_filename"] = os.path.basename(result["json_path"])
        
        return {
            "user_id": item.user_id,
            "url": str(item.url),
            "status": "success",
            "result": result
        }
    except Exception as e:
        logger.error(f"Failed processing for user '{item.user_id}': {e}", exc_info=True)
        return {
            "user_id": item.user_id,
            "url": str(item.url),
            "status": "error",
            "detail": str(e)
        }

@app.get("/health", summary="Check API Health")
def health_check():
    return {"status": "ok"}

@app.post("/analyze", summary="Analyze a single audio URL")
def analyze_audio(request: AudioItem):
    """
    Analyze a single audio URL and return analysis results.
    """
    logger.info(f"Received request to analyze audio for user '{request.user_id}' with URL: {request.url}")
    result = process_item_wrapper(request)
    logger.info(f"Finished processing audio for user '{request.user_id}'")
    return {"analysis_result": result}

@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.")