Upload 3 files

app.py

@@ -0,0 +1,254 @@
+import streamlit as st
+import os
+from dotenv import load_dotenv
+from crewai import Agent, Task, Crew, Process
+from crewai import LLM
+import litellm
+from crewai_tools import SerperDevTool
+from typing import Any, Dict
+import sys
+from datetime import datetime
+from queue import Queue
+import threading
+from referral_codes import validate_referral_code
+
+# Page config
+st.set_page_config(
+    page_title="Travel Planning Agent",
+    page_icon="🌍",
+    layout="wide"
+)
+
+# Load environment variables
+load_dotenv()
+
+def initialize_llm():
+    # Get API key
+    api_key = os.getenv("GEMINI_API_KEY")
+    if not api_key:
+        raise ValueError("GEMINI_API_KEY not found in environment variables")
+    
+    st.write(f"Debug: API Key found: {api_key[:10]}...")
+    
+    # Initialize Gemini LLM with API Key
+    return LLM(
+        model="gemini/gemini-1.5-pro-latest",
+        api_key=api_key,
+        temperature=0.7
+    )
+
+def create_travel_agent(llm):
+    return Agent(
+        role="Elite AI Travel Strategist",
+        goal=(
+            "Design the most optimized, personalized, and budget-conscious travel plans based on the user's unique preferences, "
+            "including origin, destination, budget, travel dates, group size, preferred activities, and travel style. "
+            "Leverage cutting-edge AI with real-time data from flights, hotels, transport services, local attractions, and events "
+            "to craft seamless, time-efficient, and experience-rich itineraries. "
+            "Ensure dynamic adaptability to handle disruptions, availability changes, and user modifications, creating a hassle-free travel experience."
+        ),
+        backstory=(
+            "I am an AI-powered **Master Travel Architect** with deep expertise in curating extraordinary journeys tailored to individual tastes, "
+            "whether for solo travelers, families, business professionals, or adventure seekers. "
+            "With an extensive knowledge of global destinations, cultural insights, seasonal trends, and logistical intricacies, "
+            "I meticulously craft itineraries that blend adventure, relaxation, and cultural immersion. "
+            "By integrating real-time APIs for flights, accommodations, activities, and local events, "
+            "I ensure each recommendation is accurate, cost-effective, and aligned with the traveler’s needs."
+        ),
+        personality=(
+            "Hyper-intelligent, intuitive, and proactive. I am a meticulous planner with a deep understanding of travel logistics, "
+            "capable of anticipating potential issues before they arise. "
+            "I balance **comfort, adventure, and budget efficiency**, ensuring every itinerary is customized for a truly unique experience. "
+            "I dynamically adjust plans based on live travel updates, offer personalized recommendations based on user preferences, "
+            "and present a seamless blend of structured scheduling with flexibility. "
+            "Whether it’s a luxury escape, an immersive cultural journey, or an action-packed adventure, I craft the **perfect** travel blueprint."
+        ),
+        tools=[SerperDevTool()],
+        allow_delegation=True,
+        humaninput=True,
+        llm=llm,
+        verbose=True
+    )
+
+def create_travel_task(agent, user_inputs):
+    return Task(
+        description=f"""
+            You are an **AI Travel Planner** responsible for crafting a **personalized, budget-conscious, and time-efficient travel plan** using advanced tools for data retrieval, analysis, and real-time updates.
+
+            The user has provided the following details:
+            - **Origin:** {user_inputs['origin']}
+            - **Destination:** {user_inputs['destination']}
+            - **Budget:** {user_inputs['budget']} (in local currency)
+            - **Preferred Activities:** {user_inputs['activities']}
+            - **Transport Preference:** {user_inputs['transport']}
+            - **Time Available:** {user_inputs['time']}
+
+            ## **Step 1: Generate an Optimized Itinerary**
+            - **Retrieve real-time data** (e.g., weather, crowd levels, and local events) to suggest the best times to visit attractions.
+            - Create a **structured, multi-day travel plan** that maximizes available time while balancing **sightseeing, relaxation, and local experiences**.
+            - Incorporate **seasonal and cultural insights** to enhance the travel experience.
+            - Prioritize **time-efficient routes** using mapping tools.
+
+            ## **Step 2: Smart Transport & Accommodation Planning**
+            - Suggest the most **efficient transport options** based on {user_inputs['transport']}.
+            - Compare **flights, trains, buses, or car rentals** for cost, duration, and convenience.
+            - Recommend **accommodations** within budget, balancing comfort, location, and safety.
+            - Provide estimated travel times and **real-time availability** for transport & hotels.
+            - Include **relevant booking links** for:
+              - **Flights:** Skyscanner, Google Flights, Kayak
+              - **Trains & Buses:** Omio, Rail Europe, FlixBus
+              - **Car Rentals:** Rentalcars, Turo, Enterprise
+              - **Hotels & Stays:** Booking.com, Airbnb, Agoda
+
+            ## **Step 3: Cost Estimation & Budgeting**
+            - **Break down estimated costs** for:
+              - **Transport:** (flights, public transport, or car rentals)
+              - **Accommodation:** (hotels, hostels, or rentals)
+              - **Meals & Dining:** (local food spots vs. premium dining)
+              - **Activities & Attractions:** (ticket prices, entry fees, and reservations)
+            - Offer **budget-friendly vs. premium options** while ensuring the plan stays within {user_inputs['budget']}.
+            - Suggest **discount passes, travel cards, and best booking platforms** for savings.
+            - Provide **direct booking links for tours, attractions, and dining** using:
+              - **Attractions & Tours:** GetYourGuide, Viator, Klook
+              - **Restaurant Reservations:** OpenTable, TheFork, Zomato
+
+            ## **Step 4: Flexibility & Real-Time Adjustments**
+            - **Retrieve live updates** on:
+              - **Transport delays, strikes, or disruptions.**
+              - **Weather conditions affecting travel plans.**
+              - **Local event schedules & closures.**
+            - Include **alternative activity suggestions** in case of last-minute changes.
+            - Provide **emergency contacts, visa requirements, safety tips**, and **last-minute booking options**.
+            - Offer a **summary dashboard** with real-time insights for easy decision-making.
+
+            ## **Step 5: AI-Powered Personalized Enhancements**
+            - Utilize **LLM-based recommendation engines** to personalize the itinerary based on travel history, reviews, and preferences.
+            - Suggest **AI-curated food experiences, offbeat destinations, and hidden gems**.
+            - Implement **interactive chatbot support** for instant Q&A and local guides.
+        """,
+        agent=agent,
+        expected_output="""
+            A structured, **highly optimized travel plan**, including:
+            - **Day-by-day itinerary** tailored to user preferences.
+            - **Real-time transport & accommodation recommendations** with direct booking links.
+            - **Estimated cost breakdown** with budget-conscious choices.
+            - **Dining & event recommendations** with reservation links.
+            - **Alternative options & live travel insights** for flexibility.
+            - **Safety, visa, and emergency contact details.**
+            - **Interactive AI-driven suggestions for a hassle-free experience.**
+            -**Links to the booking webistes for travel and hotel returned from the serperDevtool**
+        """
+    )
+
+def log_status(container, message):
+    """Simple function to log status messages using Streamlit's native components"""
+    timestamp = datetime.now().strftime("%H:%M:%S")
+    container.info(f"[{timestamp}] {message}")
+
+def main():
+    # Ensure environment variables are loaded
+    load_dotenv()
+    
+    st.title("🌍 AI Travel Planning Assistant")
+    
+    # Session state for referral code validation
+    if 'referral_validated' not in st.session_state:
+        st.session_state.referral_validated = False
+        st.session_state.user_type = None
+    
+    # Debug information in sidebar
+    st.sidebar.title("Debug Information")
+    st.sidebar.write(f"GEMINI_API_KEY present: {bool(os.getenv('GEMINI_API_KEY'))}")
+    
+    # Referral code validation
+    if not st.session_state.referral_validated:
+        st.write("Welcome! Please enter your referral code to access the travel planner.")
+        with st.form("referral_form"):
+            referral_code = st.text_input("Referral Code", key="referral_input")
+            submit_code = st.form_submit_button("Submit")
+            
+            if submit_code:
+                is_valid, user_type, message = validate_referral_code(referral_code)
+                if is_valid:
+                    st.session_state.referral_validated = True
+                    st.session_state.user_type = user_type
+                    st.success(f"Welcome! You have {user_type.title()} access.")
+                    st.experimental_rerun()
+                else:
+                    st.error(message)
+        return
+    
+    # Show user type in sidebar
+    st.sidebar.write(f"Access Level: {st.session_state.user_type.title()}")
+    
+    st.write("Let me help you plan your perfect trip!")
+
+    # Input form
+    with st.form("travel_form"):
+        col1, col2 = st.columns(2)
+        
+        with col1:
+            origin = st.text_input("Origin", placeholder="e.g., New York")
+            destination = st.text_input("Destination", placeholder="e.g., Paris")
+            budget = st.number_input("Budget (in local currency)", min_value=0.0, value=1000.0, step=100.0)
+        
+        with col2:
+            activities = st.text_input("Preferred Activities", placeholder="e.g., cultural, adventure, leisure")
+            transport = st.text_input("Preferred Transport", placeholder="e.g., flight, train, bus")
+            time = st.text_input("Time Available", placeholder="e.g., 5 days")
+
+        submit_button = st.form_submit_button("Generate Travel Plan")
+
+    if submit_button:
+        if not all([origin, destination, activities, transport, time]):
+            st.error("Please fill in all the fields!")
+            return
+
+        user_inputs = {
+            "origin": origin,
+            "destination": destination,
+            "budget": budget,
+            "activities": activities,
+            "transport": transport,
+            "time": time
+        }
+
+        with st.spinner("🤖 Generating your personalized travel plan... This may take a few minutes."):
+            try:
+                # Create an expander for logs
+                with st.expander("Progress Updates", expanded=True):
+                    status_container = st.empty()
+                    
+                    # Initialize components
+                    log_status(status_container, "Initializing LLM...")
+                    llm = initialize_llm()
+                    
+                    log_status(status_container, "Creating travel agent...")
+                    travel_agent = create_travel_agent(llm)
+                    
+                    log_status(status_container, "Creating travel task...")
+                    travel_task = create_travel_task(travel_agent, user_inputs)
+
+                    log_status(status_container, "Setting up crew...")
+                    crew = Crew(
+                        agents=[travel_agent],
+                        tasks=[travel_task],
+                        verbose=True,
+                        process=Process.sequential
+                    )
+
+                    log_status(status_container, "Starting travel plan generation...")
+                    results = crew.kickoff()
+                    
+                    log_status(status_container, "✨ Travel plan generation completed!")
+
+                # Display results
+                st.success("✨ Your travel plan is ready!")
+                st.markdown(results)
+
+            except Exception as e:
+                st.error(f"An error occurred: {str(e)}")
+                st.error("Please try again or contact support if the problem persists.")
+
+if __name__ == "__main__":
+    main()

referral_codes.py

@@ -0,0 +1,57 @@
+"""
+Store and validate referral codes for application access.
+"""
+
+# List of valid referral codes
+VALID_REFERRAL_CODES = {
+    "TRAVEL2025BETA1": {"uses_left": 50, "type": "beta"},
+    "WANDERLUST25A": {"uses_left": 30, "type": "premium"},
+    "EXPLORER2025B": {"uses_left": 30, "type": "premium"},
+    "JOURNEY25VIP": {"uses_left": 20, "type": "vip"},
+    "GLOBETROTTER1": {"uses_left": 25, "type": "standard"},
+    "ADVENTURE25C": {"uses_left": 25, "type": "standard"},
+    "DISCOVER25D": {"uses_left": 25, "type": "standard"},
+    "VOYAGE2025E": {"uses_left": 25, "type": "standard"},
+    "PASSPORT25F": {"uses_left": 25, "type": "standard"},
+    "COMPASS2025G": {"uses_left": 25, "type": "standard"},
+    "NOMAD2025H": {"uses_left": 25, "type": "standard"},
+    "WANDER25VIP": {"uses_left": 20, "type": "vip"},
+    "TRAVEL25PRO": {"uses_left": 30, "type": "premium"},
+    "EXPLORE25J": {"uses_left": 25, "type": "standard"},
+    "JOURNEY25K": {"uses_left": 25, "type": "standard"},
+    "VOYAGE25VIP": {"uses_left": 20, "type": "vip"},
+    "ATLAS2025M": {"uses_left": 25, "type": "standard"},
+    "COMPASS25N": {"uses_left": 25, "type": "standard"},
+    "DISCOVER25P": {"uses_left": 25, "type": "standard"},
+    "PASSPORT25Q": {"uses_left": 25, "type": "standard"},
+    "WANDER25PRO": {"uses_left": 30, "type": "premium"},
+    "TRAVEL25VIP": {"uses_left": 20, "type": "vip"},
+    "EXPLORE25T": {"uses_left": 25, "type": "standard"},
+    "JOURNEY25U": {"uses_left": 25, "type": "standard"},
+    "VOYAGE25PRO": {"uses_left": 30, "type": "premium"}
+}
+
+def validate_referral_code(code):
+    """
+    Validate a referral code and return its type if valid.
+    
+    Args:
+        code (str): The referral code to validate
+        
+    Returns:
+        tuple: (is_valid, code_type, message)
+    """
+    if not code:
+        return False, None, "Please enter a referral code"
+    
+    code = code.upper().strip()
+    if code not in VALID_REFERRAL_CODES:
+        return False, None, "Invalid referral code"
+    
+    code_data = VALID_REFERRAL_CODES[code]
+    if code_data["uses_left"] <= 0:
+        return False, None, "This referral code has expired"
+    
+    # Decrease the number of uses left
+    VALID_REFERRAL_CODES[code]["uses_left"] -= 1
+    return True, code_data["type"], "Valid referral code"

requirements.txt

@@ -0,0 +1,5 @@
+streamlit==1.31.1
+crewai==0.12.2
+python-dotenv==1.0.0
+litellm==1.16.9
+crewai-tools==0.0.15