import gradio as gr
from service.chatbot import process_question


def handle_input(question):
    result = process_question(question)
    return result["answer_model_1"], result["answer_model_2"], result["best_answer"]


iface = gr.Interface(
    fn=handle_input,
    inputs=gr.Textbox(label="Pergunta"),
    outputs=[
        gr.Textbox(label="Resposta do Modelo 1"),
        gr.Textbox(label="Resposta do Modelo 2"),
        gr.Textbox(label="Melhor Resposta (segundo Avaliador)")
    ]
)

if __name__ == "__main__":
    iface.launch()