AI Travel Agent

from agno.agent import Agent
from agno.models.openai import OpenAIChat
from agno.tools.serpapi import SerpApiTools

# Researcher Agent
researcher = Agent(
    name="Researcher",
    role="Searches for travel destinations, activities, and accommodations",
    model=OpenAIChat(id="gpt-4o", api_key=openai_api_key),
    description="""
    You are a world-class travel researcher. Given a travel destination 
    and duration, generate search terms for finding relevant activities 
    and accommodations, then analyze results.
    """,
    instructions=[
        "Generate 3 search terms related to the destination and duration",
        "Search Google for each term and analyze results",
        "Return the 10 most relevant results",
    ],
    tools=[SerpApiTools(api_key=serp_api_key)],
    add_datetime_to_context=True,
)

# Planner Agent
planner = Agent(
    name="Planner",
    role="Generates itineraries based on research and preferences",
    model=OpenAIChat(id="gpt-4o", api_key=openai_api_key),
    description="""
    You are a senior travel planner. Create detailed itineraries 
    that meet user needs and preferences.
    """,
    instructions=[
        "Generate a draft itinerary with activities and accommodations",
        "Ensure well-structured, informative, and engaging content",
        "Provide balanced recommendations with proper attribution",
    ],
    add_datetime_to_context=True,
)

# Generate itinerary
research_results = researcher.run(
    f"Research {destination} for a {num_days} day trip", 
    stream=False
)

prompt = f"""
Destination: {destination}
Duration: {num_days} days
Research Results: {research_results.content}

Please create a detailed itinerary based on this research.
"""

itinerary = planner.run(prompt, stream=False)

from icalendar import Calendar, Event
from datetime import datetime, timedelta
import re

def generate_ics_content(plan_text: str, start_date: datetime = None) -> bytes:
    """Generate an ICS calendar file from a travel itinerary."""
    cal = Calendar()
    cal.add('prodid', '-//AI Travel Planner//github.com//')
    cal.add('version', '2.0')
    
    if start_date is None:
        start_date = datetime.today()
    
    # Parse itinerary by days
    day_pattern = re.compile(r'Day (\d+)[:\s]+(.*?)(?=Day \d+|$)', re.DOTALL)
    days = day_pattern.findall(plan_text)
    
    # Create events for each day
    for day_num, day_content in days:
        day_num = int(day_num)
        current_date = start_date + timedelta(days=day_num - 1)
        
        event = Event()
        event.add('summary', f"Day {day_num} Itinerary")
        event.add('description', day_content.strip())
        event.add('dtstart', current_date.date())
        event.add('dtend', current_date.date())
        event.add('dtstamp', datetime.now())
        cal.add_component(event)
    
    return cal.to_ical()