# app.py

import gradio as gr
from agent import create_agent
from dotenv import load_dotenv

load_dotenv()
agent = create_agent()


def respond_to_query(query: str) -> str:
    response = agent.run(query)
    return response


with gr.Blocks() as demo:
    gr.Markdown("# 🤖 SmolAgent GAIA Assistant")
    with gr.Row():
        user_input = gr.Textbox(label="Ask your question")
        output = gr.Textbox(label="Agent's Response")
    submit_btn = gr.Button("Submit")

    submit_btn.click(fn=respond_to_query, inputs=[user_input], outputs=[output])

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