from fastapi import FastAPI
from pydantic import BaseModel
from gradio_client import Client

app = FastAPI()

class Message(BaseModel):
    text: str

@app.post("/ask")
async def ask_searchgpt(msg: Message):
    client = Client("https://umint-searchgpt.hf.space/")
    result = client.predict(
        user_message={"text": msg.text},
        api_name="/public"
    )
    return {"response": result}