Update app.py

app.py

@@ -4,8 +4,8 @@ import requests
 import pytz
 import yaml
 from tools.final_answer import FinalAnswerTool
-from typing import List, Dict, Any, Optional
-from duckduckgo_search import DDGS
+from typing import List, Dict
+from smolagents import DuckDuckGoSearchTool, tool
 
 from Gradio_UI import GradioUI
 
@@ -36,37 +36,57 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
-# Simple Germany bounding box
-GERMANY_BBOX = (5.8663, 47.2701, 15.0419, 55.0581)
+
 
 @tool
-def DuckDuckGoSearchTool(query: str, city: Optional[str] = None, limit: int = 10) -> List[Dict[str, Any]]:
-    """A tool that searches DuckDuckGo Maps for cafes in Germany.
+def search_cafes_in_germany(query: str, city: str = None, limit: int = 10) -> List[Dict]:
+    """
+    A refined tool that searches for cafes or restaurants in Germany using DuckDuckGo.
+
     Args:
-        query: A string to refine the cafe search (e.g., 'vegan', 'wifi').
-        city: Optional German city name (e.g., 'Berlin', 'Munich').
-        limit: Maximum number of cafes to return.
+        query: Search keywords (e.g., 'vegan cafe', 'Indian restaurant').
+        city: Optional German city to narrow results (e.g., 'Berlin', 'Cologne').
+        limit: Maximum number of results to return (default 10).
+
+    Returns:
+        A list of dictionaries containing structured information about each place, e.g.,
+        [
+            {
+                "name": "Cafe ABC",
+                "address": "Some street, Berlin",
+                "website": "https://cafeabc.de",
+                "phone": "+49 123 456789",
+                "rating": 4.5,
+                "lat": 52.5200,
+                "lon": 13.4050
+            },
+            ...
+        ]
     """
+    # Build the final query string
+    final_query = f"{query} in {city}" if city else query
+
     try:
-        maps_query = f"cafe {query}".strip()
-        results = []
-
-        with DDGS() as ddgs:
-            for item in ddgs.maps(maps_query, region="de-de", bbox=GERMANY_BBOX, max_results=limit):
-                results.append({
-                    "name": item.get("title"),
-                    "address": item.get("address"),
-                    "website": item.get("url"),
-                    "phone": item.get("phone"),
-                    "rating": item.get("rating"),
-                    "lat": (item.get("coordinates") or {}).get("latitude"),
-                    "lon": (item.get("coordinates") or {}).get("longitude"),
-                })
-
-        return results
+        # Use the built-in DuckDuckGoSearchTool from smolagents
+        results = DuckDuckGoSearchTool(query=final_query, limit=limit)
+
+        # Optional: normalize fields for consistency
+        normalized = []
+        for r in results:
+            normalized.append({
+                "name": r.get("name"),
+                "address": r.get("address"),
+                "website": r.get("website"),
+                "phone": r.get("phone"),
+                "rating": r.get("rating"),
+                "lat": r.get("lat"),
+                "lon": r.get("lon"),
+            })
+        return normalized
 
     except Exception as e:
-        return [{"error": f"Failed to fetch cafes: {str(e)}"}]
+        return [{"error": f"Search failed: {str(e)}"}]
+
 
 
 
@@ -91,7 +111,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, image_generation_tool, DuckDuckGoSearchTool], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, search_cafes_in_germany], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,