Update app.py

app.py

@@ -1,10 +1,9 @@
-import os
+import os 
 import json
 import requests
 from datetime import datetime, timedelta
 import gradio as gr
 import openai
-import re
 import random
 from typing import List, Dict, Any
 
@@ -13,115 +12,49 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 OPENWEATHER_API_KEY = os.getenv("OPENWEATHER_API_KEY", "")
 
 if not OPENAI_API_KEY:
-    print("⚠️ OPENAI_API_KEY not set. AI itinerary generation will fall back to curated data.")
+    print("⚠️ OPENAI_API_KEY not set. AI itinerary generation will fall back to manual mode.")
     client = None
 else:
     client = openai.OpenAI(api_key=OPENAI_API_KEY)
 
-# Global caches
+# Global attractions cache
 ATTRACTIONS_CACHE = {}
-IMAGE_CACHE = {}
-
-# -------------------- REAL IMAGE FUNCTIONS --------------------
-
-def get_wikimedia_image(attraction_name: str, city: str) -> str | None:
-    """
-    Fetch a REAL image from Wikipedia/Wikimedia Commons.
-    Returns a URL string if found, None if not available.
-    Never returns a fake or placeholder image.
-    """
-    cache_key = f"{attraction_name}_{city}".lower().replace(" ", "_")
-    if cache_key in IMAGE_CACHE:
-        return IMAGE_CACHE[cache_key]
-
-    # Try searching Wikipedia for the attraction
-    search_queries = [
-        attraction_name,
-        f"{attraction_name} {city}",
-    ]
-
-    for query in search_queries:
-        try:
-            # Step 1: Search Wikipedia for the page
-            search_url = "https://en.wikipedia.org/w/api.php"
-            search_params = {
-                "action": "query",
-                "list": "search",
-                "srsearch": query,
-                "srlimit": 1,
-                "format": "json",
-                "origin": "*"
-            }
-            resp = requests.get(search_url, params=search_params, timeout=8)
-            if resp.status_code != 200:
-                continue
-
-            results = resp.json().get("query", {}).get("search", [])
-            if not results:
-                continue
-
-            page_title = results[0]["title"]
 
-            # Step 2: Get the main image for that Wikipedia page
-            image_params = {
-                "action": "query",
-                "titles": page_title,
-                "prop": "pageimages",
-                "pithumbsize": 800,
-                "format": "json",
-                "origin": "*"
-            }
-            img_resp = requests.get(search_url, params=image_params, timeout=8)
-            if img_resp.status_code != 200:
-                continue
-
-            pages = img_resp.json().get("query", {}).get("pages", {})
-            for page in pages.values():
-                thumbnail = page.get("thumbnail", {})
-                if thumbnail.get("source"):
-                    url = thumbnail["source"]
-                    IMAGE_CACHE[cache_key] = url
-                    return url
-
-        except Exception:
-            continue
-
-    # No real image found — return None (never fake)
-    IMAGE_CACHE[cache_key] = None
-    return None
-
-
-def get_city_hero_image(city: str) -> str | None:
-    """Get a real Wikipedia image for the city itself as a hero/banner."""
-    cache_key = f"city_hero_{city}".lower().replace(" ", "_")
-    if cache_key in IMAGE_CACHE:
-        return IMAGE_CACHE[cache_key]
+# Image cache to avoid repeated API calls
+IMAGE_CACHE = {}
 
+# -------------------- IMAGE FUNCTIONS --------------------
+def get_wikipedia_image(attraction_name: str) -> str:
+    """Fetch an image for an attraction using Wikipedia API."""
+    if attraction_name in IMAGE_CACHE:
+        return IMAGE_CACHE[attraction_name]
     try:
         search_url = "https://en.wikipedia.org/w/api.php"
-        image_params = {
+        params = {
             "action": "query",
-            "titles": city,
-            "prop": "pageimages",
-            "pithumbsize": 1200,
             "format": "json",
-            "origin": "*"
+            "prop": "pageimages",
+            "piprop": "original",
+            "titles": attraction_name
         }
-        resp = requests.get(search_url, params=image_params, timeout=8)
-        if resp.status_code == 200:
-            pages = resp.json().get("query", {}).get("pages", {})
-            for page in pages.values():
-                thumbnail = page.get("thumbnail", {})
-                if thumbnail.get("source"):
-                    url = thumbnail["source"]
-                    IMAGE_CACHE[cache_key] = url
-                    return url
-    except Exception:
+        response = requests.get(search_url, params=params, timeout=5)
+        data = response.json()
+        pages = data.get("query", {}).get("pages", {})
+        for page in pages.values():
+            if "original" in page:
+                img_url = page["original"]["source"]
+                IMAGE_CACHE[attraction_name] = img_url
+                return img_url
+    except:
         pass
+    # Fallback generic image if Wikipedia fails
+    fallback = "https://upload.wikimedia.org/wikipedia/commons/6/65/No_image_available_600_x_450.svg"
+    IMAGE_CACHE[attraction_name] = fallback
+    return fallback
 
-    IMAGE_CACHE[cache_key] = None
-    return None
-
+def get_attraction_image(attraction_name: str, city: str) -> str:
+    """Fetch an attraction image using Wikipedia only (no Unsplash)."""
+    return get_wikipedia_image(attraction_name)
 
 # -------------------- WEATHER FUNCTION --------------------
 def get_weather(city: str) -> dict:
@@ -134,190 +67,104 @@ def get_weather(city: str) -> dict:
             "humidity": 60,
             "wind_speed": 10,
             "precipitation": 0,
-            "note": "Demo mode – add OpenWeather API key for real data"
+            "note": "Demo mode - Add OpenWeather API key for real data"
         }
     try:
         url = "https://api.openweathermap.org/data/2.5/weather"
-        params = {"q": city, "appid": OPENWEATHER_API_KEY, "units": "metric"}
+        params = {'q': city, 'appid': OPENWEATHER_API_KEY, 'units': 'metric'}
         response = requests.get(url, params=params, timeout=10)
         data = response.json()
         if response.status_code != 200:
             return {"error": f"Weather API error: {data.get('message', 'unknown')}"}
+        
         return {
             "city": city,
-            "temperature": data["main"]["temp"],
-            "feels_like": data["main"]["feels_like"],
-            "condition": data["weather"][0]["description"],
-            "humidity": data["main"]["humidity"],
-            "wind_speed": data["wind"]["speed"],
-            "precipitation": data.get("rain", {}).get("1h", 0),
+            "temperature": data['main']['temp'],
+            "feels_like": data['main']['feels_like'],
+            "condition": data['weather'][0]['description'],
+            "humidity": data['main']['humidity'],
+            "wind_speed": data['wind']['speed'],
+            "precipitation": data.get('rain', {}).get('1h', 0)
         }
     except Exception as e:
         return {"error": f"Weather service unavailable: {str(e)}"}
 
-
-# -------------------- ATTRACTIONS VIA OPENAI --------------------
+# -------------------- ATTRACTIONS FUNCTIONS --------------------
 def fetch_attractions_via_openai(city: str) -> List[Dict]:
-    """
-    Fetch REAL, specific attractions for ANY city using OpenAI.
-    The prompt strictly forbids generic placeholders.
-    """
+    """Fetch attractions for ANY city using OpenAI with rich descriptions."""
     if not client:
-        return get_curated_fallback(city)
-
-    prompt = f"""
-You are an expert travel guide with deep knowledge of {city}.
-
-List the 6 most famous and iconic real tourist attractions in {city}.
-
-STRICT RULES:
-- Use ONLY real, named, well-known attractions that actually exist in {city}.
-- Do NOT use generic names like "City Center", "Main Square", "Cathedral", "Museum", or "{city} [generic word]".
-- Each attraction must be a specific, recognizable landmark or site.
-- Entry fees should be realistic for {city} in USD (0 for free).
-
-Return ONLY a JSON array (no markdown, no extra text) with this exact structure:
-[
-  {{
-    "name": "Exact Real Attraction Name",
-    "description": "Vivid 2-sentence description of this specific place.",
-    "entry_fee": 0,
-    "duration_hours": 2,
-    "best_time": "morning"
-  }}
-]
-
-best_time must be one of: morning, afternoon, evening, anytime.
-"""
-
+        return get_fallback_attractions(city)
+    
     try:
+        prompt = f"""
+You are a local expert in {city}. List the top 6-8 must-visit tourist attractions in {city}. 
+For each attraction, provide:
+1. Name (the actual famous attraction name, not generic)
+2. A vivid, engaging 2-sentence description
+3. Entry fee (in USD, use numbers only - 0 for free)
+4. Visit duration in hours
+5. Best time to visit (morning/afternoon/evening)
+
+Return ONLY a JSON array with these keys: name, description, entry_fee, duration_hours, best_time.
+Use REAL attractions specific to {city}.
+"""
         response = client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[{"role": "user", "content": prompt}],
-            temperature=0.5,
-            max_tokens=1800,
+            temperature=0.7,
+            max_tokens=1500
         )
         content = response.choices[0].message.content.strip()
-
-        # Strip markdown fences if present
-        content = re.sub(r"^```json\s*", "", content)
-        content = re.sub(r"^```\s*", "", content)
-        content = re.sub(r"\s*```$", "", content)
-
+        if content.startswith('```json'):
+            content = content[7:]
+        if content.startswith('```'):
+            content = content[3:]
+        if content.endswith('```'):
+            content = content[:-3]
         attractions = json.loads(content)
-        if isinstance(attractions, dict):
-            attractions = attractions.get("attractions", [attractions])
-
+        if isinstance(attractions, dict) and "attractions" in attractions:
+            attractions = attractions["attractions"]
+        elif not isinstance(attractions, list):
+            attractions = [attractions]
         result = []
-        for a in attractions[:6]:
+        for a in attractions[:8]:
             fee = a.get("entry_fee", 0)
-            fee_display = "Free" if fee == 0 else f"${fee}"
+            fee_display = "Free" if fee == 0 else f"${fee}" if isinstance(fee, (int, float)) else str(fee)
             result.append({
-                "name": a.get("name", ""),
+                "name": a.get("name", "Unknown"),
                 "entry_fee": fee_display,
                 "duration_hours": a.get("duration_hours", 2),
-                "description": a.get("description", ""),
-                "best_time": a.get("best_time", "anytime"),
+                "description": a.get("description", f"A must-visit attraction in {city}"),
+                "best_time": a.get("best_time", "anytime")
             })
-
-        # Filter out any that look generic (safety net)
-        generic_keywords = ["city center", "main square", "cathedral", "museum", "old town", "market"]
-        result = [
-            a for a in result
-            if not any(
-                a["name"].lower() == f"{city.lower()} {kw}" or a["name"].lower() == kw
-                for kw in generic_keywords
-            )
-        ]
-
-        ATTRACTIONS_CACHE[city.strip().lower()] = result
+        ATTRACTIONS_CACHE[city.lower()] = result
         return result
-
     except Exception as e:
         print(f"OpenAI attractions error: {e}")
-        return get_curated_fallback(city)
+        return get_fallback_attractions(city)
 
-
-def get_curated_fallback(city: str) -> List[Dict]:
-    """
-    Curated real attractions for major cities when OpenAI is unavailable.
-    Only real named places — no generics.
-    """
-    curated = {
-        "nairobi": [
-            {"name": "Nairobi National Park", "description": "The world's only national park within a capital city, home to lions, rhinos, giraffes and over 400 bird species against a dramatic city skyline backdrop.", "entry_fee": "$60", "duration_hours": 4, "best_time": "morning"},
-            {"name": "David Sheldrick Wildlife Trust", "description": "World-famous elephant orphanage where you can watch baby elephants being fed and bathed — an unforgettable wildlife encounter.", "entry_fee": "$7", "duration_hours": 1, "best_time": "morning"},
-            {"name": "Giraffe Centre", "description": "A sanctuary for endangered Rothschild giraffes where you can hand-feed them from an elevated platform just inches from their faces.", "entry_fee": "$15", "duration_hours": 1, "best_time": "morning"},
-            {"name": "Karen Blixen Museum", "description": "The beautifully preserved farmhouse of 'Out of Africa' author Karen Blixen, offering a glimpse into colonial Kenya's history and culture.", "entry_fee": "$10", "duration_hours": 1.5, "best_time": "afternoon"},
-            {"name": "Nairobi National Museum", "description": "Kenya's flagship museum showcasing natural history, cultural heritage, prehistoric fossils and stunning art collections.", "entry_fee": "$12", "duration_hours": 2, "best_time": "afternoon"},
-            {"name": "Bomas of Kenya", "description": "A vibrant cultural center featuring traditional homesteads of Kenya's 42 tribes and daily performances of traditional music and dance.", "entry_fee": "$8", "duration_hours": 2, "best_time": "afternoon"},
-        ],
+def get_fallback_attractions(city: str) -> List[Dict]:
+    """Provide fallback attractions for any city."""
+    city_lower = city.lower()
+    real_attractions = {
         "paris": [
-            {"name": "Eiffel Tower", "description": "Iconic iron lattice tower offering breathtaking panoramic views of Paris from three observation levels. Built in 1889, it remains the world's most visited paid monument.", "entry_fee": "$25", "duration_hours": 2.5, "best_time": "evening"},
-            {"name": "Louvre Museum", "description": "World's largest art museum housing over 35,000 works including the Mona Lisa and Venus de Milo, set in a stunning 12th-century palace.", "entry_fee": "$20", "duration_hours": 4, "best_time": "morning"},
-            {"name": "Musée d'Orsay", "description": "Housed in a stunning Beaux-Arts railway station, this museum holds the world's finest collection of Impressionist and Post-Impressionist art.", "entry_fee": "$16", "duration_hours": 3, "best_time": "morning"},
-            {"name": "Sacré-Cœur Basilica", "description": "Magnificent white-domed basilica crowning Montmartre hill, offering panoramic views across Paris and a peaceful spiritual atmosphere.", "entry_fee": "Free", "duration_hours": 1.5, "best_time": "morning"},
-            {"name": "Palace of Versailles", "description": "Opulent royal palace with stunning Hall of Mirrors, magnificent formal gardens and the grandeur of French royal history.", "entry_fee": "$20", "duration_hours": 5, "best_time": "morning"},
-            {"name": "Seine River Cruise", "description": "Relaxing Bateaux Mouches cruise past Notre-Dame, the Louvre and the Eiffel Tower — Paris's landmarks look magical from the water.", "entry_fee": "$15", "duration_hours": 1, "best_time": "evening"},
+            {"name": "Eiffel Tower", "description": "Iconic iron lattice tower offering breathtaking panoramic views of Paris.", "entry_fee": "$25", "duration_hours": 2.5, "best_time": "evening"},
+            {"name": "Louvre Museum", "description": "World's largest art museum and historic monument.", "entry_fee": "$20", "duration_hours": 4, "best_time": "morning"},
+            {"name": "Notre-Dame Cathedral", "description": "Magnificent Gothic cathedral known for stunning architecture.", "entry_fee": "Free", "duration_hours": 1.5, "best_time": "morning"}
         ],
         "tokyo": [
-            {"name": "Senso-ji Temple", "description": "Tokyo's oldest and most famous Buddhist temple in Asakusa, with its iconic Thunder Gate and bustling Nakamise shopping street.", "entry_fee": "Free", "duration_hours": 2, "best_time": "morning"},
-            {"name": "Shibuya Crossing", "description": "The world's busiest pedestrian crossing, a mesmerizing spectacle of thousands of people crossing simultaneously in perfect choreography.", "entry_fee": "Free", "duration_hours": 1, "best_time": "evening"},
-            {"name": "Meiji Shrine", "description": "Serene Shinto shrine set in 70 hectares of forested parkland, dedicated to Emperor Meiji — a peaceful escape from city life.", "entry_fee": "Free", "duration_hours": 1.5, "best_time": "morning"},
-            {"name": "Tokyo Skytree", "description": "The world's second-tallest structure at 634m, offering unparalleled 360° views of Tokyo and on clear days, Mount Fuji.", "entry_fee": "$18", "duration_hours": 2, "best_time": "evening"},
-            {"name": "Tsukiji Outer Market", "description": "Famous seafood and food market where you can enjoy the freshest sushi breakfast, street food and browse unique Japanese kitchen goods.", "entry_fee": "Free", "duration_hours": 2, "best_time": "morning"},
-            {"name": "Shinjuku Gyoen National Garden", "description": "Stunning garden blending Japanese, French and English landscape styles — the most popular cherry blossom viewing spot in Tokyo.", "entry_fee": "$2", "duration_hours": 2, "best_time": "afternoon"},
-        ],
-        "new york": [
-            {"name": "Statue of Liberty", "description": "America's most iconic landmark, a gift from France standing 93 meters tall on Liberty Island, welcoming visitors since 1886.", "entry_fee": "$24", "duration_hours": 4, "best_time": "morning"},
-            {"name": "Central Park", "description": "Manhattan's 843-acre green oasis with iconic landmarks, lakes, meadows and world-class attractions — New York's living room.", "entry_fee": "Free", "duration_hours": 3, "best_time": "morning"},
-            {"name": "The Metropolitan Museum of Art", "description": "One of the world's greatest art museums with over 2 million works spanning 5,000 years of history across 17 curatorial departments.", "entry_fee": "$30", "duration_hours": 4, "best_time": "morning"},
-            {"name": "Brooklyn Bridge", "description": "Iconic 1883 suspension bridge offering spectacular Manhattan skyline views — best experienced by walking across it.", "entry_fee": "Free", "duration_hours": 1, "best_time": "morning"},
-            {"name": "Times Square", "description": "The glittering 'Crossroads of the World' — an electrifying spectacle of towering billboards, Broadway theaters and constant energy.", "entry_fee": "Free", "duration_hours": 1.5, "best_time": "evening"},
-            {"name": "9/11 Memorial & Museum", "description": "A deeply moving tribute to the 2001 attacks, featuring twin reflecting pools where the towers once stood and a powerful underground museum.", "entry_fee": "$29", "duration_hours": 3, "best_time": "morning"},
-        ],
-        "london": [
-            {"name": "Tower of London", "description": "Historic royal palace and fortress housing the Crown Jewels, with 1,000 years of history as a palace, prison and place of execution.", "entry_fee": "$32", "duration_hours": 3, "best_time": "morning"},
-            {"name": "British Museum", "description": "One of the world's greatest museums with over 8 million works spanning human history, including the Rosetta Stone and Elgin Marbles.", "entry_fee": "Free", "duration_hours": 3, "best_time": "morning"},
-            {"name": "Buckingham Palace", "description": "The official London residence of the British monarch, famous for the Changing of the Guard ceremony and State Rooms open in summer.", "entry_fee": "$30", "duration_hours": 2, "best_time": "morning"},
-            {"name": "Westminster Abbey", "description": "UNESCO-listed Gothic church where British monarchs are crowned and buried, with 1,000 years of royal history within its walls.", "entry_fee": "$28", "duration_hours": 2, "best_time": "morning"},
-            {"name": "Tate Modern", "description": "World-class modern and contemporary art gallery in a spectacular converted power station on the South Bank of the Thames.", "entry_fee": "Free", "duration_hours": 2.5, "best_time": "afternoon"},
-            {"name": "Hyde Park", "description": "London's most famous Royal Park covering 350 acres, home to the Serpentine Gallery, Speaker's Corner and the Diana Memorial Fountain.", "entry_fee": "Free", "duration_hours": 2, "best_time": "afternoon"},
-        ],
-        "rome": [
-            {"name": "Colosseum", "description": "The world's greatest amphitheater, built in 70-80 AD and capable of holding 50,000 spectators for gladiatorial contests and public spectacles.", "entry_fee": "$18", "duration_hours": 3, "best_time": "morning"},
-            {"name": "Vatican Museums & Sistine Chapel", "description": "One of the world's most important art collections, culminating in Michelangelo's breathtaking ceiling fresco in the Sistine Chapel.", "entry_fee": "$20", "duration_hours": 4, "best_time": "morning"},
-            {"name": "Trevi Fountain", "description": "Rome's most famous Baroque fountain, where legend says tossing a coin guarantees your return to Rome — best visited at dawn.", "entry_fee": "Free", "duration_hours": 1, "best_time": "morning"},
-            {"name": "Roman Forum", "description": "The ancient heart of Rome, a sprawling ruined complex of temples, arches and government buildings that was once the center of the Roman Empire.", "entry_fee": "$18", "duration_hours": 2, "best_time": "morning"},
-            {"name": "Pantheon", "description": "The best-preserved ancient building in Rome, a 2,000-year-old temple with a remarkable unreinforced concrete dome and open oculus.", "entry_fee": "$6", "duration_hours": 1, "best_time": "morning"},
-            {"name": "Borghese Gallery", "description": "A world-class museum in a stunning villa park, housing a breathtaking collection of Bernini sculptures and Caravaggio paintings.", "entry_fee": "$15", "duration_hours": 2, "best_time": "afternoon"},
-        ],
-        "bali": [
-            {"name": "Tanah Lot Temple", "description": "Bali's most iconic sea temple perched dramatically on a rocky outcrop in the ocean, best seen at sunset when the sky turns golden.", "entry_fee": "$4", "duration_hours": 2, "best_time": "evening"},
-            {"name": "Ubud Monkey Forest", "description": "A sacred sanctuary home to over 700 Balinese long-tailed macaques, set among ancient Hindu temples and lush tropical forest.", "entry_fee": "$5", "duration_hours": 1.5, "best_time": "morning"},
-            {"name": "Tegalalang Rice Terraces", "description": "Stunning UNESCO-recognized rice terraces north of Ubud, carved into the hillside with traditional Balinese subak irrigation — breathtaking at sunrise.", "entry_fee": "$2", "duration_hours": 2, "best_time": "morning"},
-            {"name": "Uluwatu Temple", "description": "Clifftop sea temple perched 70m above the Indian Ocean, famous for its nightly Kecak fire dance performances at sunset.", "entry_fee": "$3", "duration_hours": 2, "best_time": "evening"},
-            {"name": "Mount Batur", "description": "Active volcano offering a spectacular pre-dawn trek rewarded with breathtaking sunrise views over the caldera lake and surrounding mountains.", "entry_fee": "$40", "duration_hours": 6, "best_time": "morning"},
-            {"name": "Besakih Temple", "description": "Bali's most important and largest Hindu temple, the 'Mother Temple' complex of 23 temples sprawling up the slopes of Mount Agung.", "entry_fee": "$15", "duration_hours": 3, "best_time": "morning"},
-        ],
-        "bangkok": [
-            {"name": "Wat Pho (Temple of the Reclining Buddha)", "description": "Home to a magnificent 46-meter gilded reclining Buddha statue and one of Bangkok's oldest temples, also the birthplace of traditional Thai massage.", "entry_fee": "$4", "duration_hours": 1.5, "best_time": "morning"},
-            {"name": "Grand Palace", "description": "Bangkok's most dazzling complex of ornate buildings, temples and halls that served as the official residence of Thai kings since 1782.", "entry_fee": "$15", "duration_hours": 3, "best_time": "morning"},
-            {"name": "Wat Arun (Temple of Dawn)", "description": "Stunning riverside temple adorned with colorful porcelain mosaic spires — most spectacular at dawn and dusk from across the Chao Phraya.", "entry_fee": "$2", "duration_hours": 1.5, "best_time": "morning"},
-            {"name": "Chatuchak Weekend Market", "description": "One of the world's largest markets with over 15,000 stalls selling everything from street food to antiques, clothing and exotic pets.", "entry_fee": "Free", "duration_hours": 3, "best_time": "morning"},
-            {"name": "Chao Phraya River Cruise", "description": "Scenic boat ride along Bangkok's 'River of Kings', passing temples, palaces and traditional wooden houses on stilts.", "entry_fee": "$10", "duration_hours": 2, "best_time": "evening"},
-            {"name": "Jim Thompson House", "description": "Beautifully preserved traditional Thai house museum of the American silk entrepreneur, showcasing Asian art and antiquities in a lush garden setting.", "entry_fee": "$6", "duration_hours": 1.5, "best_time": "afternoon"},
-        ],
+            {"name": "Senso-ji Temple", "description": "Ancient Buddhist temple in Asakusa.", "entry_fee": "Free", "duration_hours": 2, "best_time": "morning"},
+            {"name": "Shibuya Crossing", "description": "Famous pedestrian scramble crossing.", "entry_fee": "Free", "duration_hours": 1, "best_time": "evening"},
+            {"name": "Tokyo Tower", "description": "Iconic red and white tower offering observation decks.", "entry_fee": "$12", "duration_hours": 1.5, "best_time": "evening"}
+        ]
     }
-
-    city_lower = city.strip().lower()
-    for key, attractions in curated.items():
-        if key in city_lower or city_lower in key:
+    for key, attractions in real_attractions.items():
+        if key in city_lower:
             return attractions
-
-    # If city not in curated list and OpenAI is unavailable, return empty with message
-    return []
-
+    return [
+        {"name": f"{city} City Center", "description": f"The vibrant heart of {city}.", "entry_fee": "Free", "duration_hours": 2, "best_time": "daytime"},
+        {"name": f"{city} Main Square", "description": f"The central gathering place in {city}.", "entry_fee": "Free", "duration_hours": 1.5, "best_time": "afternoon"}
+    ]
 
 def get_attractions(city: str) -> Dict:
     """Get attractions from cache or fetch new ones."""
@@ -325,8 +172,7 @@ def get_attractions(city: str) -> Dict:
     if city_clean in ATTRACTIONS_CACHE:
         return {"city": city, "attractions": ATTRACTIONS_CACHE[city_clean], "source": "cache"}
     attractions = fetch_attractions_via_openai(city)
-    return {"city": city, "attractions": attractions, "source": "api"}
-
+    return {"city": city, "attractions": attractions, "source": "OpenAI"}
 
 # -------------------- BUDGET CALCULATION --------------------
 def calculate_budget(num_days: int, budget_amount: float = None) -> Dict:
@@ -348,297 +194,288 @@ def calculate_budget(num_days: int, budget_amount: float = None) -> Dict:
     else:
         level = "moderate"
         daily_rates = {"accommodation": 100, "food": 60, "transport": 25, "activities": 20}
-
+    accommodation = daily_rates["accommodation"] * num_days
+    food = daily_rates["food"] * num_days
+    transport = daily_rates["transport"] * num_days
+    activities = daily_rates["activities"] * num_days
+    total = accommodation + food + transport + activities
     return {
         "level": level,
-        "accommodation": daily_rates["accommodation"] * num_days,
-        "food": daily_rates["food"] * num_days,
-        "transport": daily_rates["transport"] * num_days,
-        "activities": daily_rates["activities"] * num_days,
-        "total": sum(daily_rates[k] * num_days for k in daily_rates),
-        "daily": daily_rates,
+        "accommodation": accommodation,
+        "food": food,
+        "transport": transport,
+        "activities": activities,
+        "total": total,
+        "daily": daily_rates
     }
-
-
 # -------------------- ITINERARY GENERATION --------------------
-def generate_itinerary(destination: str, start_date: str, num_days: int,
+def generate_itinerary(destination: str, start_date: str, num_days: int, 
                        budget_amount: float, budget_currency: str, departure_city: str = ""):
-    """Main itinerary generation function."""
+    """Main itinerary generation function with beautiful formatting."""
     try:
-        if not destination or not destination.strip():
-            return "<div style='color:red;padding:20px;text-align:center;'>❌ Please enter a destination city.</div>"
+        # Validate inputs
+        if not destination:
+            return "<div style='color: red; padding: 20px; text-align: center;'>❌ Please enter a destination city.</div>"
+        
         if num_days < 1 or num_days > 14:
-            return "<div style='color:red;padding:20px;text-align:center;'>❌ Number of days must be between 1 and 14.</div>"
-
-        destination = destination.strip()
+            return "<div style='color: red; padding: 20px; text-align: center;'>❌ Number of days must be between 1 and 14.</div>"
+        
+        # Get data
         weather = get_weather(destination)
         if "error" in weather:
-            return f"<div style='color:red;padding:20px;text-align:center;'>❌ {weather['error']}</div>"
-
+            return f"<div style='color: red; padding: 20px; text-align: center;'>❌ Weather error: {weather['error']}</div>"
+        
         attractions_data = get_attractions(destination)
         attractions = attractions_data["attractions"]
-
-        if not attractions:
-            return f"""
-            <div style='color:#856404;background:#fff3cd;border:1px solid #ffc107;padding:20px;border-radius:12px;text-align:center;'>
-                ⚠️ Could not find specific attractions for <strong>{destination}</strong>.<br>
-                Please check the city name or add an OpenAI API key for AI-powered recommendations.
-            </div>"""
-
+        
+        # Calculate budget
         budget_data = calculate_budget(num_days, budget_amount)
+        
+        # Format dates
         start = datetime.strptime(start_date, "%Y-%m-%d")
         end = start + timedelta(days=int(num_days) - 1)
-
-        # Fetch city hero image (real, from Wikipedia)
-        city_img_url = get_city_hero_image(destination)
-
-        return generate_beautiful_itinerary(
-            destination, weather, attractions, budget_data,
-            num_days, start, end, budget_amount, budget_currency,
-            departure_city, city_img_url
+        
+        # Generate beautiful HTML itinerary with images
+        html = generate_beautiful_itinerary(
+            destination, weather, attractions, budget_data, 
+            num_days, start, end, budget_amount, budget_currency, departure_city
         )
-
+        
+        return html
+        
     except Exception as e:
-        return f"<div style='color:red;padding:20px;text-align:center;'>❌ Unexpected error: {str(e)}</div>"
-
-
-def generate_beautiful_itinerary(destination, weather, attractions, budget_data,
-                                 num_days, start_date, end_date, budget_amount,
-                                 budget_currency, departure_city, city_img_url):
-    """Create a beautiful itinerary HTML — only showing REAL images."""
-
+        return f"<div style='color: red; padding: 20px; text-align: center;'>❌ An unexpected error occurred: {str(e)}</div>"
+
+def generate_beautiful_itinerary(destination, weather, attractions, budget_data, 
+                                 num_days, start_date, end_date, budget_amount, 
+                                 budget_currency, departure_city):
+    """Create a stunning, visually appealing itinerary with images."""
+    
+    # Weather details
     weather_temp = f"{weather['temperature']:.1f}°C" if isinstance(weather['temperature'], (int, float)) else str(weather['temperature'])
-    weather_condition = weather["condition"].capitalize()
-
+    weather_condition = weather['condition'].capitalize()
+    
+    # Budget warning
     budget_warning = ""
-    if budget_amount and budget_data["total"] > budget_amount:
+    if budget_amount and budget_data['total'] > budget_amount:
         budget_warning = f"""
-        <div style="background:#fff3cd;border-left:4px solid #ffc107;padding:15px;border-radius:8px;margin:15px 0;">
-            <strong>⚠️ Budget Alert:</strong> Estimated costs (${budget_data['total']:.0f}) exceed your budget
-            ({budget_currency} {budget_amount:.0f}). Consider choosing budget-friendly options.
-        </div>"""
-
-    # Hero banner — show real city image or a clean text-only banner
-    if city_img_url:
-        hero_style = f"""
-            background: linear-gradient(rgba(40,0,80,0.55), rgba(40,0,80,0.55)),
-                        url('{city_img_url}') center/cover no-repeat;
-        """
-    else:
-        hero_style = "background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);"
-
-    hero_html = f"""
-    <div style="{hero_style} padding:50px 30px; border-radius:20px; color:white;
-                 margin-bottom:30px; text-align:center;">
-        <h1 style="margin:0;font-size:2.8em;text-shadow:0 2px 8px rgba(0,0,0,0.4);">🌍 {destination}</h1>
-        <p style="margin:10px 0 0;font-size:1.15em;opacity:0.95;">
-            {num_days} Days of Adventure &nbsp;·&nbsp;
-            {start_date.strftime('%B %d')} – {end_date.strftime('%B %d, %Y')}
-        </p>
-        {f"<p style='margin:6px 0 0;opacity:0.85;'>✈️ Departing from: {departure_city}</p>" if departure_city else ""}
-    </div>"""
-
-    # Quick stats
-    stats_html = f"""
-    <div style="display:grid;grid-template-columns:repeat(auto-fit,minmax(180px,1fr));gap:16px;margin-bottom:28px;">
-        <div style="background:#f8f9fa;padding:18px;border-radius:12px;text-align:center;">
-            <div style="font-size:2em;">🌤️</div>
-            <strong>{weather_temp}</strong><br><small>{weather_condition}</small>
-        </div>
-        <div style="background:#f8f9fa;padding:18px;border-radius:12px;text-align:center;">
-            <div style="font-size:2em;">📅</div>
-            <strong>{num_days} Days</strong><br><small>Full Itinerary</small>
+        <div style="background: #fff3cd; border-left: 4px solid #ffc107; padding: 15px; border-radius: 8px; margin: 15px 0;">
+            <strong>⚠️ Budget Alert:</strong> Estimated costs (${budget_data['total']:.0f}) exceed your budget (${budget_amount:.0f}). 
+            Consider reducing days or choosing budget-friendly options like street food and free attractions.
         </div>
-        <div style="background:#f8f9fa;padding:18px;border-radius:12px;text-align:center;">
-            <div style="font-size:2em;">🎯</div>
-            <strong>{len(attractions)} Attractions</strong><br><small>To Explore</small>
-        </div>
-        <div style="background:#f8f9fa;padding:18px;border-radius:12px;text-align:center;">
-            <div style="font-size:2em;">💰</div>
-            <strong>{budget_currency} {budget_data['total']:.0f}</strong><br><small>Estimated Total</small>
-        </div>
-    </div>"""
-
-    # Budget breakdown
-    level_names = {"budget": "Budget", "moderate": "Moderate", "comfortable": "Comfortable", "luxury": "Luxury"}
-    level_display = level_names.get(budget_data["level"], "Moderate")
-    budget_html = f"""
-    <div style="background:linear-gradient(135deg,#667eea 0%,#764ba2 100%);padding:22px;border-radius:12px;color:white;margin:20px 0;">
-        <h3 style="margin:0 0 15px 0;">💰 Budget Breakdown ({level_display} Travel Style)</h3>
-        <div style="display:grid;grid-template-columns:repeat(auto-fit,minmax(160px,1fr));gap:15px;">
-            <div><strong>🏨 Accommodation</strong><br>{budget_currency} {budget_data['accommodation']:.0f}<br><small>({budget_currency} {budget_data['daily']['accommodation']}/day)</small></div>
-            <div><strong>🍽️ Food & Dining</strong><br>{budget_currency} {budget_data['food']:.0f}<br><small>({budget_currency} {budget_data['daily']['food']}/day)</small></div>
-            <div><strong>🚗 Local Transport</strong><br>{budget_currency} {budget_data['transport']:.0f}<br><small>({budget_currency} {budget_data['daily']['transport']}/day)</small></div>
-            <div><strong>🎟️ Activities</strong><br>{budget_currency} {budget_data['activities']:.0f}<br><small>({budget_currency} {budget_data['daily']['activities']}/day)</small></div>
-            <div style="border-top:2px solid rgba(255,255,255,0.3);padding-top:10px;grid-column:1/-1;">
-                <strong>💰 Total Estimated Cost:</strong>
-                <span style="font-size:1.25em;"> {budget_currency} {budget_data['total']:.0f}</span>
-                {f" &nbsp;(Your budget: {budget_currency} {budget_amount:.0f})" if budget_amount else ""}
-            </div>
-        </div>
-    </div>"""
-
-    # Top Attractions — real images only
-    def attraction_image_html(attr, size="large"):
-        """Render attraction with real image or image-free card."""
-        img_url = get_wikimedia_image(attr["name"], destination)
-        time_icon = {"morning": "🌅", "afternoon": "☀️", "evening": "🌙"}.get(attr.get("best_time", ""), "🕐")
-
-        if size == "large" and img_url:
-            return f"""
-            <div style="background:white;border-radius:12px;margin-bottom:20px;overflow:hidden;
-                        box-shadow:0 2px 10px rgba(0,0,0,0.08);">
-                <div style="display:flex;flex-wrap:wrap;">
-                    <div style="flex:0 0 220px;">
-                        <img src="{img_url}" style="width:220px;height:160px;object-fit:cover;display:block;"
-                             alt="{attr['name']}" onerror="this.parentElement.style.display='none'">
-                    </div>
-                    <div style="flex:1;padding:18px;min-width:200px;">
-                        <strong style="font-size:1.1em;color:#333;">📍 {attr['name']}</strong>
-                        <div style="color:#555;margin:8px 0;font-size:0.95em;line-height:1.5;">{attr['description']}</div>
-                        <div style="display:flex;gap:15px;font-size:0.85em;color:#777;">
-                            <span>⏱️ {attr['duration_hours']} hrs</span>
-                            <span>🎟️ {attr['entry_fee']}</span>
-                            <span>{time_icon} Best: {attr.get('best_time','anytime')}</span>
-                        </div>
-                    </div>
-                </div>
-            </div>"""
-        elif size == "large":
-            # No image available — clean text card
-            return f"""
-            <div style="background:white;border-radius:12px;margin-bottom:20px;overflow:hidden;
-                        box-shadow:0 2px 10px rgba(0,0,0,0.08);padding:18px;">
-                <strong style="font-size:1.1em;color:#333;">📍 {attr['name']}</strong>
-                <div style="color:#555;margin:8px 0;font-size:0.95em;line-height:1.5;">{attr['description']}</div>
-                <div style="display:flex;gap:15px;font-size:0.85em;color:#777;">
-                    <span>⏱️ {attr['duration_hours']} hrs</span>
-                    <span>🎟️ {attr['entry_fee']}</span>
-                    <span>{time_icon} Best: {attr.get('best_time','anytime')}</span>
-                </div>
-            </div>"""
-
-        # Small card for daily schedule
-        if img_url:
-            return f"""
-            <div style="display:flex;gap:14px;margin-bottom:18px;padding:12px;
-                        background:#f8f9fa;border-radius:10px;align-items:flex-start;">
-                <img src="{img_url}" style="width:90px;height:90px;object-fit:cover;border-radius:8px;flex-shrink:0;"
-                     alt="{attr['name']}" onerror="this.style.display='none'">
-                <div>
-                    <strong style="color:#333;">{attr['name']}</strong><br>
-                    <span style="font-size:0.87em;color:#666;">{attr['description'][:110]}…</span>
-                    <div style="font-size:0.78em;color:#888;margin-top:5px;">
-                        ⏱️ {attr['duration_hours']} hrs &nbsp;|&nbsp; 🎟️ {attr['entry_fee']}
-                    </div>
+        """
+    
+    # Attractions list with images
+    attractions_html = ""
+    for attr in attractions[:6]:
+        time_icon = "🌅" if attr.get('best_time') == "morning" else "☀️" if attr.get('best_time') == "afternoon" else "🌙"
+        # Get image for this attraction
+        img_url = get_attraction_image(attr['name'], destination)
+        
+        attractions_html += f"""
+        <div style="background: white; border-radius: 12px; margin-bottom: 20px; overflow: hidden; box-shadow: 0 2px 8px rgba(0,0,0,0.1); transition: transform 0.3s;">
+            <div style="display: flex; flex-wrap: wrap;">
+                <div style="flex: 0 0 200px; overflow: hidden;">
+                    <img src="{img_url}" style="width: 100%; height: 150px; object-fit: cover;" alt="{attr['name']}">
                 </div>
-            </div>"""
-        else:
-            return f"""
-            <div style="display:flex;gap:14px;margin-bottom:18px;padding:12px;
-                        background:#f8f9fa;border-radius:10px;">
-                <div>
-                    <strong style="color:#333;">📍 {attr['name']}</strong><br>
-                    <span style="font-size:0.87em;color:#666;">{attr['description'][:110]}…</span>
-                    <div style="font-size:0.78em;color:#888;margin-top:5px;">
-                        ⏱️ {attr['duration_hours']} hrs &nbsp;|&nbsp; 🎟️ {attr['entry_fee']}
+                <div style="flex: 1; padding: 15px;">
+                    <div style="display: flex; justify-content: space-between; align-items: start;">
+                        <div style="flex: 1;">
+                            <strong style="font-size: 1.1em; color: #333;">📍 {attr['name']}</strong>
+                            <div style="color: #666; margin: 8px 0; font-size: 0.95em;">{attr['description']}</div>
+                            <div style="display: flex; gap: 15px; margin-top: 8px; font-size: 0.85em;">
+                                <span>⏱️ {attr['duration_hours']} hrs</span>
+                                <span>🎟️ {attr['entry_fee']}</span>
+                                <span>{time_icon} Best: {attr.get('best_time', 'anytime')}</span>
+                            </div>
+                        </div>
                     </div>
                 </div>
-            </div>"""
-
-    attractions_html = "".join(attraction_image_html(a, "large") for a in attractions[:6])
-
-    # Daily itinerary
+            </div>
+        </div>
+        """
+    
+    # Daily itinerary with images
     daily_html = ""
     per_day = max(1, len(attractions) // max(1, num_days))
+    
     for day in range(1, num_days + 1):
-        current_date = start_date + timedelta(days=day - 1)
+        current_date = start_date + timedelta(days=day-1)
         date_str = current_date.strftime("%A, %B %d")
-        start_idx = (day - 1) * per_day
+        
+        start_idx = (day-1) * per_day
         end_idx = min(day * per_day, len(attractions))
         day_attractions = attractions[start_idx:end_idx]
-        if not day_attractions and day <= len(attractions):
-            day_attractions = [attractions[day - 1]]
-
+        
         if day_attractions:
             daily_html += f"""
-            <div style="background:white;border-radius:12px;margin-bottom:20px;overflow:hidden;
-                        box-shadow:0 2px 8px rgba(0,0,0,0.06);">
-                <div style="background:linear-gradient(135deg,#667eea 0%,#764ba2 100%);
-                             padding:12px 20px;color:white;">
-                    <h3 style="margin:0;font-size:1.2em;">Day {day} · {date_str}</h3>
+            <div style="background: white; border-radius: 12px; margin-bottom: 20px; overflow: hidden; box-shadow: 0 2px 8px rgba(0,0,0,0.05);">
+                <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 12px 20px; color: white;">
+                    <h3 style="margin: 0; font-size: 1.2em;">Day {day} · {date_str}</h3>
                 </div>
-                <div style="padding:20px;">
-                    {"".join(attraction_image_html(a, "small") for a in day_attractions)}
-                    <div style="margin-top:15px;padding-top:15px;border-top:1px solid #eee;">
-                        <div style="margin-bottom:8px;">🍽️ <strong>Lunch:</strong> Ask locals for the best neighbourhood restaurant</div>
-                        <div>🌙 <strong>Evening:</strong> Explore night markets, enjoy live music or a cultural performance</div>
+                <div style="padding: 20px;">
+            """
+            
+            for attr in day_attractions:
+                img_url = get_attraction_image(attr['name'], destination)
+                daily_html += f"""
+                    <div style="display: flex; gap: 15px; margin-bottom: 20px; padding: 10px; background: #f8f9fa; border-radius: 8px;">
+                        <img src="{img_url}" style="width: 80px; height: 80px; object-fit: cover; border-radius: 8px;" alt="{attr['name']}">
+                        <div style="flex: 1;">
+                            <strong>{attr['name']}</strong><br>
+                            <span style="font-size: 0.85em; color: #666;">{attr['description'][:100]}...</span>
+                            <div style="font-size: 0.75em; color: #888; margin-top: 5px;">
+                                ⏱️ {attr['duration_hours']} hrs | 🎟️ {attr['entry_fee']}
+                            </div>
+                        </div>
+                    </div>
+                """
+            
+            daily_html += f"""
+                    <div style="margin-top: 15px; padding-top: 15px; border-top: 1px solid #e0e0e0;">
+                        <div style="margin-bottom: 10px;">
+                            <span style="font-size: 1.1em;">🍽️</span> <strong>Lunch Recommendation:</strong> Try authentic local cuisine at a nearby restaurant
+                        </div>
+                        <div>
+                            <span style="font-size: 1.1em;">🌙</span> <strong>Evening Activity:</strong> Explore local markets, enjoy a cultural show, or relax at a cafe
+                        </div>
                     </div>
                 </div>
-            </div>"""
-
-    # Tips
+            </div>
+            """
+    
+    # Budget breakdown
+    level_names = {"budget": "Budget", "moderate": "Moderate", "comfortable": "Comfortable", "luxury": "Luxury"}
+    level_display = level_names.get(budget_data['level'], "Moderate")
+    
+    budget_html = f"""
+    <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 20px; border-radius: 12px; color: white; margin: 20px 0;">
+        <h3 style="margin: 0 0 15px 0;">💰 Budget Breakdown ({level_display} Travel Style)</h3>
+        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(180px, 1fr)); gap: 15px;">
+            <div>
+                <strong>🏨 Accommodation</strong><br>
+                ${budget_data['accommodation']:.0f}<br>
+                <small>(${budget_data['daily']['accommodation']}/day)</small>
+            </div>
+            <div>
+                <strong>🍽️ Food & Dining</strong><br>
+                ${budget_data['food']:.0f}<br>
+                <small>(${budget_data['daily']['food']}/day)</small>
+            </div>
+            <div>
+                <strong>🚗 Local Transport</strong><br>
+                ${budget_data['transport']:.0f}<br>
+                <small>(${budget_data['daily']['transport']}/day)</small>
+            </div>
+            <div>
+                <strong>🎟️ Activities & Tours</strong><br>
+                ${budget_data['activities']:.0f}<br>
+                <small>(${budget_data['daily']['activities']}/day)</small>
+            </div>
+            <div style="border-top: 2px solid rgba(255,255,255,0.3); padding-top: 10px; grid-column: 1/-1;">
+                <strong>💰 Total Estimated Cost</strong><br>
+                <span style="font-size: 1.2em;">${budget_data['total']:.0f}</span>
+                {f" (Your budget: ${budget_amount:.0f})" if budget_amount else ""}
+            </div>
+        </div>
+    </div>
+    """
+    
+    # Travel tips with icons
     tips_html = """
-    <div style="background:#f0f4ff;padding:22px;border-radius:12px;margin:20px 0;">
-        <h3 style="margin:0 0 15px 0;color:#667eea;">💡 Smart Travel Tips</h3>
-        <div style="display:grid;grid-template-columns:repeat(auto-fit,minmax(200px,1fr));gap:14px;">
-            <div>🎫 <strong>Book in Advance</strong><br>Save time and money by booking popular sites online</div>
-            <div>🚇 <strong>Public Transport</strong><br>A day pass often gives unlimited travel at lower cost</div>
-            <div>📱 <strong>Offline Maps</strong><br>Download Google Maps offline before you arrive</div>
-            <div>💵 <strong>Local Currency</strong><br>Carry some cash for markets and smaller vendors</div>
-            <div>🌍 <strong>Learn Basic Phrases</strong><br>A few local words are always warmly received</div>
-            <div>📸 <strong>Go Early</strong><br>Visit popular attractions first thing to beat the crowds</div>
+    <div style="background: #f0f4ff; padding: 20px; border-radius: 12px; margin: 20px 0;">
+        <h3 style="margin: 0 0 15px 0; color: #667eea;">💡 Smart Travel Tips</h3>
+        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(200px, 1fr)); gap: 15px;">
+            <div>🎫 <strong>Book in Advance</strong><br>Save time and money by booking popular attractions online</div>
+            <div>🚇 <strong>Public Transport</strong><br>Get a day pass for unlimited travel and better savings</div>
+            <div>📱 <strong>Offline Maps</strong><br>Download Google Maps offline to navigate without data</div>
+            <div>💵 <strong>Local Currency</strong><br>Carry cash for markets and small vendors</div>
+            <div>🌍 <strong>Learn Basic Phrases</strong><br>A few local words go a long way with locals</div>
+            <div>📸 <strong>Early Bird</strong><br>Visit popular spots early morning to avoid crowds</div>
         </div>
-    </div>"""
-
-    dest_encoded = destination.replace(" ", "+")
-    dep_encoded = departure_city.lower().replace(" ", "-") if departure_city else ""
-
-    booking_html = f"""
-    <div style="text-align:center;padding:22px;background:#f8f9fa;border-radius:12px;margin-top:20px;">
-        <h3 style="margin:0 0 12px 0;">Ready to Book Your Trip?</h3>
-        <p style="margin:0;">
-            🛏️ <a href="https://www.booking.com/searchresults.html?ss={dest_encoded}" target="_blank"
-               style="color:#667eea;text-decoration:none;font-weight:600;">Search Hotels</a>
-            &nbsp;|&nbsp;
-            ✈️ <a href="https://www.skyscanner.net/transport/flights/{dep_encoded}/{destination.lower().replace(' ','-')}/"
-               target="_blank" style="color:#667eea;text-decoration:none;font-weight:600;">Search Flights</a>
-            &nbsp;|&nbsp;
-            🎟️ <a href="https://www.tripadvisor.com/Search?q={dest_encoded}" target="_blank"
-               style="color:#667eea;text-decoration:none;font-weight:600;">Read Reviews</a>
-        </p>
-    </div>"""
-
-    footer_html = """
-    <div style="text-align:center;padding:20px;margin-top:16px;font-size:0.85em;color:#888;">
-        <p>✨ TravelBuddy AI — Powered by OpenAI · Real-time Weather · Wikipedia Images · Smart Recommendations</p>
-        <p>🌍 Plan your perfect adventure with confidence</p>
-    </div>"""
-
-    return f"""
-    <div style="font-family:-apple-system,BlinkMacSystemFont,'Segoe UI',Roboto,sans-serif;max-width:100%;">
-        {hero_html}
-        {stats_html}
+    </div>
+    """
+    
+    # Complete HTML
+    full_html = f"""
+    <div style="font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, sans-serif; max-width: 100%;">
+        <!-- Hero Section -->
+        <div style="background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 40px; border-radius: 20px; color: white; margin-bottom: 30px; text-align: center;">
+            <h1 style="margin: 0; font-size: 2.5em;">🌍 {destination}</h1>
+            <p style="margin: 10px 0 0; opacity: 0.9; font-size: 1.1em;">
+                {num_days} Days of Adventure • {start_date.strftime('%B %d')} - {end_date.strftime('%B %d, %Y')}
+            </p>
+            {f"<p style='margin: 5px 0 0; opacity: 0.8;'>✈️ From: {departure_city}</p>" if departure_city else ""}
+        </div>
+        
+        <!-- Quick Stats -->
+        <div style="display: grid; grid-template-columns: repeat(auto-fit, minmax(200px, 1fr)); gap: 20px; margin-bottom: 30px;">
+            <div style="background: #f8f9fa; padding: 15px; border-radius: 12px; text-align: center;">
+                <div style="font-size: 2em;">🌤️</div>
+                <strong>{weather_temp}</strong><br>
+                <small>{weather_condition}</small>
+            </div>
+            <div style="background: #f8f9fa; padding: 15px; border-radius: 12px; text-align: center;">
+                <div style="font-size: 2em;">📅</div>
+                <strong>{num_days} Days</strong><br>
+                <small>Full Itinerary</small>
+            </div>
+            <div style="background: #f8f9fa; padding: 15px; border-radius: 12px; text-align: center;">
+                <div style="font-size: 2em;">🎯</div>
+                <strong>{len(attractions)}+ Attractions</strong><br>
+                <small>To Explore</small>
+            </div>
+            <div style="background: #f8f9fa; padding: 15px; border-radius: 12px; text-align: center;">
+                <div style="font-size: 2em;">💰</div>
+                <strong>${budget_data['total']:.0f}</strong><br>
+                <small>Estimated Total</small>
+            </div>
+        </div>
+        
         {budget_warning}
+        
+        <!-- Budget Section -->
         {budget_html}
-        <div style="background:white;padding:22px;border-radius:12px;margin:20px 0;box-shadow:0 2px 10px rgba(0,0,0,0.05);">
-            <h2 style="margin:0 0 18px 0;color:#667eea;">✨ Top Attractions in {destination}</h2>
+        
+        <!-- Top Attractions with Images -->
+        <div style="background: white; padding: 20px; border-radius: 12px; margin: 20px 0; box-shadow: 0 2px 10px rgba(0,0,0,0.05);">
+            <h2 style="margin: 0 0 15px 0; color: #667eea;">✨ Top Attractions in {destination}</h2>
             {attractions_html}
         </div>
-        <div style="background:white;padding:22px;border-radius:12px;margin:20px 0;box-shadow:0 2px 10px rgba(0,0,0,0.05);">
-            <h2 style="margin:0 0 18px 0;color:#667eea;">📅 Your {num_days}-Day Itinerary</h2>
+        
+        <!-- Daily Itinerary with Images -->
+        <div style="background: white; padding: 20px; border-radius: 12px; margin: 20px 0; box-shadow: 0 2px 10px rgba(0,0,0,0.05);">
+            <h2 style="margin: 0 0 15px 0; color: #667eea;">📅 Your {num_days}-Day Itinerary</h2>
             {daily_html}
         </div>
+        
+        <!-- Travel Tips -->
         {tips_html}
-        {booking_html}
-        {footer_html}
-    </div>"""
-
+        
+        <!-- Booking Links -->
+        <div style="text-align: center; padding: 20px; margin-top: 20px; background: #f8f9fa; border-radius: 12px;">
+            <h3 style="margin: 0 0 15px 0;">Ready to Book Your Trip?</h3>
+            <p>
+                🛏️ <a href="https://www.booking.com/searchresults.html?ss={destination.replace(' ', '+')}" target="_blank" style="color: #667eea; text-decoration: none;">Search Hotels</a> | 
+                ✈️ <a href="https://www.skyscanner.net/transport/flights/{departure_city.lower() if departure_city else ''}/{destination.lower()}/" target="_blank" style="color: #667eea; text-decoration: none;">Search Flights</a> | 
+                🎟️ <a href="https://www.tripadvisor.com/Search?q={destination}" target="_blank" style="color: #667eea; text-decoration: none;">Read Reviews</a>
+            </p>
+        </div>
+        
+        <!-- Footer -->
+        <div style="text-align: center; padding: 20px; margin-top: 20px; font-size: 0.85em; color: #666;">
+            <p>✨ TravelBuddy AI • Powered by OpenAI • Real-time Weather • Beautiful Images • Smart Recommendations</p>
+            <p>🌍 Plan your perfect adventure with confidence</p>
+        </div>
+    </div>
+    """
+    
+    return full_html
 
 # -------------------- GRADIO INTERFACE --------------------
 css = """
 .gradio-container {
-    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, Oxygen, Ubuntu, sans-serif;
     max-width: 1200px;
     margin: 0 auto;
 }
@@ -646,31 +483,44 @@ css = """
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
     border: none !important;
     font-weight: bold !important;
+    padding: 12px 30px !important;
     font-size: 1.1em !important;
+    transition: transform 0.2s !important;
+}
+.gr-button-primary:hover {
+    transform: translateY(-2px) !important;
+}
+input, select, textarea {
+    border-radius: 8px !important;
+    border: 1px solid #e0e0e0 !important;
+}
+label {
+    font-weight: 500 !important;
+    color: #333 !important;
 }
-input, select, textarea { border-radius: 8px !important; border: 1px solid #e0e0e0 !important; }
-label { font-weight: 500 !important; color: #333 !important; }
 """
 
 with gr.Blocks(css=css, title="✨ TravelBuddy AI - Smart Travel Planner") as demo:
     gr.HTML("""
-    <div style="text-align:center;background:linear-gradient(135deg,#667eea 0%,#764ba2 100%);
-                padding:40px;border-radius:20px;margin-bottom:30px;">
-        <h1 style="color:white;margin:0;font-size:2.5em;">✨ TravelBuddy AI</h1>
-        <p style="color:white;margin:10px 0 0;opacity:0.95;font-size:1.1em;">
-            Your Intelligent Travel Companion — Personalized Itineraries for Any Destination Worldwide
+    <div style="text-align: center; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); padding: 40px; border-radius: 20px; margin-bottom: 30px;">
+        <h1 style="color: white; margin: 0; font-size: 2.5em;">✨ TravelBuddy AI</h1>
+        <p style="color: white; margin: 10px 0 0; opacity: 0.95; font-size: 1.1em;">
+            Your Intelligent Travel Companion - Create Beautiful, Personalized Itineraries for Any Destination Worldwide
         </p>
-    </div>""")
-
+    </div>
+    """)
+    
     with gr.Row(equal_height=True):
         with gr.Column(scale=2):
             with gr.Group():
                 gr.Markdown("### 🎯 Where's Your Next Adventure?")
                 destination = gr.Textbox(
                     label="Destination",
-                    placeholder="e.g., Nairobi, Paris, Tokyo, Cairo, Buenos Aires, Sydney…",
-                    lines=1, show_label=False
+                    placeholder="e.g., Paris, Tokyo, New York, Bali, Cape Town...",
+                    lines=1,
+                    show_label=False
                 )
+            
             with gr.Group():
                 gr.Markdown("### 📅 When Are You Traveling?")
                 with gr.Row():
@@ -680,71 +530,73 @@ with gr.Blocks(css=css, title="✨ TravelBuddy AI - Smart Travel Planner") as de
                         placeholder="YYYY-MM-DD"
                     )
                     num_days = gr.Slider(
-                        label="Duration (Days)", minimum=1, maximum=14, value=3, step=1
+                        label="Duration (Days)",
+                        minimum=1,
+                        maximum=14,
+                        value=3,
+                        step=1
                     )
+        
         with gr.Column(scale=1):
             with gr.Group():
                 gr.Markdown("### 💰 Your Budget")
                 with gr.Row():
                     budget_amount = gr.Number(
-                        label="Total Budget (Optional)", placeholder="Enter amount", value=None
+                        label="Total Budget (Optional)",
+                        placeholder="Enter amount",
+                        value=None
                     )
                     budget_currency = gr.Dropdown(
-                        ["USD", "EUR", "GBP", "JPY", "KES", "RWF", "ZAR", "NGN",
-                         "EGP", "CAD", "AUD", "CHF", "INR", "BRL", "MXN"],
-                        label="Currency", value="USD"
+                        ["USD", "EUR", "GBP", "JPY", "CAD", "AUD", "CHF"],
+                        label="Currency",
+                        value="USD"
                     )
                 gr.HTML("""
-                <div style="background:#e8f0fe;padding:10px;border-radius:8px;margin-top:10px;">
-                    <small>💡 We'll suggest the best travel style based on your budget.</small>
-                </div>""")
+                <div style="background: #e8f0fe; padding: 10px; border-radius: 8px; margin-top: 10px;">
+                    <small>💡 <strong>Smart Tip:</strong> We'll automatically suggest the best travel style based on your budget and trip duration!</small>
+                </div>
+                """)
+            
             with gr.Group():
                 gr.Markdown("### ✈️ Departure Info")
                 departure_city = gr.Textbox(
                     label="Departure City (Optional)",
-                    placeholder="e.g., Kigali, London, Lagos, Mumbai",
+                    placeholder="e.g., New York, London, Sydney",
                     lines=1
                 )
-
+    
     with gr.Row():
-        generate_btn = gr.Button(
-            "✨ Generate My Personalized Itinerary", variant="primary", size="lg"
-        )
-
+        generate_btn = gr.Button("✨ Generate My Personalized Itinerary", variant="primary", size="lg")
+    
     output = gr.HTML()
-
+    
     generate_btn.click(
         fn=generate_itinerary,
         inputs=[destination, start_date, num_days, budget_amount, budget_currency, departure_city],
-        outputs=output,
+        outputs=output
     )
-
+    
+    # Examples section
     gr.HTML("""
-    <div style="text-align:center;padding:20px;margin-top:20px;border-top:1px solid #eee;">
+    <div style="text-align: center; padding: 20px; margin-top: 20px; border-top: 1px solid #eee;">
         <h3>🌟 Popular Destinations to Try</h3>
-        <div style="display:flex;gap:10px;justify-content:center;flex-wrap:wrap;">
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Nairobi';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🦒 Nairobi</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Paris';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🗼 Paris</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Tokyo';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🗾 Tokyo</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='New York';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🗽 New York</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Bali';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🏝️ Bali</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Cairo';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🏛️ Cairo</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Cape Town';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🌍 Cape Town</button>
-            <button onclick="document.querySelector('input[placeholder*=Nairobi]').value='Bangkok';"
-                style="background:#f0f4ff;border:none;padding:8px 16px;border-radius:20px;cursor:pointer;">🍜 Bangkok</button>
+        <div style="display: flex; gap: 10px; justify-content: center; flex-wrap: wrap;">
+            <button onclick="document.querySelector('#destination input').value='Paris';" style="background: #f0f4ff; border: none; padding: 8px 16px; border-radius: 20px; cursor: pointer;">🗼 Paris</button>
+            <button onclick="document.querySelector('#destination input').value='Tokyo';" style="background: #f0f4ff; border: none; padding: 8px 16px; border-radius: 20px; cursor: pointer;">🗾 Tokyo</button>
+            <button onclick="document.querySelector('#destination input').value='New York';" style="background: #f0f4ff; border: none; padding: 8px 16px; border-radius: 20px; cursor: pointer;">🗽 New York</button>
+            <button onclick="document.querySelector('#destination input').value='Bali';" style="background: #f0f4ff; border: none; padding: 8px 16px; border-radius: 20px; cursor: pointer;">🏝️ Bali</button>
+            <button onclick="document.querySelector('#destination input').value='Rome';" style="background: #f0f4ff; border: none; padding: 8px 16px; border-radius: 20px; cursor: pointer;">🏛️ Rome</button>
+            <button onclick="document.querySelector('#destination input').value='Bangkok';" style="background: #f0f4ff; border: none; padding: 8px 16px; border-radius: 20px; cursor: pointer;">🍜 Bangkok</button>
         </div>
     </div>
-
-    <div style="text-align:center;padding:16px;margin-top:10px;border-top:1px solid #eee;color:#888;">
-        <small>Powered by OpenAI · OpenWeather · Wikipedia Images · For travellers from every corner of the world ✨</small>
-    </div>""")
+    
+    <div style="text-align: center; padding: 20px; margin-top: 20px; border-top: 1px solid #eee; color: #666;">
+        <small>Powered by OpenAI, Weather API, Unsplash • Smart travel planning with beautiful visuals ✨</small>
+    </div>
+    """)
 
 if __name__ == "__main__":
-    demo.launch(share=False, server_name="0.0.0.0")
+    demo.launch(share=False, server_name="0.0.0.0")
+    
+    
+