Update app.py

Fixed Image Tool

app.py

@@ -100,6 +100,12 @@ 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,
+        "lisbon": 0, "porto": 0, "brussels": 1, "zurich": 1, "geneva": 1,
+        "milan": 1, "florence": 1, "venice": 1, "naples": 1, "dublin": 0,
+        "edinburgh": 0, "glasgow": 0, "manchester": 0, "birmingham": 0,
+        "krakow": 1, "gdansk": 1, "wroclaw": 1, "riga": 2, "vilnius": 2,
+        "tallinn": 2, "sofia": 2, "bucharest": 2, "zagreb": 1, "ljubljana": 1,
+        "belgrade": 1, "sarajevo": 1, "nicosia": 2, "valletta": 1, "reykjavik": 0,
     }
     
     origin_key = origin_city.lower().strip()
@@ -214,25 +220,25 @@ 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")
+        destination: Destination city name (e.g., "Lisbon")
 
     Returns:
-        JSON-formatted string containing two image URLs: landmark and street scene
+        JSON-formatted string containing two image URLs with keys "landmark_image" and "street_scene_image"
     """
     try:
-        # Clean destination name for URL
-        clean_dest = re.sub(r"[^a-zA-Z\s]", "", destination).strip().replace(" ", "%20")
+        # Clean destination name for URL (remove special chars, keep spaces for Unsplash)
+        clean_dest = re.sub(r"[^a-zA-Z\s]", "", destination).strip()
         
-        # Landmark image
-        landmark_url = f"https://source.unsplash.com/800x600/?{clean_dest},landmark,architecture"
+        # Landmark image - use landmark/architecture keywords
+        landmark_url = f"https://source.unsplash.com/800x600/?{clean_dest},landmark,architecture,travel"
         
-        # Street scene image
-        street_url = f"https://source.unsplash.com/800x600/?{clean_dest},street,people,culture"
+        # Street scene image - use street/people/culture keywords
+        street_url = f"https://source.unsplash.com/800x600/?{clean_dest},street,people,culture,travel"
         
         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"}'
+    except Exception:
+        # Fallback generic travel images with correct JSON keys
+        return '{"landmark_image": "https://source.unsplash.com/800x600/?europe,landmark,travel", "street_scene_image": "https://source.unsplash.com/800x600/?europe,street,people"}'
 
 @tool
 def assemble_catalogue(
@@ -250,9 +256,9 @@ def assemble_catalogue(
     Compile all travel research into a beautiful, structured Markdown travel catalogue.
 
     Args:
-        destination: Destination city name (e.g., "Barcelona")
-        origin: Traveler's home city (e.g., "New York")
-        dates: Travel date range (e.g., "Oct 15-19, 2026")
+        destination: Destination city name (e.g., "Lisbon")
+        origin: Traveler's home city (e.g., "London")
+        dates: Travel date range (e.g., "Sep 20-22, 2026")
         budget_summary: Formatted budget conversion string
         weather: Weather forecast with packing advice
         timezone_info: Time difference and jet lag guidance
@@ -266,12 +272,16 @@ def assemble_catalogue(
     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")
+        img1 = images.get("landmark_image", "https://source.unsplash.com/800x600/?travel,landmark")
+        img2 = images.get("street_scene_image", "https://source.unsplash.com/800x600/?travel,street")
     except:
         img1 = "https://source.unsplash.com/800x600/?travel,landmark"
         img2 = "https://source.unsplash.com/800x600/?travel,street"
     
+    # Count days from itinerary for header
+    import re
+    day_count = len(re.findall(r'DAY\s+\d+', itinerary))
+    
     return f"""# 🌍 {destination} Travel Catalogue
 *Planned from {origin} • {dates} • {budget_summary}*
 
@@ -287,7 +297,7 @@ def assemble_catalogue(
 
 ---
 
-## 🗓️ Your Personalized {len([d for d in itinerary.split('DAY ') if d.strip() and d[0].isdigit()])}-Day Itinerary
+## 🗓️ Your Personalized {day_count}-Day Itinerary
 {itinerary}
 
 ---
@@ -327,7 +337,7 @@ def assemble_catalogue(
 if __name__ == "__main__":
     print("🚀 Initializing Travel Catalogue Creator...")
     
-    # Load tools (no external image gen tool needed - using custom generate_travel_images)
+    # Load tools
     web_search = DuckDuckGoSearchTool()
     
     model = HfApiModel(
@@ -336,9 +346,9 @@ if __name__ == "__main__":
         model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     )
     
-    # System prompt enforcing strict 8-step workflow with mandatory image generation
+    # ENHANCED SYSTEM PROMPT: Prevent code generation, enforce pure tool usage
     system_prompt = """
-You are TravelCatalogueCreator, an expert travel planner. ALWAYS follow this EXACT 8-step workflow:
+You are TravelCatalogueCreator, an expert travel planner. You MUST follow this EXACT 8-step workflow using ONLY tool calls - NEVER write Python code:
 
 1. RESEARCH: Use DuckDuckGoSearchTool to find:
    - Top 4-5 attractions in {destination}
@@ -368,18 +378,19 @@ You are TravelCatalogueCreator, an expert travel planner. ALWAYS follow this EXA
    - trip_type="city" (default unless beach/mountain specified)
 
 7. IMAGES: CALL generate_travel_images EXACTLY ONCE with {destination}
-   → Returns JSON with two image URLs (landmark + street scene)
+   → Returns JSON with "landmark_image" and "street_scene_image" keys
    → CRITICAL: You MUST use this tool - do not skip image generation
 
 8. ASSEMBLE: Use assemble_catalogue with ALL previous outputs including image URLs JSON
 
-CRITICAL RULES:
-• NEVER skip image generation (Step 7 is mandatory)
+RULES YOU MUST FOLLOW:
+• NEVER write Python code blocks - ONLY use tool calls
 • 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
-• Budget must be converted to local currency before itinerary planning
+• ALWAYS extract concrete values (temps, rates, times) from tool outputs before next step
+• Budget must be converted to local currency BEFORE itinerary planning
 • You MUST complete all 8 steps before finishing
+• If a tool fails, retry ONCE with simplified parameters before proceeding
+• Keep responses concise and focused on travel planning
 """
 
     agent = CodeAgent(