|
|
import os |
|
|
import logging |
|
|
from fastapi import FastAPI, Request, HTTPException |
|
|
from fastapi.responses import JSONResponse |
|
|
from fastapi.middleware.cors import CORSMiddleware |
|
|
from pydantic import BaseModel |
|
|
|
|
|
from src.services.graph_service import GraphInterviewProcessor |
|
|
|
|
|
from langtrace_python_sdk import langtrace |
|
|
|
|
|
langtrace.init(api_key=os.getenv("LANGTRACE_API_KEY")) |
|
|
|
|
|
logging.basicConfig(level=logging.INFO) |
|
|
logger = logging.getLogger(__name__) |
|
|
|
|
|
app = FastAPI( |
|
|
title="Interview Simulation API", |
|
|
description="API for interview simulations.", |
|
|
version="1.0.0", |
|
|
docs_url="/docs", |
|
|
redoc_url="/redoc" |
|
|
) |
|
|
|
|
|
app.add_middleware( |
|
|
CORSMiddleware, |
|
|
allow_origins=["*"], |
|
|
allow_credentials=True, |
|
|
allow_methods=["*"], |
|
|
allow_headers=["*"], |
|
|
) |
|
|
|
|
|
class HealthCheck(BaseModel): |
|
|
status: str = "ok" |
|
|
|
|
|
@app.get("/", response_model=HealthCheck, tags=["Status"]) |
|
|
async def health_check(): |
|
|
return HealthCheck() |
|
|
|
|
|
@app.post("/simulate-interview/") |
|
|
async def simulate_interview(request: Request): |
|
|
""" |
|
|
This endpoint receives the interview data, instantiates the graph processor |
|
|
and starts the conversation. |
|
|
""" |
|
|
logger = logging.getLogger(__name__) |
|
|
try: |
|
|
payload = await request.json() |
|
|
|
|
|
if not all(k in payload for k in ["user_id", "job_offer_id", "cv_document", "job_offer"]): |
|
|
raise HTTPException(status_code=400, detail="Missing data in payload (user_id, job_offer_id, cv_document, job_offer).") |
|
|
|
|
|
logger.info(f"Starting simulation for user: {payload['user_id']}") |
|
|
|
|
|
processor = GraphInterviewProcessor(payload) |
|
|
result = processor.invoke(payload.get("messages", [])) |
|
|
|
|
|
return JSONResponse(content=result) |
|
|
|
|
|
except ValueError as ve: |
|
|
logger.error(f"Data validation error: {ve}", exc_info=True) |
|
|
return JSONResponse(content={"error": str(ve)}, status_code=400) |
|
|
except Exception as e: |
|
|
logger.error(f"Internal error in simulate-interview endpoint: {e}", exc_info=True) |
|
|
return JSONResponse( |
|
|
content={"error": "An internal error occurred on the assistant's server."}, |
|
|
status_code=500 |
|
|
) |
|
|
|
|
|
if __name__ == "__main__": |
|
|
import uvicorn |
|
|
port = int(os.getenv("PORT", 8002)) |
|
|
uvicorn.run(app, host="0.0.0.0", port=port) |
|
|
|
