# server.py
import traceback
import uuid
from fastapi import APIRouter, Request
from fastapi.responses import JSONResponse
from gradio_client import Client
from pydantic import BaseModel
from app import chat

router = APIRouter()


class Message(BaseModel):
    text: str


@router.post("/chat")
async def handle_chat(msg: Message):
    try:
        thread_id = uuid.uuid4()
        reply_text = chat(
            debug_mode=False, message=msg.text, history=None, thread_id=thread_id
        )

        return {"reply": reply_text}

    except Exception as e:
        traceback.print_exc()
        return {"reply": f"Error: {e}"}