Update app.py

fixed budget parsing

app.py

@@ -9,18 +9,15 @@ from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
 from Gradio_UI import GradioUI
 
 # ======================
-# TRAVEL TOOLS (SMOLAGENTS-COMPLIANT DOCSTRINGS)
+# 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
     """
@@ -32,17 +29,14 @@ def get_weather_forecast(location: str, travel_dates: str) -> str:
         cur = data["current_condition"][0]
         temp = int(cur["temp_C"])
         cond = cur["lang_en"][0]["value"].lower()
-        
         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"
-        
         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} typical weather: 15-25°C. Pack versatile layers, light jacket, and compact umbrella."
@@ -51,22 +45,18 @@ def get_weather_forecast(location: str, travel_dates: str) -> str:
 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]
-        
         if from_curr == to_curr:
-            return f"{amount:,.0f} {from_curr}"
-        
+            return f"{amount:,.0f} {from_curr} = {amount:,.0f} {to_curr} (1 {from_curr} = 1.00 {to_curr})"
         resp = requests.get(
             f"https://api.frankfurter.app/latest",
             params={"from": from_curr, "to": to_curr},
@@ -77,17 +67,15 @@ def convert_currency(amount: float, from_currency: str, to_currency: str) -> str
         converted = amount * rate
         return f"{amount:,.0f} {from_curr} = {converted:,.0f} {to_curr} (1 {from_curr} = {rate:.2f} {to_curr})"
     except Exception:
-        return f"{amount:,.0f} {from_currency} (conversion rate unavailable)"
+        return f"{amount:,.0f} {from_currency} = {amount:,.0f} {to_currency} (conversion 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
     """
@@ -108,14 +96,11 @@ 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:
@@ -127,13 +112,11 @@ def get_time_difference(origin_city: str, destination_city: str) -> str:
 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
     """
@@ -141,10 +124,8 @@ def generate_packing_list(destination: str, weather_summary: str, trip_days: int
     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"])
-    
     tops = min(trip_days, trip_days // 2 + 2)
     bottoms = min(trip_days // 2 + 1, 4)
-    
     clothes = [f"• Tops ({tops})", f"• Bottoms ({bottoms})"]
     if cold:
         clothes.extend(["• Warm jacket/coat", "• Long underwear (if very cold)", "• Warm socks (x3)"])
@@ -152,15 +133,12 @@ def generate_packing_list(destination: str, weather_summary: str, trip_days: int
         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"])
-    
     if trip_type == "beach":
         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"])
-    
     return (
         f"🎒 SMART PACKING LIST ({trip_days}-day {trip_type} trip to {destination})\n"
         "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━\n"
@@ -183,35 +161,28 @@ def generate_packing_list(destination: str, weather_summary: str, trip_days: int
 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: Total budget converted to local currency (float)
+        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
     """
     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"]
-    
-    daily_budget = budget_local / max(days, 1)
     lines = [f"🗓️ {days}-DAY REALISTIC ITINERARY: {destination}", "━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━"]
-    
     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: ~{daily_budget:,.0f} local currency",
+            f"\nDAY {d} | Budget: ~{budget_local:,.0f} local currency",
             f"  09:00 - 12:00  {primary} (arrive early to avoid crowds)",
-            f"  12:30 - 14:00  Lunch at local spot (budget: ~{daily_budget * 0.25:,.0f})",
+            f"  12:30 - 14:00  Lunch at local spot (budget: ~{budget_local * 0.25:,.0f})",
             f"  14:30 - 17:00  {secondary}",
-            f"  19:00+         Dinner + evening stroll (budget: ~{daily_budget * 0.35:,.0f})"
+            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)
 
@@ -219,15 +190,13 @@ def build_itinerary(destination: str, attractions: str, budget_local: float, day
 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_dest = re.sub(r"[^a-zA-Z\s]", "", destination).strip()
+        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}"}}'
@@ -248,7 +217,6 @@ 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")
@@ -259,7 +227,6 @@ 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
     """
@@ -277,49 +244,36 @@ def assemble_catalogue(
     
     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);">
-    <p><em>Iconic {destination} landmark</em></p>
-  </div>
-  <div style="flex: 1; text-align: center;">
-    <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 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);">
+<p><em>Iconic {destination} landmark</em></p>
+</div>
+<div style="flex: 1; text-align: center;">
+<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>
-
 ---
-
-> 💡 **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  
+> 💡 **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
 """
 
@@ -328,14 +282,12 @@ def assemble_catalogue(
 # ======================
 if __name__ == "__main__":
     print("🚀 Initializing Travel Catalogue Creator...")
-    
     # Load system prompt from YAML
     with open("prompts.yaml", "r") as f:
         prompt_config = yaml.safe_load(f)
     
-    # Load tools
+    # Pre-instantiate tools (CRITICAL: tools must be pre-instantiated)
     web_search = DuckDuckGoSearchTool()
-    
     model = HfApiModel(
         max_tokens=2048,
         temperature=0.3,
@@ -345,7 +297,7 @@ if __name__ == "__main__":
     agent = CodeAgent(
         model=model,
         tools=[
-            web_search,
+            web_search,  # Pre-instantiated search tool (called as "web_search" in code)
             get_weather_forecast,
             convert_currency,
             get_time_difference,