import gradio as gr
from agent import run_agent

examples = [
    "What's the capital of France?",
    "Solve: 345 * 12",
    "Write a Python function to reverse a list.",
    "What's the square root of 169?",
    "Who discovered penicillin?"
]

def ask_agent(input_text):
    return run_agent(input_text)

iface = gr.Interface(fn=ask_agent, 
                     inputs="text", 
                     outputs="text", 
                     title="Final Assignment Agent",
                     description="Enter your question and the agent will answer it.",
                     examples=examples)

iface.launch()