app.py

from fastapi import FastAPI
from pydantic import BaseModel
from router import route_request
import uuid

app = FastAPI()


class ChatRequest(BaseModel):
    prompt: str
    user_id: str | None = None


@app.post("/chat")
async def chat(request: ChatRequest):

    # Auto-generate user_id if not provided
    user_id = request.user_id or str(uuid.uuid4())

    result = await route_request(request.prompt, user_id)

    return {
        "user_id": user_id,
        "response": result.get("response", result)
    }