Update app.py

app.py

@@ -1,7 +1,12 @@
 #!/usr/bin/env python
 # coding=utf-8
+"""
+Travel Catalogue Creator - AI Agent for Personalized Travel Planning
+Creates comprehensive travel guides with weather, itineraries, packing lists & images
+"""
 import os
 import re
+import json
 import requests
 import yaml
 from typing import Optional
@@ -11,52 +16,70 @@ from Gradio_UI import GradioUI
 # ======================
 # TRAVEL TOOLS (SMOLAGENTS-COMPLIANT)
 # ======================
+
 @tool
 def get_weather_forecast(location: str, travel_dates: str) -> str:
     """
     Get weather forecast for destination using wttr.in API.
+    
     Args:
         location: Destination city name (e.g., "Barcelona")
         travel_dates: Travel date range string (e.g., "October 15-19")
+    
     Returns:
         Weather summary string with temperature and packing recommendations
     """
     try:
+        # Clean location name for API
         clean = re.sub(r"[^a-zA-Z0-9\s]", "", location).replace(" ", "+")
         resp = requests.get(f"http://wttr.in/{clean}?format=j1", timeout=10)
         resp.raise_for_status()
         data = resp.json()
+        
         cur = data["current_condition"][0]
         temp = int(cur["temp_C"])
         cond = cur["lang_en"][0]["value"].lower()
+        
+        # Generate packing recommendations based on temperature
         if temp > 25:
             pack = "Light clothing, shorts, breathable fabrics, sunscreen, hat"
         elif temp > 15:
             pack = "Light layers, long sleeves, light jacket"
         else:
             pack = "Warm layers, jacket, beanie recommended"
+        
+        # Add rain gear if needed
         if "rain" in cond or "shower" in cond:
             pack += " + compact umbrella + waterproof jacket"
-        return f"{location} forecast: {temp}°C, {cond}. Packing: {pack}"
-    except Exception:
+        
+        return f"{location} forecast ({travel_dates}): {temp}°C, {cond}. Packing: {pack}"
+    except Exception as e:
+        # Fallback to generic forecast
         return f"{location} typical weather: 15-25°C. Pack versatile layers, light jacket, and compact umbrella."
 
+
 @tool
 def convert_currency(amount: float, from_currency: str, to_currency: str) -> str:
     """
     Convert currency amount using Frankfurter API (no API key required).
+    
     Args:
         amount: Numeric amount to convert (e.g., 1200.0)
         from_currency: Source currency ISO code (e.g., "USD")
         to_currency: Target currency ISO code (e.g., "EUR")
+    
     Returns:
         Formatted string showing conversion result and exchange rate
     """
     try:
         from_curr = from_currency.upper()[:3]
         to_curr = to_currency.upper()[:3]
+        
+        # Handle same currency
         if from_curr == to_curr:
             return f"{amount:,.0f} {from_curr} = {amount:,.0f} {to_curr} (1 {from_curr} = 1.00 {to_curr})"
+        
+        # Get exchange rate
         resp = requests.get(
             f"https://api.frankfurter.app/latest",
             params={"from": from_curr, "to": to_curr},
@@ -65,20 +88,26 @@ def convert_currency(amount: float, from_currency: str, to_currency: str) -> str
         resp.raise_for_status()
         rate = resp.json()["rates"][to_curr]
         converted = amount * rate
+        
         return f"{amount:,.0f} {from_curr} = {converted:,.0f} {to_curr} (1 {from_curr} = {rate:.2f} {to_curr})"
-    except Exception:
+    except Exception as e:
+        # Fallback to showing amount without conversion
         return f"{amount:,.0f} {from_currency} = {amount:,.0f} {to_currency} (rate unavailable)"
 
+
 @tool
 def get_time_difference(origin_city: str, destination_city: str) -> str:
     """
     Calculate time difference between origin and destination cities using heuristic mapping.
+    
     Args:
         origin_city: Traveler's home city name (e.g., "New York")
         destination_city: Travel destination city name (e.g., "Paris")
+    
     Returns:
         Human-readable string describing time difference and jet lag advice
     """
+    # Extended timezone mapping (UTC offsets)
     city_tz = {
         "new york": -5, "london": 0, "paris": 1, "tokyo": 9, "sydney": 10,
         "los angeles": -8, "chicago": -6, "mumbai": 5.5, "dubai": 4,
@@ -96,11 +125,13 @@ def get_time_difference(origin_city: str, destination_city: str) -> str:
         "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()
     dest_key = destination_city.lower().strip()
     origin_tz = city_tz.get(origin_key, 0)
     dest_tz = city_tz.get(dest_key, 0)
     diff = dest_tz - origin_tz
+    
     if diff == 0:
         return f"No time difference between {origin_city} and {destination_city}."
     elif diff > 0:
@@ -108,37 +139,73 @@ def get_time_difference(origin_city: str, destination_city: str) -> str:
     else:
         return f"{destination_city} is {abs(diff)} hours behind {origin_city}. Prepare for westward jet lag (may cause insomnia)."
 
+
 @tool
 def generate_packing_list(destination: str, weather_summary: str, trip_days: int, trip_type: str) -> str:
     """
     Generate customized packing checklist based on destination, weather, duration and trip type.
+    
     Args:
         destination: Destination city or region name (e.g., "Barcelona")
         weather_summary: Weather forecast string containing temperature and conditions
         trip_days: Total number of travel days (integer >= 1)
         trip_type: Type of trip: "city", "beach", "mountain", or "mixed"
+    
     Returns:
         Formatted multi-section packing checklist as string
     """
     weather_lower = weather_summary.lower()
+    
+    # Analyze weather conditions
     cold = any(w in weather_lower for w in ["cold", "cool", "10°c", "11°c", "12°c", "13°c", "14°c", "jacket", "below 15"])
     rain = any(w in weather_lower for w in ["rain", "shower", "drizzle", "umbrella", "precipitation"])
     hot = any(w in weather_lower for w in ["hot", "warm", "25°c", "26°c", "27°c", "28°c", "29°c", "30°c", "above 25"])
+    
+    # Calculate clothing quantities
     tops = min(trip_days, trip_days // 2 + 2)
     bottoms = min(trip_days // 2 + 1, 4)
+    
+    # Build clothing list
     clothes = [f"• Tops ({tops})", f"• Bottoms ({bottoms})"]
+    
     if cold:
-        clothes.extend(["• Warm jacket/coat", "• Long underwear (if very cold)", "• Warm socks (x3)"])
+        clothes.extend([
+            "• Warm jacket/coat",
+            "• Long underwear (if very cold)",
+            "• Warm socks (x3)"
+        ])
     elif hot:
-        clothes.extend(["• Light breathable fabrics", "• Sun hat", "• Sunglasses", "• Reef-safe sunscreen (SPF 50+)"])
+        clothes.extend([
+            "• Light breathable fabrics",
+            "• Sun hat",
+            "• Sunglasses",
+            "• Reef-safe sunscreen (SPF 50+)"
+        ])
     else:
         clothes.append("• Light jacket or sweater (essential)")
+    
     if rain:
-        clothes.extend(["• Compact travel umbrella", "• Quick-dry clothing", "• Waterproof shoes or sandals"])
+        clothes.extend([
+            "• Compact travel umbrella",
+            "• Quick-dry clothing",
+            "• Waterproof shoes or sandals"
+        ])
+    
+    # Add trip-type specific items
     if trip_type == "beach":
-        clothes.extend(["• Swimsuit (x2)", "• Beach towel", "• Flip-flops/sandals", "• Beach bag"])
+        clothes.extend([
+            "• Swimsuit (x2)",
+            "• Beach towel",
+            "• Flip-flops/sandals",
+            "• Beach bag"
+        ])
     elif trip_type == "mountain":
-        clothes.extend(["• Sturdy hiking shoes", "• Moisture-wicking base layers", "• Trekking socks (x3)", "• Daypack"])
+        clothes.extend([
+            "• Sturdy hiking shoes",
+            "• Moisture-wicking base layers",
+            "• Trekking socks (x3)",
+            "• Daypack"
+        ])
     
     return (
         f"🎒 SMART PACKING LIST ({trip_days}-day {trip_type} trip to {destination})\n"
@@ -158,25 +225,36 @@ def generate_packing_list(destination: str, weather_summary: str, trip_days: int
         "\n💡 Pro tip: Roll clothes to save space. Pack one versatile outfit per day + 1 extra day buffer."
     )
 
+
 @tool
 def build_itinerary(destination: str, attractions: str, budget_local: float, days: int) -> str:
     """
     Create realistic day-by-day travel itinerary with time allocations and budget guidance.
+    
     Args:
         destination: Destination city name (e.g., "Barcelona")
         attractions: Comma-separated list of attraction names (e.g., "Sagrada Familia, Park Guell")
         budget_local: Daily budget in local currency (float)
         days: Number of full travel days (integer >= 1)
+    
     Returns:
         Formatted multi-day itinerary with time slots and daily budget allocation
     """
+    # Parse attractions list
     att_list = [a.strip() for a in attractions.split(",") if a.strip()]
     if not att_list:
         att_list = ["Old Town exploration", "Local museum", "Scenic viewpoint", "Local market"]
-    lines = [f"🗓️ {days}-DAY REALISTIC ITINERARY: {destination}", "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"]
+    
+    lines = [
+        f"🗓️ {days}-DAY REALISTIC ITINERARY: {destination}",
+        "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"
+    ]
+    
+    # Generate day-by-day schedule
     for d in range(1, days + 1):
         primary = att_list[(d - 1) % len(att_list)]
         secondary = att_list[d % len(att_list)] if len(att_list) > 1 else "Leisurely café break"
+        
         lines.extend([
             f"\nDAY {d} | Budget: ~{budget_local:,.0f} local currency",
             f"  09:00 - 12:00  {primary} (arrive early to avoid crowds)",
@@ -184,25 +262,40 @@ def build_itinerary(destination: str, attractions: str, budget_local: float, day
             f"  14:30 - 17:00  {secondary}",
             f"  19:00+         Dinner + evening stroll (budget: ~{budget_local * 0.35:,.0f})"
         ])
+    
     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., "Lisbon")
+    
     Returns:
         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")
+        
         landmark_url = f"https://source.unsplash.com/800x600/?{clean_dest},landmark,architecture,travel"
         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:
-        return '{"landmark_image": "https://source.unsplash.com/800x600/?europe,landmark,travel", "street_scene_image": "https://source.unsplash.com/800x600/?europe,street,people"}'
+        
+        return json.dumps({
+            "landmark_image": landmark_url,
+            "street_scene_image": street_url
+        })
+    except Exception as e:
+        # Fallback to generic travel images
+        return json.dumps({
+            "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(
@@ -218,6 +311,7 @@ def assemble_catalogue(
 ) -> str:
     """
     Compile all travel research into a beautiful, structured Markdown travel catalogue.
+    
     Args:
         destination: Destination city name (e.g., "Lisbon")
         origin: Traveler's home city (e.g., "London")
@@ -228,10 +322,11 @@ def assemble_catalogue(
         itinerary: Day-by-day schedule with time allocations
         packing_list: Complete customized packing checklist
         image_urls_json: JSON string with "landmark_image" and "street_scene_image" URLs
+    
     Returns:
         Complete Markdown-formatted travel catalogue with embedded images
     """
-    import json
+    # Parse image URLs
     try:
         images = json.loads(image_urls_json)
         img1 = images.get("landmark_image", "https://source.unsplash.com/800x600/?travel,landmark")
@@ -240,25 +335,36 @@ def assemble_catalogue(
         img1 = "https://source.unsplash.com/800x600/?travel,landmark"
         img2 = "https://source.unsplash.com/800x600/?travel,street"
     
-    import re
+    # Count days in itinerary
     day_count = len(re.findall(r'DAY\s+\d+', itinerary))
     
     return f"""# 🌍 {destination} Travel Catalogue
 *Planned from {origin} • {dates} • {budget_summary}*
+
 ---
+
 ## ⏰ Time Zone Adjustment
 {timezone_info}
+
 ---
+
 ## 🌤️ Weather Forecast & Packing Guidance
 {weather}
+
 ---
+
 ## 🗓️ Your Personalized {day_count}-Day Itinerary
 {itinerary}
+
 ---
+
 ## 🎒 Complete Packing Checklist
 {packing_list}
+
 ---
+
 ## 📸 Visual Inspiration
+
 <div style="display: flex; gap: 20px; margin: 20px 0;">
 <div style="flex: 1; text-align: center;">
 <img src="{img1}" alt="{destination} landmark" style="max-width: 100%; border-radius: 8px; box-shadow: 0 4px 8px rgba(0,0,0,0.1);">
@@ -269,36 +375,49 @@ def assemble_catalogue(
 <p><em>Vibrant local atmosphere</em></p>
 </div>
 </div>
+
 ---
+
 > 💡 **Travel Pro Tips**
 > • Download offline Google Maps before departure
 > • Learn 5 basic phrases in the local language
 > • Keep digital copies of passport/insurance in cloud storage
 > • Budget 10-15% extra for spontaneous experiences
 > • Public transport passes often save 30%+ vs single tickets
+
+---
+
+**Happy Travels! ✈️**
 """
 
+
 # ======================
 # AGENT SETUP
 # ======================
-if __name__ == "__main__":
+
+def create_agent():
+    """Initialize and return the Travel Catalogue Creator agent"""
     print("🚀 Initializing Travel Catalogue Creator...")
+    
     # Load system prompt from YAML
     with open("prompts.yaml", "r") as f:
         prompt_config = yaml.safe_load(f)
     
-    # Pre-instantiate tools (CRITICAL: tools must be pre-instantiated)
+    # Pre-instantiate DuckDuckGo search tool
     web_search = DuckDuckGoSearchTool()
+    
+    # Configure model with error handling
     model = HfApiModel(
         max_tokens=2048,
         temperature=0.3,
         model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     )
     
+    # Create agent with all tools
     agent = CodeAgent(
         model=model,
         tools=[
-            web_search,  # Pre-instantiated search tool (called as "web_search" in code)
+            web_search,  # Pre-instantiated search tool
             get_weather_forecast,
             convert_currency,
             get_time_difference,
@@ -307,13 +426,25 @@ if __name__ == "__main__":
             generate_travel_images,
             assemble_catalogue,
         ],
-        max_steps=15,  # Reduced from 25 to prevent runaway execution
+        max_steps=15,
         verbosity_level=1,
         name="TravelCatalogueCreator",
-        description="Creates comprehensive, personalized travel catalogues with images",
-        prompt_templates=prompt_config,
+        description="Creates comprehensive, personalized travel catalogues with weather forecasts, custom itineraries, packing lists and visual inspiration",
+        system_prompt=prompt_config.get("system_prompt", "")
     )
     
-    # Launch UI
+    print("✅ Agent initialized successfully!")
+    return agent
+
+
+if __name__ == "__main__":
+    # Create agent
+    agent = create_agent()
+    
+    # Create uploads directory
     os.makedirs("./uploads", exist_ok=True)
-    GradioUI(agent, file_upload_folder="./uploads").launch()
+    
+    # Launch Gradio UI
+    print("🌐 Launching Gradio interface...")
+    ui = GradioUI(agent, file_upload_folder="./uploads")
+    ui.launch(share=False, server_name="0.0.0.0", server_port=7860)