Update travel.py

travel.py

@@ -92,19 +92,7 @@ itinerary_agent = Agent(
     llm=llm,
 )
 
-# 7. Final Report Generation Agent
-final_report_agent = Agent(
-    role="Final Report Generation Agent",
-    goal="Generate a comprehensive, visually appealing travel itinerary document that is easy to follow and includes all necessary details."
-        "1. Compile outputs from all agents into a cohesive travel plan. "
-        "2. Create clearly defined, organized daily schedules with times, locations, and activity durations. "
-        "3. Include practical information such as estimated costs, contact details, and reservation information. "
-        "4. Implement a clean, travel-themed design with proper spacing, alignment, and highlights. "
-        "5. Output the final itinerary as valid HTML with inline CSS for styling.",
-    backstory="A creative travel document specialist who transforms complex travel plans into beautiful, functional itineraries that enhance the travel experience.",
-    personality="Creative, detail-oriented, and user-focused.",
-    llm=llm,
-)
+
 
 # ------------------------------------------------------------------------------
 # Define Tasks
@@ -218,27 +206,27 @@ itinerary_task = Task(
     description="""Create a day-by-day itinerary for a {duration} trip to {destination} from {origin}, incorporating all recommendations.
     
     Detailed requirements:
-    - Begin with arrival logistics including airport transfer options, check-in times, and first-day orientation activities
+     - Begin with arrival logistics including airport transfer options, check-in times, and first-day orientation activities
     - Structure each day with:
         * Morning, afternoon, and evening activity blocks with precise timing
         * Estimated travel times between locations using various transportation methods
         * Buffer time for rest, spontaneous exploration, and unexpected delays
         * Meal recommendations with reservation details and backup options
         * Sunset/sunrise opportunities for optimal photography or experiences
-    - Apply intelligent sequencing to:
+     - Apply intelligent sequencing to:
         * Group attractions by geographic proximity to minimize transit time
         * Schedule indoor activities strategically for predicted weather patterns
         * Balance high-energy activities with relaxation periods
         * Alternate between cultural immersion and entertainment experiences
         * Account for opening days/hours of attractions and potential closures
-    - Include practical timing considerations:
+     - Include practical timing considerations:
         * Museum/attraction fatigue limitations
         * Jet lag recovery for first 1-2 days
         * Time zone adjustment strategies
         * Local rush hours and traffic patterns to avoid
         * Cultural norms for meal times and business hours
-    - End with departure logistics including check-out procedures, airport transfer timing, and luggage considerations
-    - Add specialized planning elements:
+     - End with departure logistics including check-out procedures, airport transfer timing, and luggage considerations
+     - Add specialized planning elements:
         * Local festivals or events coinciding with the travel dates
         * Free time blocks for personal exploration or shopping
         * Contingency recommendations for weather disruptions
@@ -264,46 +252,7 @@ itinerary_task = Task(
     Format should include both a condensed overview calendar and detailed daily breakdowns with time, activity, location, notes, and contingency plans."""
 )
 
-final_report_task = Task(
-    description="""Generate a comprehensive, visually appealing travel itinerary document for a {duration} trip to {destination} from {origin}. 
-    
-    Technical requirements:
-    - Implement using Tailwind CSS for responsive styling and utility classes
-    - Enhance with Alpine.js for lightweight interactive components 
-    - Include Mapbox GL JS integration for interactive maps with points of interest
-    - Use Chart.js for visualizing trip budget breakdowns
-    - Implement with DaisyUI or Flowbite components for consistent UI elements
-    - Ensure cross-browser compatibility and mobile-first responsive design
-    - Meet WCAG 2.1 AA accessibility standards
-    
-    UI/UX features:
-    - Create collapsible day-by-day itinerary sections
-    - Include interactive map with customized markers for attractions, hotels, and restaurants
-    - Implement dark/light mode toggle with user preference storage
-    - Add weather forecast integration via OpenWeatherMap API
-    - Design printable version with QR codes for digital access
-    - Include local emergency information in an accessible sidebar
-    - Create animated transitions between sections for improved user experience
-    - Implement offline capabilities with service workers""",
-    
-    agent=final_report_agent,
-    
-    expected_output="""Complete HTML travel itinerary featuring:
-    1. Responsive layout using Tailwind CSS that adapts to all device sizes
-    2. Interactive components powered by Alpine.js for seamless user interactions
-    3. Dynamic map visualization with Mapbox GL JS showing the complete travel route
-    4. Collapsible sections for daily activities with smooth animations
-    5. Dark/light mode toggle that persists user preferences
-    6. Weather forecast widgets integrated with OpenWeatherMap
-    7. Printable version with optimized styling for physical documents
-    8. Trip budget visualization with interactive charts
-    9. Offline-capable implementation with service worker caching
-    10. Accessibility-compliant design including keyboard navigation and screen reader support
-    11. Performance optimization with lazy-loading images and content
-    12. Custom PDF export functionality
-    
-    All code should be clean, well-commented, and follow modern web development best practices."""
-)
+
 
 # ------------------------------------------------------------------------------
 # Helper Function to Run a Task with Full Agent & Task Information
@@ -469,25 +418,10 @@ Dining Recommendations:
     )
     print("✓ Itinerary creation completed")
     
-    # Step 7: Generate Final Report
-    print("Generating your final travel itinerary document...")
-    final_context = f"""{combined_info}
-
-Day-by-Day Itinerary:
-{itinerary}
-"""
-    
-    final_itinerary = run_task(
-        final_report_task, 
-        final_context.format(
-            duration=user_input['duration'],
-            origin=user_input['origin'],
-            destination=user_input['destination']
-        )
-    )
-    print("✓ Final itinerary document generated")
+    # Return the completed itinerary
+    print("✓ Itinerary generation completed")
     
-    return final_itinerary
+    return itinerary
 
 # ------------------------------------------------------------------------------
 # Save Itinerary to File
@@ -495,7 +429,7 @@ Day-by-Day Itinerary:
 def save_itinerary_to_file(itinerary, user_input, output_dir=None):
     # Create a filename based on the destination and date
     date_str = datetime.now().strftime("%Y-%m-%d")
-    filename = f"{user_input['destination'].replace(' ', '_')}_{date_str}_itinerary.html"
+    filename = f"{user_input['destination'].replace(' ', '_')}_{date_str}_itinerary.txt"
     
     # If output directory is specified, ensure it exists and use it
     if output_dir: