Update app.py

app.py

@@ -1,4 +1,4 @@
-from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool
+from smolagents import CodeAgent,DuckDuckGoSearchTool, HfApiModel,load_tool,tool, wikipedia
 import datetime
 import requests
 import pytz
@@ -33,6 +33,19 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@tool
+def get_wiki_summary(query1: str) -> str:
+    """A tool that provides summary from wikipedia for a particular query.
+    Args:
+        query1: A string representing a valid title for Wikipedia (e.g., "Donald Trump")
+    """
+    try:
+        wikipedia.set_lang("en")
+        summary = get_wiki_summary(query1)
+        return summary
+    except Exception as e:
+        return f"No page found for that topic"
+
 
 final_answer = FinalAnswerTool()