test_app.py

test_app.py

@@ -0,0 +1,97 @@
+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()