Update application file

app.py

@@ -298,7 +298,7 @@ def generate_fallback_plan(destination, days, interests, budget):
     - budget: Trip budget category
     
     Returns:
-    A formatted travel plan string compatible with format_travel_plan function
+    A formatted travel plan string
     """
     # Basic activity templates
     basic_activities = {
@@ -314,21 +314,26 @@ def generate_fallback_plan(destination, days, interests, budget):
     }
     
     # Prepare the travel plan
-    travel_plan_lines = []
+    travel_plan_lines = [
+        f"# Travel Plan Details",
+        f"## Destination: {destination}",
+        f"## Duration: {days} days",
+        f"## Interests: {', '.join(interests)}",
+        f"## Budget: {budget}\n",
+        "## Itinerary:\n"
+    ]
     
     # Generate activities for each day
     for day in range(1, days + 1):
         day_lines = [f"Day {day}:"]
         
         # Select activities based on interests
-        day_activities = []
         available_interests = interests[:2]  # Use up to 2 interests per day
         
         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-specific note
         budget_text = {
@@ -343,68 +348,44 @@ def generate_fallback_plan(destination, days, interests, budget):
         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]
-    })
-    
-    # Use format_travel_plan to format the fallback plan
-    return format_travel_plan(df, destination, days, ', '.join(interests), budget)
+    return "\n".join(travel_plan_lines)
 
-def format_travel_plan(sample_data, destination, days, interests=None, budget=None):
+def format_travel_plan(travel_plan, destination, days, interests=None, budget=None):
     """
-    Format travel plan using data from create_sample_data()
+    Format travel plan string
     
     Parameters:
-    - sample_data: DataFrame from create_sample_data()
+    - travel_plan: Generated travel plan string
     - destination: Target destination
     - days: Number of trip days
-    - interests: Optional interests filter
-    - budget: Optional budget filter
+    - interests: Optional interests (default: None)
+    - budget: Optional budget (default: None)
     
     Returns:
     Formatted travel plan as a string
     """
-    # Filter the DataFrame based on input criteria
-    filtered_data = sample_data[
-        (sample_data['destination'] == destination) & 
-        (sample_data['days'] == days)
-    ]
-    
-    # Apply additional filters if provided
-    if interests:
-        filtered_data = filtered_data[filtered_data['interests'] == interests]
-    
-    if budget:
-        filtered_data = filtered_data[filtered_data['budget'] == budget]
+    # If the input is already a properly formatted string, return it
+    if isinstance(travel_plan, str):
+        return travel_plan
     
-    # Check if any matching data exists
-    if filtered_data.empty:
-        return f"No travel plans found for:\nDestination: {destination}\nDays: {days}"
-    
-    # Select the first matching travel plan
-    selected_plan = filtered_data.iloc[0]
-    
-    # Format the plan
-    formatted_plan = [
-        f"# Travel Plan Details\n",
-        f"## Destination: {selected_plan['destination']}\n",
-        f"## Duration: {selected_plan['days']} days\n",
-        f"## Interests: {selected_plan['interests']}\n",
-        f"## Budget: {selected_plan['budget']}\n\n",
-        "## Itinerary:\n"
-    ]
-    
-    # Add the travel plan directly from the CSV
-    formatted_plan.append(selected_plan['travel_plan'])
-    
-    return "\n".join(formatted_plan)
+    # If it's a DataFrame or Series, extract the travel plan
+    try:
+        # Attempt to get the travel plan from DataFrame or Series
+        if hasattr(travel_plan, 'iloc'):
+            # If it's a DataFrame or Series, get the first row's travel plan
+            travel_plan = travel_plan.iloc[0]['travel_plan']
+        elif hasattr(travel_plan, 'get'):
+            # If it's a dictionary-like object
+            travel_plan = travel_plan.get('travel_plan', '')
+    except Exception:
+        # Fallback to generating a new plan if extraction fails
+        return generate_fallback_plan(destination, days, interests or [], budget or 'Moderate')
+    
+    # Ensure travel_plan is a string
+    if not isinstance(travel_plan, str):
+        return generate_fallback_plan(destination, days, interests or [], budget or 'Moderate')
+    
+    return travel_plan
 
 
 def main():