Update app.py

app.py

@@ -16,6 +16,24 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
 # --- Basic Agent Definition ---
 # ----- THIS IS WERE YOU CAN BUILD WHAT YOU WANT ------
 
+class WikipediaSearchTool(Tool):
+    name = "wikipedia_search"
+    description = "Search Wikipedia and return a summary for a given query"
+
+    inputs = {"query": {"type": "string", "description": "Search term"}}
+    output_type = "string"
+
+    def forward(self, query: str) -> str:
+        try:
+            results = wikipedia.search(query)
+            if not results:
+                return "No results found."
+            page = wikipedia.page(results[0])
+            return page.summary[:1000]  # limit to 1000 chars for safety
+        except Exception as e:
+            return f"Error: {e}"
+
+
 class BasicAgent:
     def __init__(self):
         print("BasicAgent initialized.")
@@ -32,13 +50,7 @@ class BasicAgent:
         print(answer)'''
     
         # Add tools
-        tools = [WikipediaSearchTool(
-               user_agent="MyResearchBot (kanittha.kra@gmail.com)",
-               language="en",
-               content_type="summary",  # or "text"
-               extract_format="WIKI",
-           )
-        ]
+        tools = [tools = [WikipediaSearchTool()]
 
         '''query = "Who won the 2024 Olympic gold in swimming?"
 
@@ -60,7 +72,6 @@ class BasicAgent:
             return fixed_answer
         except Exception as e:
             return f"AGENT ERROR: {e}"
-        
 
 
     '''def __call__(self, question: str) -> str:
@@ -251,7 +262,7 @@ if __name__ == "__main__":
     # Check for SPACE_HOST and SPACE_ID at startup for information
     space_host_startup = os.getenv("SPACE_HOST")
     space_id_startup = os.getenv("SPACE_ID") # Get SPACE_ID at startup
-
+    
     if space_host_startup:
         print(f"✅ SPACE_HOST found: {space_host_startup}")
         print(f"   Runtime URL should be: https://{space_host_startup}.hf.space")
@@ -268,4 +279,11 @@ if __name__ == "__main__":
     print("-"*(60 + len(" App Starting ")) + "\n")
 
     print("Launching Gradio Interface for Basic Agent Evaluation...")
+
+    tool = WikipediaSearchTool()
+    query = "Artificial intelligence"
+    result = tool.forward(query)
+    print("🔍 Query:", query)
+    print("📄 Result:", result[:500], "...")
+    
     demo.launch(debug=True, share=False)