Update app.py

Fixed DuckDuckGo import warning

app.py

@@ -3,13 +3,12 @@
 import os
 import re
 import requests
-from datetime import datetime
 from typing import Optional
-from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, load_tool, tool
+from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
 from Gradio_UI import GradioUI
 
 # ======================
-# TRAVEL TOOLS (SMOLAGENTS-COMPLIANT DOCSTRINGS)
+# TRAVEL TOOLS (SMOLAGENTS-COMPLIANT)
 # ======================
 
 @tool
@@ -101,21 +100,6 @@ def get_time_difference(origin_city: str, destination_city: str) -> str:
         "shanghai": 8, "hong kong": 8, "manila": 8, "kuala lumpur": 8,
         "jakarta": 7, "delhi": 5.5, "rio de janeiro": -3, "sao paulo": -3,
         "buenos aires": -3, "mexico city": -6, "toronto": -5, "vancouver": -8,
-        "montreal": -5, "miami": -5, "las vegas": -8, "san francisco": -8,
-        "seattle": -8, "denver": -7, "dallas": -6, "houston": -6, "atlanta": -5,
-        "boston": -5, "philadelphia": -5, "washington dc": -5, "orlando": -5,
-        "phoenix": -7, "minneapolis": -6, "detroit": -5, "pittsburgh": -5,
-        "cleveland": -5, "cincinnati": -5, "st louis": -6, "kansas city": -6,
-        "new orleans": -6, "memphis": -6, "nashville": -6, "charlotte": -5,
-        "raleigh": -5, "baltimore": -5, "milwaukee": -6, "indianapolis": -5,
-        "louisville": -5, "portland": -8, "salt lake city": -7, "albuquerque": -7,
-        "tucson": -7, "fresno": -8, "sacramento": -8, "oakland": -8, "san jose": -8,
-        "las palmas": 0, "tenerife": 0, "lisbon": 0, "madrid": 1, "barcelona": 1,
-        "valencia": 1, "seville": 1, "malaga": 1, "bilbao": 1, "zaragoza": 1,
-        "granada": 1, "alicante": 1, "murcia": 1, "palma": 1, "ibiza": 1,
-        "menorca": 1, "fuerteventura": 0, "lanzarote": 0, "tenerife": 0,
-        "gran canaria": 0, "la palma": 0, "el hierro": 0, "la gomera": 0,
-        "santa cruz de tenerife": 0, "las palmas de gran canaria": 0,
     }
     
     origin_key = origin_city.lower().strip()
@@ -224,6 +208,32 @@ def build_itinerary(destination: str, attractions: str, budget_local: float, day
     lines.append("\n💡 Pro tips: Book major attractions online in advance. Use public transport day passes for 30%+ savings.")
     return "\n".join(lines)
 
+@tool
+def generate_travel_images(destination: str) -> str:
+    """
+    Generate two realistic placeholder image URLs for the destination using Unsplash API (no key required).
+
+    Args:
+        destination: Destination city name (e.g., "Barcelona")
+
+    Returns:
+        JSON-formatted string containing two image URLs: landmark and street scene
+    """
+    try:
+        # Clean destination name for URL
+        clean_dest = re.sub(r"[^a-zA-Z\s]", "", destination).strip().replace(" ", "%20")
+        
+        # Landmark image
+        landmark_url = f"https://source.unsplash.com/800x600/?{clean_dest},landmark,architecture"
+        
+        # Street scene image
+        street_url = f"https://source.unsplash.com/800x600/?{clean_dest},street,people,culture"
+        
+        return f'{{"landmark_image": "{landmark_url}", "street_scene_image": "{street_url}"}}'
+    except Exception as e:
+        # Fallback generic travel images
+        return '{"landmark_image": "https://source.unsplash.com/800x600/?travel,landmark", "street_scene_image": "https://source.unsplash.com/800x600/?travel,street"}'
+
 @tool
 def assemble_catalogue(
     destination: str,
@@ -234,8 +244,7 @@ def assemble_catalogue(
     timezone_info: str,
     itinerary: str,
     packing_list: str,
-    image_url_1: str,
-    image_url_2: str
+    image_urls_json: str
 ) -> str:
     """
     Compile all travel research into a beautiful, structured Markdown travel catalogue.
@@ -249,12 +258,20 @@ def assemble_catalogue(
         timezone_info: Time difference and jet lag guidance
         itinerary: Day-by-day schedule with time allocations
         packing_list: Complete customized packing checklist
-        image_url_1: URL of first generated destination image (landmark)
-        image_url_2: URL of second generated destination image (street scene)
+        image_urls_json: JSON string with "landmark_image" and "street_scene_image" URLs
 
     Returns:
         Complete Markdown-formatted travel catalogue with embedded images
     """
+    import json
+    try:
+        images = json.loads(image_urls_json)
+        img1 = images.get("landmark_image", "https://source.unsplash.com/800x600/?travel")
+        img2 = images.get("street_scene_image", "https://source.unsplash.com/800x600/?street")
+    except:
+        img1 = "https://source.unsplash.com/800x600/?travel,landmark"
+        img2 = "https://source.unsplash.com/800x600/?travel,street"
+    
     return f"""# 🌍 {destination} Travel Catalogue
 *Planned from {origin} • {dates} • {budget_summary}*
 
@@ -284,11 +301,11 @@ def assemble_catalogue(
 
 <div style="display: flex; gap: 20px; margin: 20px 0;">
   <div style="flex: 1; text-align: center;">
-    <img src="{image_url_1}" alt="{destination} landmark" style="max-width: 100%; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.1);">
+    <img src="{img1}" alt="{destination} landmark" style="max-width: 100%; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.1);">
     <p><em>Iconic {destination} landmark</em></p>
   </div>
   <div style="flex: 1; text-align: center;">
-    <img src="{image_url_2}" alt="{destination} street scene" style="max-width: 100%; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.1);">
+    <img src="{img2}" alt="{destination} street scene" style="max-width: 100%; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.1);">
     <p><em>Vibrant local atmosphere</em></p>
   </div>
 </div>
@@ -310,18 +327,9 @@ def assemble_catalogue(
 if __name__ == "__main__":
     print("🚀 Initializing Travel Catalogue Creator...")
     
-    # Load tools
+    # Load tools (no external image gen tool needed - using custom generate_travel_images)
     web_search = DuckDuckGoSearchTool()
     
-    # Try to load image generation tool (fallback if unavailable)
-    try:
-        image_gen = load_tool("black-forest-labs/FLUX.1-schnell", trust_remote_code=True)
-    except Exception:
-        try:
-            image_gen = load_tool("stabilityai/stable-diffusion-3-medium", trust_remote_code=True)
-        except Exception as e:
-            raise RuntimeError(f"Failed to load image generation tool: {e}")
-    
     model = HfApiModel(
         max_tokens=2048,
         temperature=0.3,
@@ -359,14 +367,14 @@ You are TravelCatalogueCreator, an expert travel planner. ALWAYS follow this EXA
    - trip_days=[duration]
    - trip_type="city" (default unless beach/mountain specified)
 
-7. IMAGES: CALL image_generation_tool EXACTLY TWICE:
-   First call: "{destination} iconic landmark photorealistic travel photo, golden hour, 4k"
-   Second call: "{destination} vibrant street scene with locals, authentic atmosphere, travel photography"
+7. IMAGES: CALL generate_travel_images EXACTLY ONCE with {destination}
+   → Returns JSON with two image URLs (landmark + street scene)
+   → CRITICAL: You MUST use this tool - do not skip image generation
 
-8. ASSEMBLE: Use assemble_catalogue with ALL previous outputs including BOTH image URLs
+8. ASSEMBLE: Use assemble_catalogue with ALL previous outputs including image URLs JSON
 
 CRITICAL RULES:
-• NEVER skip image generation (Step 7 is mandatory - 2 calls required)
+• NEVER skip image generation (Step 7 is mandatory)
 • NEVER call final_answer directly – always use assemble_catalogue first
 • If a tool fails, retry ONCE with simplified parameters before proceeding
 • Always extract concrete values (temps, rates, times) from tool outputs
@@ -378,12 +386,12 @@ CRITICAL RULES:
         model=model,
         tools=[
             web_search,
-            image_gen,
             get_weather_forecast,
             convert_currency,
             get_time_difference,
             generate_packing_list,
             build_itinerary,
+            generate_travel_images,  # Custom reliable image generator
             assemble_catalogue,
         ],
         max_steps=25,