Update application file

app.py

@@ -16,7 +16,6 @@ from torch.utils.data import Dataset
 from datetime import datetime
 import numpy as np
 from random import choice
-import re
 
 class TravelDataset(Dataset):
     def __init__(self, data, tokenizer, max_length=512):
@@ -289,10 +288,18 @@ def generate_travel_plan(destination, days, interests, budget, model, tokenizer)
         return fallback_plan
     
 def generate_fallback_plan(destination, days, interests, budget):
-    """Generate a basic fallback plan if the model fails"""
-    # Start with the overview section
-    fallback_plan = f"# Emergency Travel Plan for {destination}\n\n"
+    """
+    Generate a basic fallback travel plan if the model fails
+    
+    Parameters:
+    - destination: Target travel destination
+    - days: Number of days for the trip
+    - interests: Primary interests for the trip
+    - budget: Trip budget category
     
+    Returns:
+    A formatted travel plan string compatible with format_travel_plan function
+    """
     # Basic activity templates
     basic_activities = {
         'Culture': ['Visit museums', 'Explore historical sites', 'Attend local events'],
@@ -306,9 +313,12 @@ def generate_fallback_plan(destination, days, interests, budget):
         'Museums': ['Visit main museums', 'Join guided tours', 'See special exhibits']
     }
     
-    # Generate exactly the requested number of days
+    # Prepare the travel plan
+    travel_plan_lines = []
+    
+    # Generate activities for each day
     for day in range(1, days + 1):
-        fallback_plan += f"\nDay {day}:\n"
+        day_lines = [f"Day {day}:"]
         
         # Select activities based on interests
         day_activities = []
@@ -317,23 +327,35 @@ def generate_fallback_plan(destination, days, interests, budget):
         for interest in available_interests:
             if interest in basic_activities:
                 activity = random.choice(basic_activities[interest])
+                day_lines.append(f"- {activity}")
                 day_activities.append(activity)
         
-        # Add budget-appropriate text
+        # Add budget-specific note
         budget_text = {
             'Budget': 'Focus on free and affordable activities',
             'Moderate': 'Mix of affordable and premium experiences',
             'Luxury': 'Premium experiences and exclusive access'
-        }.get(budget, '')
+        }.get(budget, 'Balanced travel experiences')
+        
+        day_lines.append(f"Note: {budget_text}")
         
-        # Format the day's activities
-        fallback_plan += f"Morning: {day_activities[0] if day_activities else 'Explore the area'}\n"
-        if len(day_activities) > 1:
-            fallback_plan += f"Afternoon/Evening: {day_activities[1]}\n"
-        fallback_plan += f"Note: {budget_text}\n"
+        travel_plan_lines.extend(day_lines)
+        travel_plan_lines.append("")  # Add blank line between days
+    
+    # Create a travel plan string
+    travel_plan_str = "\n".join(travel_plan_lines)
+    
+    # Create a temporary DataFrame to simulate the CSV data structure
+    df = pd.DataFrame({
+        'destination': [destination],
+        'days': [days],
+        'interests': [', '.join(interests)],
+        'budget': [budget],
+        'travel_plan': [travel_plan_str]
+    })
     
-    # Format the fallback plan using the same formatter
-    return format_travel_plan(fallback_plan, destination, days)
+    # Use format_travel_plan to format the fallback plan
+    return format_travel_plan(df, destination, days, ', '.join(interests), budget)
 
 def format_travel_plan(sample_data, destination, days, interests=None, budget=None):
     """