Add application file

app.py

@@ -262,8 +262,7 @@ def generate_travel_plan(destination, days, interests, budget, model, tokenizer)
             top_k=50,
             top_p=0.9,
             do_sample=True,
-            repetition_penalty=1.2,  # Additional repetition avoidance
-            num_return_sequences=1
+            repetition_penalty=1.2  # Additional repetition avoidance
         )
         
         # Decode output
@@ -272,76 +271,11 @@ def generate_travel_plan(destination, days, interests, budget, model, tokenizer)
         # Handle empty output
         if not travel_plan.strip():
             raise ValueError("Generated plan is empty")
-        
-        # Ensure the plan has the correct number of days
-        plan_lines = travel_plan.split('\n')
-        formatted_lines = []
-        activity_templates = {
-            'Budget': [
-                "Explore local free attractions",
-                "Visit public parks and gardens",
-                "Take self-guided walking tours",
-                "Visit markets and street food venues",
-                "Use public transportation"
-            ],
-            'Moderate': [
-                "Join group tours and activities",
-                "Visit popular attractions",
-                "Try local restaurants",
-                "Use mix of public and private transport",
-                "Book mid-range accommodations"
-            ],
-            'Luxury': [
-                "Book private guided tours",
-                "Experience fine dining",
-                "Visit exclusive attractions",
-                "Use private transportation",
-                "Stay at luxury accommodations"
-            ]
-        }
-        
-        # Ensure we have enough content for each day
-        for day in range(1, days + 1):
-            day_content = next(
-                (line for line in plan_lines if f"Day {day}:" in line),
-                None
-            )
             
-            if day_content:
-                formatted_lines.append(day_content)
-            else:
-                # Generate fallback content based on budget and interests
-                budget_activities = activity_templates[budget]
-                interests_activities = [
-                    f"Explore {destination}'s {interest.lower()} attractions"
-                    for interest in interests
-                ]
-                activities = budget_activities + interests_activities
-                
-                fallback_content = (
-                    f"Day {day}: {random.choice(activities)}. "
-                    f"Also {random.choice(activities).lower()}."
-                )
-                formatted_lines.append(fallback_content)
-        
-        # Join the lines back together
-        final_plan = '\n'.join(formatted_lines)
-        
-        # Add a trip overview at the beginning
-        overview = (
-            f"Trip Overview:\n"
-            f"Destination: {destination}\n"
-            f"Duration: {days} days\n"
-            f"Budget Level: {budget}\n"
-            f"Interests: {interests_str}\n\n"
-        )
-        
-        final_plan = overview + final_plan
-        
-        # Log successful generation
-        print(f"Successfully generated plan for {destination} ({days} days)")
+        # Format the plan using the new formatting function
+        formatted_plan = format_travel_plan(travel_plan, days)
         
-        return final_plan
+        return formatted_plan
         
     except Exception as e:
         error_msg = f"Error generating travel plan: {str(e)}"
@@ -353,6 +287,7 @@ def generate_travel_plan(destination, days, interests, budget, model, tokenizer)
     
 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"
     
     # Basic activity templates
@@ -368,51 +303,87 @@ 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
     for day in range(1, days + 1):
-        fallback_plan += f"\n## Day {day}\n"
+        fallback_plan += f"\nDay {day}:\n"
+        
         # Select activities based on interests
         day_activities = []
-        for interest in interests[:2]:  # Use up to 2 interests per day
+        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_activities.append(activity)
         
         # Add budget-appropriate text
         budget_text = {
-            'Budget': 'Focus on free and affordable activities.',
-            'Moderate': 'Mix of affordable and premium experiences.',
-            'Luxury': 'Premium experiences and exclusive access.'
+            'Budget': 'Focus on free and affordable activities',
+            'Moderate': 'Mix of affordable and premium experiences',
+            'Luxury': 'Premium experiences and exclusive access'
         }.get(budget, '')
         
+        # 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"
     
-    return fallback_plan
+    # Format the fallback plan using the same formatter
+    return format_travel_plan(fallback_plan, days)
 
 def format_travel_plan(plan, days):
-    """Format the generated travel plan into a readable structure"""
-    formatted_plan = "# Your Travel Itinerary\n\n"
+    """Format the travel plan to ensure exact number of days and proper formatting"""
+    formatted_plan = []
+    current_day = None
+    current_activities = []
     
-    # Split the plan into days (split by newlines)
-    day_plans = plan.split('\n')
+    # Process the plan line by line
+    for line in plan.split('\n'):
+        line = line.strip()
+        if not line:
+            continue
+        
+        # Check for day headers
+        if line.lower().startswith('day'):
+            # Save previous day's activities if they exist
+            if current_day is not None and current_activities:
+                formatted_plan.append((current_day, '\n'.join(current_activities)))
+            
+            # Extract day number
+            try:
+                day_num = int(''.join(filter(str.isdigit, line.split(':')[0])))
+                current_day = day_num
+                current_activities = []
+                # Add any activities that appear on the same line after the colon
+                if ':' in line:
+                    activity = line.split(':', 1)[1].strip()
+                    if activity:
+                        current_activities.append(activity)
+            except ValueError:
+                continue
+        else:
+            # Add activity to current day
+            if current_day is not None:
+                current_activities.append(line)
     
-    # Filter out empty lines and ensure we don't exceed the requested number of days
-    day_plans = [plan.strip() for plan in day_plans if plan.strip()][:days]
+    # Add the last day if there are pending activities
+    if current_day is not None and current_activities:
+        formatted_plan.append((current_day, '\n'.join(current_activities)))
     
-    # Format each day
-    for day_plan in day_plans:
-        if day_plan.startswith("Day"):
-            # Extract day number
-            day_num = day_plan.split(':')[0].replace('Day ', '')
-            # Extract activities
-            activities = day_plan.split(':', 1)[1].strip()
-            
-            formatted_plan += f"\n## Day {day_num}\n"
-            formatted_plan += f"{activities}\n"
+    # Sort days and ensure we have exactly the requested number of days
+    formatted_plan.sort(key=lambda x: x[0])
+    final_plan = []
+    
+    # Generate the final formatted plan with exactly the requested number of days
+    for day in range(1, days + 1):
+        # Find the matching day's activities or use a placeholder
+        day_content = next((content for d, content in formatted_plan if d == day), 
+                         "Explore local attractions and sights")
+        final_plan.append(f"Day {day}:\n{day_content}\n")
     
-    return formatted_plan
+    # Join all days with proper spacing
+    return '\n'.join(final_plan)
 
 def main():
     st.set_page_config(