Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 os.system("pip install 'smolagents[openai]'")  # ✅ Installa SmolAgents con OpenAI
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool, OpenAIServerModel
 import datetime
+import wikipediaapi
 import requests
 import pytz
 import yaml
@@ -41,6 +42,25 @@ final_answer = FinalAnswerTool()
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud' 
 
+@tool
+def search_wikipedia(query: str, lang: str = "it") -> str:
+    """Fetches a summary from Wikipedia for a given topic.
+    
+    Args:
+        query: The search term for Wikipedia.
+        lang: Language of the Wikipedia page (default: Italian).
+    
+    Returns:
+        A short summary from Wikipedia.
+    """
+    wiki_wiki = wikipediaapi.Wikipedia(lang)  # Creiamo un'istanza di Wikipedia in italiano
+    page = wiki_wiki.page(query)  # Cerchiamo la pagina
+
+    if not page.exists():
+        return f"Nessuna pagina trovata per '{query}' su Wikipedia."
+
+    return page.summary[:500] + "..."  # Restituiamo un riassunto di massimo 500 caratteri
+
 
 model = OpenAIServerModel(
     max_tokens=2096,
@@ -64,7 +84,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, image_generation_tool],
+    tools=[final_answer, get_current_time_in_timezone, image_generation_tool, search_wikipedia],
     max_steps=6,
     verbosity_level=1,
     grammar=None,