Update app.py

app.py

@@ -7,7 +7,8 @@ from io import StringIO
 import logging
 #from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, OpenAIServerModel
 from pathlib import Path
-from prompt_settings import verification_of_final_answer, yaml_template
+from prompt_settings import verification_of_final_answer, yaml_templateù
+from duckduckgo_search import ddg
 
 from llama_index.core import (
     VectorStoreIndex,
@@ -56,6 +57,12 @@ class BasicAgent:
                 fn=extract_ingredients,
                 description="Extracts and returns a comma-separated, alphabetized list of ingredients for a pie filling from a transcription string."
             )
+
+            search_tool = FunctionTool.from_defaults(
+                name="web_search",
+                fn=web_search,
+                description="Performs a DuckDuckGo search and returns the top 3 results."
+            )
             
             # Registra il tool
             #Settings.tools = [ingredient_tool]
@@ -67,7 +74,7 @@ class BasicAgent:
             )
     
             # Prepara l'agente
-            self.agent = OpenAIAgent.from_tools([ingredient_tool], llm=llm, verbose=True)
+            self.agent = OpenAIAgent.from_tools([ingredient_tool, search_tool], llm=llm, verbose=True)
     
             # Client OpenAI per chiamate esterne (immagini/audio)
            
@@ -535,6 +542,13 @@ def extract_ingredients(transcription: str) -> str:
     unique_ingredients = sorted(set(match.strip() for match in matches))
     return ", ".join(unique_ingredients)
 
+def web_search(query: str) -> str:
+    results = ddg(query, max_results=3)
+    if not results:
+        return "No results found."
+    return "\n".join([f"{res['title']} - {res['href']}" for res in results])
+
+
 
 def print_coso(scritta: str):
     print(f"coso {scritta}")