Spaces:
Sleeping
Sleeping
| # Travel Companion AI Agent | |
| # This agent helps travelers with destination information, local time, weather forecasts, | |
| # currency conversion, language translation, and visual destination previews. | |
| from smolagents import CodeAgent, HfApiModel, load_tool, tool | |
| import datetime | |
| import requests | |
| import pytz | |
| import yaml | |
| import random | |
| import json | |
| import os | |
| from tools.final_answer import FinalAnswerTool | |
| from Gradio_UI import GradioUI | |
| # Load the image generation tool once, outside the function | |
| image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True) | |
| # ==================== ORIGINAL TOOLS (ENHANCED) ==================== | |
| def generate_destination_preview(destination: str) -> str: | |
| """Generates a vibrant, artistic preview image of a travel destination. | |
| Args: | |
| destination: The travel destination to visualize (e.g., 'Paris', 'Tokyo', 'Bali'). | |
| Returns: | |
| A link to the generated destination preview image. | |
| """ | |
| # List of visual styles for variety | |
| styles = [ | |
| "sunrise golden hour", | |
| "blue hour twilight", | |
| "vibrant daytime", | |
| "dramatic sunset", | |
| "night lights" | |
| ] | |
| # Select a random style for variety | |
| style = random.choice(styles) | |
| # Construct a detailed prompt for the AI model | |
| prompt = f"A beautiful travel photograph of {destination}, {style}, photorealistic, high-resolution, travel photography, highly detailed landmark view" | |
| # Use the pre-loaded image generation tool | |
| try: | |
| image_url = image_generation_tool(prompt) | |
| return f"Here's a preview of {destination}: {image_url}" | |
| except Exception as e: | |
| return f"Error generating image of {destination}: {str(e)}" | |
| def get_local_time(destination: str) -> str: | |
| """Get the current local time at a travel destination. | |
| Args: | |
| destination: A city or location name (e.g., 'Paris', 'Tokyo', 'New York'). | |
| Returns: | |
| The current local time at the specified destination. | |
| """ | |
| # Map of common tourist destinations to their timezones | |
| destination_timezones = { | |
| "london": "Europe/London", | |
| "paris": "Europe/Paris", | |
| "rome": "Europe/Rome", | |
| "madrid": "Europe/Madrid", | |
| "berlin": "Europe/Berlin", | |
| "amsterdam": "Europe/Amsterdam", | |
| "athens": "Europe/Athens", | |
| "istanbul": "Europe/Istanbul", | |
| "dubai": "Asia/Dubai", | |
| "new delhi": "Asia/Kolkata", | |
| "mumbai": "Asia/Kolkata", | |
| "bangkok": "Asia/Bangkok", | |
| "singapore": "Asia/Singapore", | |
| "tokyo": "Asia/Tokyo", | |
| "seoul": "Asia/Seoul", | |
| "beijing": "Asia/Shanghai", | |
| "shanghai": "Asia/Shanghai", | |
| "hong kong": "Asia/Hong_Kong", | |
| "sydney": "Australia/Sydney", | |
| "melbourne": "Australia/Melbourne", | |
| "auckland": "Pacific/Auckland", | |
| "fiji": "Pacific/Fiji", | |
| "honolulu": "Pacific/Honolulu", | |
| "anchorage": "America/Anchorage", | |
| "los angeles": "America/Los_Angeles", | |
| "san francisco": "America/Los_Angeles", | |
| "las vegas": "America/Los_Angeles", | |
| "denver": "America/Denver", | |
| "chicago": "America/Chicago", | |
| "houston": "America/Chicago", | |
| "new york": "America/New_York", | |
| "miami": "America/New_York", | |
| "toronto": "America/Toronto", | |
| "mexico city": "America/Mexico_City", | |
| "rio de janeiro": "America/Sao_Paulo", | |
| "sao paulo": "America/Sao_Paulo", | |
| "buenos aires": "America/Argentina/Buenos_Aires", | |
| "cairo": "Africa/Cairo", | |
| "cape town": "Africa/Johannesburg", | |
| "johannesburg": "Africa/Johannesburg", | |
| "nairobi": "Africa/Nairobi" | |
| } | |
| try: | |
| # Normalize the destination name | |
| normalized_dest = destination.lower().strip() | |
| # Find the closest matching timezone | |
| timezone = None | |
| for city, tz in destination_timezones.items(): | |
| if city in normalized_dest or normalized_dest in city: | |
| timezone = tz | |
| break | |
| if not timezone: | |
| return f"I don't have timezone information for {destination}. Please try a major city nearby." | |
| # Get current time in that timezone | |
| tz = pytz.timezone(timezone) | |
| local_time = datetime.datetime.now(tz) | |
| # Format the result | |
| formatted_time = local_time.strftime("%I:%M %p on %A, %B %d, %Y") | |
| time_diff = local_time.utcoffset().total_seconds() / 3600 | |
| sign = "+" if time_diff >= 0 else "" | |
| return f"The current local time in {destination} is {formatted_time} (UTC{sign}{int(time_diff)})" | |
| except Exception as e: | |
| return f"Error getting local time for {destination}: {str(e)}" | |
| # ==================== NEW TRAVEL-SPECIFIC TOOLS ==================== | |
| def get_weather_forecast(destination: str, days: int = 3) -> str: | |
| """Get the weather forecast for a travel destination. | |
| Args: | |
| destination: City or location name | |
| days: Number of days to forecast (default: 3) | |
| Returns: | |
| Weather forecast information for trip planning | |
| """ | |
| try: | |
| # In a production environment, you would use a real API key | |
| API_KEY = os.environ.get("WEATHER_API_KEY", "demo_key") | |
| # For demo purposes, we'll generate simulated weather data | |
| # In a real implementation, you would call an actual weather API | |
| weather_conditions = ["Sunny", "Partly Cloudy", "Cloudy", "Light Rain", "Heavy Rain", "Thunderstorms", "Windy", "Foggy", "Snow", "Clear"] | |
| # Create a deterministic but seemingly random forecast based on destination name | |
| seed = sum(ord(c) for c in destination) | |
| random.seed(seed) | |
| # Generate forecast data | |
| forecast_text = f"🌦️ Weather forecast for {destination}:\n\n" | |
| today = datetime.datetime.now() | |
| for i in range(days): | |
| day = today + datetime.timedelta(days=i) | |
| day_name = day.strftime("%A") | |
| date = day.strftime("%b %d") | |
| # "Random" but deterministic weather for the demo | |
| condition = weather_conditions[random.randint(0, len(weather_conditions)-1)] | |
| temp_high = random.randint(15, 35) # Celsius | |
| temp_low = temp_high - random.randint(5, 15) | |
| precipitation = random.randint(0, 100) if "Rain" in condition or "Snow" in condition or "Thunder" in condition else 0 | |
| forecast_text += f"• {day_name}, {date}: {condition}, {temp_low}°C to {temp_high}°C" | |
| if precipitation > 0: | |
| forecast_text += f", {precipitation}% chance of precipitation" | |
| forecast_text += "\n" | |
| # Add packing recommendations based on conditions | |
| coldest = min([forecast_text.count("Snow"), forecast_text.count("0°C")]) | |
| rainiest = forecast_text.count("Rain") + forecast_text.count("Thunder") | |
| forecast_text += "\n🧳 Packing tips: " | |
| if coldest > 0: | |
| forecast_text += "Bring warm layers and a heavy jacket. " | |
| elif "5°C" in forecast_text or "6°C" in forecast_text or "7°C" in forecast_text: | |
| forecast_text += "Pack a warm jacket and layers. " | |
| if rainiest > 0: | |
| forecast_text += "Don't forget an umbrella and waterproof footwear. " | |
| if "Sunny" in forecast_text and "30°C" in forecast_text: | |
| forecast_text += "Bring sunscreen, sunglasses, and light clothing. " | |
| return forecast_text | |
| except Exception as e: | |
| return f"Error retrieving weather data for {destination}: {str(e)}" | |
| def convert_currency(amount: float, from_currency: str, to_currency: str) -> str: | |
| """Convert an amount between currencies for travel budgeting. | |
| Args: | |
| amount: The amount to convert | |
| from_currency: Source currency code (e.g., USD, EUR) | |
| to_currency: Target currency code (e.g., JPY, GBP) | |
| Returns: | |
| Converted amount and exchange rate information | |
| """ | |
| try: | |
| # In a production environment, you would use a real API key | |
| # For demo purposes, we'll use fixed exchange rates | |
| # In a real implementation, you would call an actual currency API | |
| # Common exchange rates (as of early 2025, for demo purposes) | |
| exchange_rates = { | |
| "USD": {"EUR": 0.92, "GBP": 0.79, "JPY": 149.50, "CAD": 1.35, "AUD": 1.52, "CNY": 7.20, "INR": 83.20, "MXN": 17.05}, | |
| "EUR": {"USD": 1.09, "GBP": 0.86, "JPY": 163.00, "CAD": 1.47, "AUD": 1.66, "CNY": 7.85, "INR": 90.70, "MXN": 18.60}, | |
| "GBP": {"USD": 1.27, "EUR": 1.16, "JPY": 189.30, "CAD": 1.71, "AUD": 1.92, "CNY": 9.10, "INR": 105.30, "MXN": 21.60}, | |
| "JPY": {"USD": 0.0067, "EUR": 0.0061, "GBP": 0.0053, "CAD": 0.0090, "AUD": 0.0102, "CNY": 0.0482, "INR": 0.5565, "MXN": 0.1141}, | |
| "CAD": {"USD": 0.74, "EUR": 0.68, "GBP": 0.58, "JPY": 110.70, "AUD": 1.13, "CNY": 5.33, "INR": 61.60, "MXN": 12.60}, | |
| "AUD": {"USD": 0.66, "EUR": 0.60, "GBP": 0.52, "JPY": 98.40, "CAD": 0.89, "CNY": 4.73, "INR": 54.70, "MXN": 11.20}, | |
| "CNY": {"USD": 0.14, "EUR": 0.13, "GBP": 0.11, "JPY": 20.80, "CAD": 0.19, "AUD": 0.21, "INR": 11.60, "MXN": 2.37}, | |
| "INR": {"USD": 0.012, "EUR": 0.011, "GBP": 0.0095, "JPY": 1.80, "CAD": 0.016, "AUD": 0.018, "CNY": 0.086, "MXN": 0.205}, | |
| "MXN": {"USD": 0.059, "EUR": 0.054, "GBP": 0.046, "JPY": 8.77, "CAD": 0.079, "AUD": 0.089, "CNY": 0.422, "INR": 4.88} | |
| } | |
| # Normalize currency codes | |
| from_currency = from_currency.upper().strip() | |
| to_currency = to_currency.upper().strip() | |
| # Validate currencies | |
| if from_currency not in exchange_rates: | |
| return f"Sorry, I don't have exchange rate data for {from_currency}." | |
| if to_currency not in exchange_rates[from_currency] and to_currency != from_currency: | |
| return f"Sorry, I don't have exchange rate data from {from_currency} to {to_currency}." | |
| # If same currency, return original amount | |
| if from_currency == to_currency: | |
| return f"{amount} {from_currency} = {amount} {to_currency}" | |
| # Get exchange rate and calculate conversion | |
| rate = exchange_rates[from_currency][to_currency] | |
| converted_amount = amount * rate | |
| # Format the result | |
| return f"💱 {amount:,.2f} {from_currency} = {converted_amount:,.2f} {to_currency}\n\nExchange rate: 1 {from_currency} = {rate} {to_currency}\n\n(Note: Actual rates may vary. For planning purposes only.)" | |
| except Exception as e: | |
| return f"Error converting currency: {str(e)}" | |
| def translate_phrase(text: str, language: str) -> str: | |
| """Translate common travel phrases to a local language. | |
| Args: | |
| text: Text to translate (e.g., "Hello", "Thank you", "Where is the bathroom?") | |
| language: Target language (e.g., 'Spanish', 'Japanese', 'French') | |
| Returns: | |
| Translated text with pronunciation guide | |
| """ | |
| try: | |
| # Common travel phrases in various languages | |
| # In a production environment, you would use a real translation API | |
| language = language.lower().strip() | |
| text_lower = text.lower().strip() | |
| phrase_translations = { | |
| "hello": { | |
| "spanish": {"text": "Hola", "pronunciation": "oh-lah"}, | |
| "french": {"text": "Bonjour", "pronunciation": "bohn-zhoor"}, | |
| "italian": {"text": "Ciao", "pronunciation": "chow"}, | |
| "german": {"text": "Hallo", "pronunciation": "hah-loh"}, | |
| "japanese": {"text": "こんにちは (Konnichiwa)", "pronunciation": "kohn-nee-chee-wah"}, | |
| "mandarin": {"text": "你好 (Nǐ hǎo)", "pronunciation": "nee how"}, | |
| "arabic": {"text": "مرحبا (Marhaba)", "pronunciation": "mar-ha-ba"}, | |
| "russian": {"text": "Здравствуйте (Zdravstvuyte)", "pronunciation": "zdrah-stvooy-tye"}, | |
| "portuguese": {"text": "Olá", "pronunciation": "oh-lah"}, | |
| "thai": {"text": "สวัสดี (Sawatdee)", "pronunciation": "sa-wat-dee"} | |
| }, | |
| "thank you": { | |
| "spanish": {"text": "Gracias", "pronunciation": "grah-see-ahs"}, | |
| "french": {"text": "Merci", "pronunciation": "mair-see"}, | |
| "italian": {"text": "Grazie", "pronunciation": "graht-see-eh"}, | |
| "german": {"text": "Danke", "pronunciation": "dahn-kuh"}, | |
| "japanese": {"text": "ありがとう (Arigatou)", "pronunciation": "ah-ree-gah-toh"}, | |
| "mandarin": {"text": "谢谢 (Xièxiè)", "pronunciation": "shyeh-shyeh"}, | |
| "arabic": {"text": "شكرا (Shukran)", "pronunciation": "shoo-kran"}, | |
| "russian": {"text": "Спасибо (Spasibo)", "pronunciation": "spah-see-boh"}, | |
| "portuguese": {"text": "Obrigado/a", "pronunciation": "oh-bree-gah-doo/dah"}, | |
| "thai": {"text": "ขอบคุณ (Khop khun)", "pronunciation": "kop-koon"} | |
| }, | |
| "excuse me": { | |
| "spanish": {"text": "Disculpe", "pronunciation": "dees-kool-peh"}, | |
| "french": {"text": "Excusez-moi", "pronunciation": "ex-koo-zay mwah"}, | |
| "italian": {"text": "Scusi", "pronunciation": "skoo-zee"}, | |
| "german": {"text": "Entschuldigung", "pronunciation": "ent-shool-di-goong"}, | |
| "japanese": {"text": "すみません (Sumimasen)", "pronunciation": "soo-mee-mah-sen"}, | |
| "mandarin": {"text": "对不起 (Duìbùqǐ)", "pronunciation": "dway-boo-chee"}, | |
| "arabic": {"text": "عفوا (Afwan)", "pronunciation": "af-wan"}, | |
| "russian": {"text": "Извините (Izvinite)", "pronunciation": "eez-vee-nee-tye"}, | |
| "portuguese": {"text": "Com licença", "pronunciation": "com lee-sen-sah"}, | |
| "thai": {"text": "ขอโทษ (Kho thot)", "pronunciation": "kor-toht"} | |
| }, | |
| "where is the bathroom": { | |
| "spanish": {"text": "¿Dónde está el baño?", "pronunciation": "don-deh es-tah el ban-yo"}, | |
| "french": {"text": "Où sont les toilettes?", "pronunciation": "oo son lay twa-let"}, | |
| "italian": {"text": "Dov'è il bagno?", "pronunciation": "doh-veh eel ban-yo"}, | |
| "german": {"text": "Wo ist die Toilette?", "pronunciation": "vo ist dee twa-let-te"}, | |
| "japanese": {"text": "トイレはどこですか (Toire wa doko desu ka)", "pronunciation": "toy-reh wah doh-koh des-kah"}, | |
| "mandarin": {"text": "厕所在哪里 (Cèsuǒ zài nǎlǐ)", "pronunciation": "tsuh-swor dzeye nah-lee"}, | |
| "arabic": {"text": "أين الحمام (Ayna al-hammam)", "pronunciation": "eye-nah al-ham-mam"}, | |
| "russian": {"text": "Где туалет (Gde tualet)", "pronunciation": "g-dyeh too-ah-lyet"}, | |
| "portuguese": {"text": "Onde fica o banheiro?", "pronunciation": "on-jee fee-ka oo ban-yay-roo"}, | |
| "thai": {"text": "ห้องน้ำอยู่ที่ไหน (Hong nam yu tee nai)", "pronunciation": "hong nam yoo tee nai"} | |
| }, | |
| "how much": { | |
| "spanish": {"text": "¿Cuánto cuesta?", "pronunciation": "kwan-toh kwes-tah"}, | |
| "french": {"text": "Combien ça coûte?", "pronunciation": "kom-bee-en sa koot"}, | |
| "italian": {"text": "Quanto costa?", "pronunciation": "kwan-toh kos-tah"}, | |
| "german": {"text": "Wie viel kostet das?", "pronunciation": "vee feel kos-tet das"}, | |
| "japanese": {"text": "いくらですか (Ikura desu ka)", "pronunciation": "ee-koo-rah des-kah"}, | |
| "mandarin": {"text": "多少钱 (Duōshǎo qián)", "pronunciation": "dwor-shaow chyen"}, | |
| "arabic": {"text": "كم الثمن (Kam althaman)", "pronunciation": "kam al-tha-man"}, | |
| "russian": {"text": "Сколько это стоит (Skol'ko eto stoit)", "pronunciation": "skol-ka eh-ta stoh-eet"}, | |
| "portuguese": {"text": "Quanto custa?", "pronunciation": "kwan-too koos-tah"}, | |
| "thai": {"text": "ราคาเท่าไหร่ (Raka tao rai)", "pronunciation": "ra-ka tao-rai"} | |
| } | |
| } | |
| # Find the phrase key that most closely matches the input text | |
| matched_phrase = None | |
| for phrase in phrase_translations: | |
| if phrase in text_lower or text_lower in phrase: | |
| matched_phrase = phrase | |
| break | |
| if not matched_phrase: | |
| return f"I don't have a translation for '{text}'. Try common travel phrases like 'hello', 'thank you', 'excuse me', etc." | |
| # Find the language that most closely matches the input language | |
| matched_language = None | |
| for lang in phrase_translations[matched_phrase]: | |
| if lang in language or language in lang: | |
| matched_language = lang | |
| break | |
| if not matched_language: | |
| return f"I don't have translations for {language}. Try languages like Spanish, French, Italian, German, Japanese, etc." | |
| # Get the translation | |
| translation = phrase_translations[matched_phrase][matched_language] | |
| return f"🗣️ '{text}' in {matched_language.capitalize()}:\n\n{translation['text']}\n\nPronunciation: {translation['pronunciation']}" | |
| except Exception as e: | |
| return f"Error translating text: {str(e)}" | |
| def get_visa_requirements(nationality: str, destination: str) -> str: | |
| """Check visa requirements for traveling to a destination. | |
| Args: | |
| nationality: Traveler's passport country (e.g., 'US', 'UK', 'Canada') | |
| destination: Country to visit (e.g., 'Japan', 'France', 'Brazil') | |
| Returns: | |
| Visa requirement information | |
| """ | |
| try: | |
| # Normalize inputs | |
| nationality = nationality.lower().strip() | |
| destination = destination.lower().strip() | |
| # Map of common country names to their normalized forms | |
| country_mapping = { | |
| "us": "united states", "usa": "united states", "united states of america": "united states", | |
| "uk": "united kingdom", "britain": "united kingdom", "great britain": "united kingdom", | |
| "uae": "united arab emirates", | |
| # Add more mappings as needed | |
| } | |
| # Apply mappings if available | |
| nationality = country_mapping.get(nationality, nationality) | |
| destination = country_mapping.get(destination, destination) | |
| # Skip if same country (generally no visa needed for citizens) | |
| if nationality == destination: | |
| return f"As a citizen of {nationality.title()}, you generally don't need a visa to visit your own country." | |
| # Sample visa requirement data | |
| # In a production environment, this would be a comprehensive database or API call | |
| visa_data = { | |
| "united states": { | |
| "european union": "No visa required for stays up to 90 days", | |
| "united kingdom": "No visa required for stays up to 6 months", | |
| "japan": "No visa required for stays up to 90 days", | |
| "australia": "Electronic Travel Authority (ETA) required", | |
| "china": "Visa required, must apply in advance", | |
| "india": "e-Visa available, apply online before travel", | |
| "brazil": "No visa required for stays up to 90 days", | |
| "mexico": "No visa required for stays up to 180 days", | |
| "south africa": "No visa required for stays up to 90 days", | |
| "thailand": "No visa required for stays up to 30 days" | |
| }, | |
| "united kingdom": { | |
| "european union": "No visa required for stays up to 90 days", | |
| "united states": "ESTA required for entry", | |
| "japan": "No visa required for stays up to 90 days", | |
| "australia": "eVisitor visa required", | |
| "china": "Visa required, must apply in advance", | |
| "india": "e-Visa available, apply online before travel", | |
| "brazil": "No visa required for stays up to 90 days", | |
| "mexico": "No visa required for stays up to 180 days", | |
| "south africa": "No visa required for stays up to 90 days", | |
| "thailand": "No visa required for stays up to 30 days" | |
| }, | |
| # Add more countries as needed | |
| } | |
| # Check if we have data for this nationality | |
| if nationality not in visa_data: | |
| return f"I don't have specific visa information for citizens of {nationality.title()}. Please check with the embassy of {destination.title()} for accurate visa requirements." | |
| # Check if we have data for this destination | |
| if destination not in visa_data[nationality]: | |
| return f"I don't have specific visa information for {nationality.title()} citizens traveling to {destination.title()}. Please check with the embassy of {destination.title()} for accurate visa requirements." | |
| # Get the visa requirements | |
| requirements = visa_data[nationality][destination] | |
| return f"🛂 Visa requirements for {nationality.title()} citizens traveling to {destination.title()}:\n\n{requirements}\n\n(Note: Visa requirements may change. Always verify with the official embassy or consulate before travel.)" | |
| except Exception as e: | |
| return f"Error retrieving visa information: {str(e)}" | |
| # ==================== AGENT SETUP ==================== | |
| final_answer = FinalAnswerTool() | |
| # # Model initialization | |
| # model = HfApiModel( | |
| # max_tokens=2096, | |
| # temperature=0.7, # Slightly higher temperature for more creative responses | |
| # model_id='Qwen/Qwen2.5-Coder-32B-Instruct', | |
| # custom_role_conversions=None, | |
| # ) | |
| model = HfApiModel( | |
| max_tokens=2096, | |
| temperature=0.7, # Slightly higher temperature for more creative responses | |
| model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud', # Use the AWS endpoint | |
| custom_role_conversions=None, | |
| ) | |
| # Load prompts | |
| with open("prompts.yaml", 'r') as stream: | |
| prompt_templates = yaml.safe_load(stream) | |
| # Add travel companion specific instructions to the prompt templates | |
| travel_agent_prompt = """ | |
| You are TravelBuddy, an AI travel companion designed to help travelers plan and navigate their journeys. | |
| Your goal is to provide helpful, accurate information about destinations, local customs, and practical travel advice. | |
| You have access to these capabilities: | |
| 1. Generate visual previews of destinations | |
| 2. Check local time at travel destinations | |
| 3. Provide weather forecasts for trip planning | |
| 4. Convert currencies for travel budgeting | |
| 5. Translate common travel phrases | |
| 6. Check visa requirements | |
| When users ask about a destination, try to provide comprehensive information by combining multiple tools. | |
| For example, if someone asks about Tokyo, consider providing the local time, weather, and a visual preview. | |
| Always be enthusiastic about travel while remaining practical and informative. | |
| Suggest off-the-beaten-path experiences when appropriate, but prioritize the specific information requested. | |
| """ | |
| # Add the travel agent prompt to the existing templates | |
| if "system_prompt" in prompt_templates: | |
| prompt_templates["system_prompt"] = travel_agent_prompt + "\n\n" + prompt_templates.get("system_prompt", "") | |
| else: | |
| prompt_templates["system_prompt"] = travel_agent_prompt | |
| # Agent setup with all travel tools | |
| agent = CodeAgent( | |
| model=model, | |
| tools=[ | |
| final_answer, | |
| generate_destination_preview, # Enhanced version of original image tool | |
| get_local_time, # Enhanced version of original time tool | |
| get_weather_forecast, # New travel tool | |
| convert_currency, # New travel tool | |
| translate_phrase, # New travel tool | |
| get_visa_requirements, # New travel tool | |
| ], | |
| max_steps=8, # Increased to allow for more tool usage in a single query | |
| verbosity_level=1, | |
| grammar=None, | |
| planning_interval=None, | |
| name="TravelBuddy", | |
| description="Your AI travel companion", | |
| prompt_templates=prompt_templates | |
| ) | |
| # Launch the UI | |
| print("Launching TravelBuddy - Your AI Travel Companion") | |
| GradioUI(agent).launch() |