Update application file

app.py

@@ -77,23 +77,23 @@ def create_sample_data():
     # Activity templates for different interests
     activities = {
         'Culture': ['Visit historical sites', 'Explore local traditions', 'Attend cultural events', 
-                   'Visit ancient monuments', 'Experience local festivals'],
+                'Visit ancient monuments', 'Experience local festivals'],
         'History': ['Tour ancient ruins', 'Visit museums', 'Explore historic districts', 
-                   'Join guided history walks', 'Visit heritage sites'],
+                'Join guided history walks', 'Visit heritage sites'],
         'Food': ['Try local cuisine', 'Join cooking classes', 'Visit food markets', 
-                 'Dine at famous restaurants', 'Food tasting tours'],
+                'Dine at famous restaurants', 'Food tasting tours'],
         'Shopping': ['Browse local markets', 'Visit shopping districts', 'Shop at boutiques', 
-                    'Explore artisan shops', 'Visit shopping centers'],
+                'Explore artisan shops', 'Visit shopping centers'],
         'Art': ['Visit art galleries', 'Attend art exhibitions', 'Join art workshops', 
                 'Visit artist studios', 'Explore street art'],
         'Museums': ['Tour famous museums', 'Visit specialty museums', 'Join museum tours', 
-                   'Explore art collections', 'Visit cultural institutes'],
+                'Explore art collections', 'Visit cultural institutes'],
         'Nature': ['Visit parks', 'Nature walks', 'Explore gardens', 'Visit natural landmarks', 
-                  'Outdoor activities'],
+                'Outdoor activities'],
         'Adventure': ['Join adventure tours', 'Try outdoor sports', 'Explore hidden spots', 
-                     'Take scenic hikes', 'Adventure activities'],
+                'Take scenic hikes', 'Adventure activities'],
         'Relaxation': ['Spa treatments', 'Visit peaceful gardens', 'Leisure activities', 
-                      'Relaxing sightseeing', 'Peaceful excursions']
+                'Relaxing sightseeing', 'Peaceful excursions']
     }
     
     def generate_daily_plan(day, total_days, interests, budget_level, destination):
@@ -140,21 +140,23 @@ def create_sample_data():
 
 @st.cache_resource
 def load_or_train_model():
-     """Load trained model or train new one"""
-     model_path = "./trained_travel_planner"
-    
-     if os.path.exists(model_path):
-         try:
-             model = T5ForConditionalGeneration.from_pretrained(model_path)
-             tokenizer = T5Tokenizer.from_pretrained(model_path)
-             if torch.cuda.is_available():
-                 model = model.cuda()
-             return model, tokenizer
-         except Exception as e:
-             st.error(f"Error loading trained model: {str(e)}")
-    
-     # If no trained model exists or loading fails, train new model
-     return train_model()
+    """Load trained model or train new one"""
+    model_path = "trained_travel_planner"
+
+    if os.path.exists(model_path):
+        try:
+            model = T5ForConditionalGeneration.from_pretrained(model_path)
+            tokenizer = T5Tokenizer.from_pretrained(model_path)
+            if torch.cuda.is_available():
+                model = model.cuda()
+            st.success("✓ Loaded existing model")
+            return model, tokenizer
+        except Exception as e:
+            st.warning("Could not load existing model, will train new one")
+            st.error(f"Error loading trained model: {str(e)}")
+
+    # If no trained model exists or loading fails, train new model
+    return train_model()
 
 def train_model():
     """Train the T5 model on travel planning data"""
@@ -356,20 +358,23 @@ def format_travel_plan(plan, days):
             
             # Extract day number
             try:
-                day_num = int(''.join(filter(str.isdigit, line.split(':')[0])))
+                day_part = line.split(':', 1)
+                day_num = int(''.join(filter(str.isdigit, day_part[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)
+                
+                # 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
-        else:
-            # Add activity to current day
-            if current_day is not None:
-                current_activities.append(line)
+        elif current_day is not None:
+            # 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:
@@ -386,15 +391,24 @@ def format_travel_plan(plan, days):
         # 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': ["Explore local attractions and sights"]}
+            {'day': day, 'activities': []}
         )
         
         # Format the day's content with markdown
         formatted_plan.append(f"### Day {day}\n")
         
-        # Format activities with bullet points
-        for activity in day_content['activities']:
-            formatted_plan.append(f"- {activity}")
+        # 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
         formatted_plan.append("\n")
@@ -436,13 +450,13 @@ def main():
     
         # Load or train model
         if 'model' not in st.session_state:
-         with st.spinner("Loading AI model... Please wait..."):
-             model, tokenizer = load_or_train_model()
-             if model is None or tokenizer is None:
-                 st.error("Failed to load/train the AI model. Please try again.")
-                 return
-             st.session_state.model = model
-             st.session_state.tokenizer = tokenizer
+            with st.spinner("Loading AI model... Please wait..."):
+                model, tokenizer = load_or_train_model()
+                if model is None or tokenizer is None:
+                    st.error("Failed to load/train the AI model. Please try again.")
+                    return
+                st.session_state.model = model
+                st.session_state.tokenizer = tokenizer
     
     # Create two columns for input form
     col1, col2 = st.columns([2, 1])
@@ -483,7 +497,7 @@ def main():
                 interests = st.multiselect(
                     "🎯 Interests",
                     ["Culture", "History", "Food", "Nature", "Shopping", 
-                     "Adventure", "Relaxation", "Art", "Museums"],
+                    "Adventure", "Relaxation", "Art", "Museums"],
                     ["Culture", "Food"],
                     help="Select up to three interests to personalize your plan"
                 )