import gradio as gr
from agent import BasicAgent

agent = BasicAgent()

def run_agent(question):
    return agent(question)

with gr.Blocks() as demo:
    gr.Markdown("# Agent Tool Tester")
    
    with gr.Row():
        question_input = gr.Textbox(label="Question")
        run_button = gr.Button("Run Agent")
    
    output = gr.Textbox(label="Agent Output")
    
    run_button.click(fn=run_agent, inputs=question_input, outputs=output)
    
    gr.Markdown("## Pre-defined Questions")
    
    with gr.Accordion("Python Interpreter"):
        gr.Examples(
            examples=[
                "What is 2 + 2?",
                "What is the square root of 144?",
            ],
            inputs=question_input
        )

    with gr.Accordion("DuckDuckGo Search"):
        gr.Examples(
            examples=[
                "What is the capital of France?",
                "Who is the current president of the United States?",
            ],
            inputs=question_input
        )
        
    with gr.Accordion("File Reader from URL"):
        gr.Examples(
            examples=[
                "Read the first 5 rows of this CSV file: https://raw.githubusercontent.com/datasciencedojo/datasets/master/titanic.csv",
            ],
            inputs=question_input
        )

    with gr.Accordion("Wikipedia"):
        gr.Examples(
            examples=[
                "Tell me about the Eiffel Tower.",
            ],
            inputs=question_input
        )

    with gr.Accordion("YouTube Transcript Reader"):
        gr.Examples(
            examples=[
                "What is the summary of the video https://www.youtube.com/watch?v=dQw4w9WgXcQ?",
            ],
            inputs=question_input
        )

    with gr.Accordion("Web Page Reader"):
        gr.Examples(
            examples=[
                "Read the content of the page https://en.wikipedia.org/wiki/Main_Page",
            ],
            inputs=question_input
        )
        
    with gr.Accordion("Audio Analyzer"):
        gr.Examples(
            examples=[
                "Transcribe the audio from this URL: https://www.kozco.com/tech/piano2-CoolEdit.mp3",
            ],
            inputs=question_input
        )
        
    with gr.Accordion("Video Analyzer"):
        gr.Examples(
            examples=[
                "Transcribe the audio from this video: https://www.w3schools.com/html/mov_bbb.mp4",
            ],
            inputs=question_input
        )
        
    with gr.Accordion("Image Analyzer"):
        gr.Examples(
            examples=[
                "What is in this image? https://www.w3schools.com/html/img_chania.jpg",
            ],
            inputs=question_input
        )

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