Update app.py

app.py

@@ -3,7 +3,7 @@ import gradio as gr
 import requests
 import inspect
 import pandas as pd
-
+import wikipidea
 # (Keep Constants as is)
 # --- Constants ---
 DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
@@ -14,22 +14,23 @@ class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
     def __call__(self, question: str) -> str:
-        print(f"BasicAgent received: {question[:50]}...")
         return "This is a default answer."
 
 class EchoAgent:
     def __init__(self):
         print("EchoAgent initialized.")
     def __call__(self, question: str) -> str:
-        print(f"EchoAgent received: {question[:50]}...")
-        return f"You said: {question}"
+        return f"You asked: {question}"
 
-class LengthAgent:
+class WikipediaAgent:
     def __init__(self):
-        print("LengthAgent initialized.")
+        print("WikipediaAgent initialized.")
     def __call__(self, question: str) -> str:
-        print(f"LengthAgent received: {question[:50]}...")
-        return f"Length of input: {len(question)}"
+        try:
+            summary = wikipedia.summary(question, sentences=1)
+            return summary
+        except Exception as e:
+            return f"Error: {str(e)}"
 
 def run_and_submit_all(agent_name: str, profile: gr.OAuthProfile | None = None):
     """
@@ -56,10 +57,10 @@ def run_and_submit_all(agent_name: str, profile: gr.OAuthProfile | None = None):
             agent = BasicAgent()
         elif agent_name == "EchoAgent":
             agent = EchoAgent()
-        elif agent_name == "LengthAgent":
-            agent = LengthAgent()
+        elif agent_name == "WikipediaAgent":
+            agent = WikipediaAgent()
         else:
-            raise ValueError(f"Unknown agent selected: {agent_name}")
+            return f"Unknown agent selected: {agent_name}", None
     except Exception as e:
         print(f"Error instantiating agent: {e}")
         return f"Error initializing agent: {e}", None
@@ -161,24 +162,24 @@ def run_and_submit_all(agent_name: str, profile: gr.OAuthProfile | None = None):
 
 # --- Build Gradio Interface using Blocks ---
 with gr.Blocks() as demo:
-    gr.Markdown("# Multi-Agent Evaluation Runner")
+    gr.Markdown("# GAIA Agent Evaluator")
 
     gr.LoginButton()
-
+    
     agent_dropdown = gr.Dropdown(
-        choices=["BasicAgent", "EchoAgent", "LengthAgent"],
-        label="Select Agent to Run",
+        choices=["BasicAgent", "EchoAgent", "WikipediaAgent"],
+        label="Select Agent",
         value="BasicAgent"
     )
 
     run_button = gr.Button("Run Evaluation & Submit All Answers")
-    status_output = gr.Textbox(label="Run Status / Submission Result", lines=5, interactive=False)
-    results_table = gr.DataFrame(label="Questions and Agent Answers", wrap=True)
+    status_output = gr.Textbox(label="Status / Score", lines=5, interactive=False)
+    results_table = gr.DataFrame(label="Agent Answers", wrap=True)
 
     run_button.click(
-    fn=run_and_submit_all,
-    inputs=[agent_dropdown],
-    outputs=[status_output, results_table]
+        fn=run_and_submit_all,
+        inputs=[agent_dropdown],
+        outputs=[status_output, results_table]
     )
 
 if __name__ == "__main__":