Update app.py

fixed agent setup

app.py

@@ -3,12 +3,13 @@
 import os
 import re
 import requests
+import yaml
 from typing import Optional
 from smolagents import CodeAgent, DuckDuckGoSearchTool, HfApiModel, tool
 from Gradio_UI import GradioUI
 
 # ======================
-# TRAVEL TOOLS (SMOLAGENTS-COMPLIANT)
+# TRAVEL TOOLS (SMOLAGENTS-COMPLIANT DOCSTRINGS)
 # ======================
 
 @tool
@@ -226,18 +227,11 @@ def generate_travel_images(destination: str) -> str:
         JSON-formatted string containing two image URLs with keys "landmark_image" and "street_scene_image"
     """
     try:
-        # 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 - use landmark/architecture keywords
         landmark_url = f"https://source.unsplash.com/800x600/?{clean_dest},landmark,architecture,travel"
-        
-        # 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:
-        # 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
@@ -278,7 +272,6 @@ def assemble_catalogue(
         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))
     
@@ -324,19 +317,22 @@ def assemble_catalogue(
 
 > 💡 **Travel Pro Tips**  
 > • Download offline Google Maps before departure  
-> • Learn 5 basic phrases in the local language (hello, thank you, please, excuse me, bill please)  
-> • Keep digital copies of passport/insurance in cloud storage + email to yourself  
+> • 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  
-> • Always carry small local currency for markets/tips
+> • Public transport passes often save 30%+ vs single tickets
 """
 
 # ======================
-# AGENT SETUP
+# AGENT SETUP (LOAD PROMPT FROM YAML)
 # ======================
 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
     web_search = DuckDuckGoSearchTool()
     
@@ -346,53 +342,6 @@ if __name__ == "__main__":
         model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
     )
     
-    # ENHANCED SYSTEM PROMPT: Prevent code generation, enforce pure tool usage
-    system_prompt = """
-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}
-   - Local customs and etiquette tips
-   - Safety considerations
-   - Typical food/transport prices
-
-2. WEATHER: Use get_weather_forecast with {destination} and {travel_dates}
-   → Capture temperature range and precipitation likelihood
-
-3. CURRENCY: Use convert_currency to convert {budget_amount} {home_currency} to {local_currency}
-   → Note the exchange rate for daily budgeting
-
-4. TIMEZONE: Use get_time_difference with {origin_city} and {destination}
-   → Calculate jet lag impact
-
-5. ITINERARY: Use build_itinerary with:
-   - destination={destination}
-   - attractions=[top attractions from Step 1]
-   - budget_local=[daily budget from Step 3]
-   - days=[trip duration]
-
-6. PACKING: Use generate_packing_list with:
-   - destination={destination}
-   - weather_summary=[output from Step 2]
-   - trip_days=[duration]
-   - trip_type="city" (default unless beach/mountain specified)
-
-7. IMAGES: CALL generate_travel_images EXACTLY ONCE with {destination}
-   → 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
-
-RULES YOU MUST FOLLOW:
-• NEVER write Python code blocks - ONLY use tool calls
-• NEVER call final_answer directly – always use assemble_catalogue first
-• 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(
         model=model,
         tools=[
@@ -402,14 +351,14 @@ RULES YOU MUST FOLLOW:
             get_time_difference,
             generate_packing_list,
             build_itinerary,
-            generate_travel_images,  # Custom reliable image generator
+            generate_travel_images,
             assemble_catalogue,
         ],
         max_steps=25,
         verbosity_level=1,
         name="TravelCatalogueCreator",
         description="Creates comprehensive, personalized travel catalogues with images",
-        prompt_templates={"system_prompt": system_prompt},
+        prompt_templates=prompt_config,
     )
     
     # Launch UI