Update application file

app.py

@@ -351,33 +351,38 @@ def format_travel_plan(plan, days):
         if line.lower().startswith('day'):
             # Save previous day's activities if they exist
             if current_day is not None and current_activities:
-                day_sections.append({
-                    'day': current_day,
-                    'activities': 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])))
-                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)
+                # Only process days within the requested range
+                if 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:
                 continue
-        elif current_day is not None:
+        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
-    if current_day is not None and current_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
@@ -549,7 +554,10 @@ def main():
                 with col1:
                     st.metric("Destination", destination)
                 with col2:
-                    st.metric("Duration", f"{days} days")
+                    if days == 1:
+                        st.metric("Duration", f"{days} day")
+                    else:
+                        st.metric("Duration", f"{days} days")
                 with col3:
                     st.metric("Budget", budget)
                 
@@ -591,7 +599,10 @@ def main():
                         with st.container():
                             st.markdown("### 📍 Destination Details")
                             st.markdown(f"**Location:** {destination}")
-                            st.markdown(f"**Duration:** {days} days")
+                            if days == 1:
+                                st.markdown(f"**Duration:** {days} day")
+                            else: 
+                                st.markdown(f"**Duration:** {days} days")
                             st.markdown(f"**Budget Level:** {budget}")
                     
                     with sum_col2: