from gradio_client import Client
from util.access_token import load_token


TOKEN = load_token("TOKEN_READ")

#Example Bot
def bot_response(input_text: str) -> str:
    client = Client("PlanBTech/Medflow-Bot", hf_token=TOKEN)
    result = client.predict(
        message=input_text,
        max_tokens=512,
        temperature=0.7,
        top_p=0.95,
        api_name="/chat",
    )

    return str(result)