Add application file

app.py

@@ -333,8 +333,9 @@ 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 to ensure exact number of days and proper formatting"""
+    """Format the travel plan with improved readability and structure"""
     formatted_plan = []
+    day_sections = []
     current_day = None
     current_activities = []
     
@@ -343,12 +344,15 @@ def format_travel_plan(plan, days):
         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)))
+                day_sections.append({
+                    'day': current_day,
+                    'activities': current_activities
+                })
             
             # Extract day number
             try:
@@ -369,21 +373,33 @@ def format_travel_plan(plan, 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)))
+        day_sections.append({
+            'day': current_day,
+            'activities': current_activities
+        })
     
     # Sort days and ensure we have exactly the requested number of days
-    formatted_plan.sort(key=lambda x: x[0])
-    final_plan = []
+    day_sections.sort(key=lambda x: x['day'])
     
-    # Generate the final formatted plan with exactly the requested number of days
+    # Generate the final formatted plan
     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")
+        day_content = next(
+            (section for section in day_sections if section['day'] == day),
+            {'day': day, 'activities': ["Explore local attractions and sights"]}
+        )
+        
+        # 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}")
+        
+        # Add spacing between days
+        formatted_plan.append("\n")
     
-    # Join all days with proper spacing
-    return '\n'.join(final_plan)
+    return "\n".join(formatted_plan)
 
 def main():
     st.set_page_config(
@@ -419,14 +435,14 @@ 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
+    # 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
     
     # Create two columns for input form
     col1, col2 = st.columns([2, 1])
@@ -513,47 +529,73 @@ def main():
                 st.session_state.tokenizer
             )
             
-            st.success("✨ Your travel plan is ready!")
+            # Create an expander for the success message with trip overview
+            with st.expander("✨ Your travel plan is ready! Click to see trip overview", expanded=True):
+                col1, col2, col3 = st.columns(3)
+                with col1:
+                    st.metric("Destination", destination)
+                with col2:
+                    st.metric("Duration", f"{days} days")
+                with col3:
+                    st.metric("Budget", budget)
+                
+                st.write("**Selected Interests:**", ", ".join(interests))
             
-            # Display the plan in tabs
-            plan_tab, summary_tab = st.tabs(["📋 Travel Plan", "ℹ️ Trip Summary"])
+            # Display the plan in tabs with improved styling
+            plan_tab, summary_tab = st.tabs(["📋 Detailed Itinerary", "ℹ️ Trip Summary"])
             
             with plan_tab:
-                st.markdown(travel_plan)
-                
-                # Add export options
-                st.download_button(
-                    label="📥 Download Plan",
-                    data=travel_plan,
-                    file_name=f"travel_plan_{destination.lower().replace(' ', '_')}.md",
-                    mime="text/markdown"
-                )
+                # Add a container for better spacing
+                with st.container():
+                    # Add trip title
+                    st.markdown(f"## 🌍 {days}-Day Trip to {destination}")
+                    st.markdown("---")
+                    
+                    # Display the formatted plan
+                    st.markdown(travel_plan)
+                    
+                    # Add export options in a nice container
+                    with st.container():
+                        st.markdown("---")
+                        col1, col2 = st.columns([1, 4])
+                        with col1:
+                            st.download_button(
+                                label="📥 Download Plan",
+                                data=travel_plan,
+                                file_name=f"travel_plan_{destination.lower().replace(' ', '_')}.md",
+                                mime="text/markdown",
+                                use_container_width=True
+                            )
             
             with summary_tab:
-                # Create three columns for summary information
-                sum_col1, sum_col2, sum_col3 = st.columns(3)
-                
-                with sum_col1:
-                    st.markdown("### 📍 Destination")
-                    st.markdown(f"**{destination}**")
-                    st.markdown("### ⏱️ Duration")
-                    st.markdown(f"**{days} days**")
-                
-                with sum_col2:
-                    st.markdown("### 💰 Budget")
-                    st.markdown(f"**{budget}**")
-                    st.markdown("### 🎯 Interests")
-                    for interest in interests:
-                        st.markdown(f"- {interest}")
-                
-                with sum_col3:
-                    st.markdown("### ⚠️ Important Notes")
-                    st.info(
-                        "- Verify opening hours\n"
-                        "- Check current prices\n"
-                        "- Confirm availability\n"
-                        "- Consider seasonal factors"
-                    )
+                # Create three columns for summary information with cards
+                with st.container():
+                    st.markdown("## Trip Overview")
+                    sum_col1, sum_col2, sum_col3 = st.columns(3)
+                    
+                    with sum_col1:
+                        with st.container():
+                            st.markdown("### 📍 Destination Details")
+                            st.markdown(f"**Location:** {destination}")
+                            st.markdown(f"**Duration:** {days} days")
+                            st.markdown(f"**Budget Level:** {budget}")
+                    
+                    with sum_col2:
+                        with st.container():
+                            st.markdown("### 🎯 Trip Focus")
+                            st.markdown("**Selected Interests:**")
+                            for interest in interests:
+                                st.markdown(f"- {interest}")
+                    
+                    with sum_col3:
+                        with st.container():
+                            st.markdown("### ⚠️ Travel Tips")
+                            st.info(
+                                "• Verify opening hours\n"
+                                "• Check current prices\n"
+                                "• Confirm availability\n"
+                                "• Consider seasonal factors"
+                            )
 
 if __name__ == "__main__":
     main()