Update application file

app.py

@@ -16,6 +16,7 @@ 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):
@@ -335,91 +336,65 @@ def generate_fallback_plan(destination, days, interests, budget):
     return format_travel_plan(fallback_plan, days)
 
 def format_travel_plan(plan, days):
-    """Format the travel plan with improved readability and structure"""
-    formatted_plan = []
-    day_sections = []
-    current_day = None
-    current_activities = []
+    """Format travel plan for 1-14 days with flexible activity distribution"""
+    # Validate days input
+    days = max(1, min(days, 14))
+    
+    # Initialize day activities dictionary
+    day_activities = {day: [] for day in range(1, days + 1)}
     
-    # Process the plan line by line
+    # Parse input plan
+    current_day = None
     for line in plan.split('\n'):
         line = line.strip()
         if not line:
             continue
-            
-        # Check for day headers
+        
+        # Detect day headers
         if line.lower().startswith('day'):
-            # Save previous day's activities if they exist
-            if current_day is not None and current_activities:
-                # Only add days that are within the requested range
-                if current_day <= days:
-                    day_sections.append({
-                        'day': current_day,
-                        'activities': current_activities
-                    })
-            
-            # Extract day number
             try:
-                day_part = line.split(':', 1)
-                day_num = int(''.join(filter(str.isdigit, day_part[0])))
-                # Only process days within the requested range
-                if day_num <= days:
+                day_num = int(''.join(filter(str.isdigit, line.split()[0])))
+                if 1 <= day_num <= days:
                     current_day = day_num
-                    current_activities = []
-                    
-                    # Process content after the day header if it exists
-                    if len(day_part) > 1:
-                        # Split activities by periods and filter out empty ones
-                        activities = [act.strip() for act in day_part[1].split('.') if act.strip()]
-                        current_activities.extend(activities)
-
             except ValueError:
+                current_day = None
                 continue
-        elif current_day is not None and current_day <= days:
-            # Only add activities for days within the requested range
-            # Split the line by periods to separate activities
-            activities = [act.strip() for act in line.split('.') if act.strip()]
-            current_activities.extend(activities)
-    
-    # Add the last day if there are pending activities and within range
-    if current_day is not None and current_activities and current_day <= days:
-        day_sections.append({
-            'day': current_day,
-            'activities': current_activities
-        })
+        
+        # Collect activities
+        elif current_day and current_day <= days:
+            # Split by multiple delimiters, filter meaningful activities
+            activities = [
+                act.strip() 
+                for act in re.split(r'[.;,]', line) 
+                if act.strip() and len(act.strip()) > 5
+            ]
+            
+            # Add activities for current day
+            for activity in activities:
+                if len(day_activities[current_day]) < 4:
+                    day_activities[current_day].append(activity)
     
-    # Sort days and ensure we have exactly the requested number of days
-    day_sections.sort(key=lambda x: x['day'])
+    # Ensure each day has activities
+    for day in range(1, days + 1):
+        if not day_activities[day]:
+            if day == 1:
+                day_activities[day].append("Explore city highlights")
+            else:
+                day_activities[day].append("Continue exploring local attractions")
     
-    # Generate the final formatted plan
+    # Generate formatted plan
+    formatted_plan = []
     for day in range(1, days + 1):
-        # Find the matching day's activities or use a placeholder
-        day_content = next(
-            (section for section in day_sections if section['day'] == day),
-            {'day': day, 'activities': []}
-        )
-        
-        # Format the day's content with markdown
         formatted_plan.append(f"### Day {day}\n")
         
-        # If no activities found for this day, add default activity
-        if not day_content['activities']:
-            formatted_plan.append("- Explore local attractions and sights")
-        else:
-            # Format each activity with bullet points
-            for activity in day_content['activities']:
-                # Clean up the activity text
-                activity = activity.strip()
-                if activity.lower().startswith('optional'):
-                    formatted_plan.append(f"- *{activity}*")
-                else:
-                    formatted_plan.append(f"- {activity}")
-        
-        # Add spacing between days
+        for activity in day_activities[day]:
+            formatted_plan.append(f"- {activity}")
+        
         formatted_plan.append("\n")
     
     return "\n".join(formatted_plan)
 
+
 def main():
     st.set_page_config(
         page_title="AI Travel Planner",

requirements.txt

@@ -6,4 +6,5 @@ accelerate
 sentencepiece
 protobuf
 typing-extensions
-packaging
+packaging
+re