Update app.py

app.py

@@ -4,13 +4,20 @@ import json
 from datetime import datetime, timedelta
 import base64
 import pandas as pd
-
 from travel import (
     destination_research_task, accommodation_task, transportation_task,
     activities_task, dining_task, itinerary_task,
     run_task
 )
 
+# st.set_page_config()는 다른 Streamlit 함수 호출보다 가장 먼저 실행되어야 합니다.
+st.set_page_config(
+    page_title="Globetrotter AI: Your AI Agent for Travelling",  # 기본값은 영어
+    page_icon="✈️",
+    layout="wide",
+    initial_sidebar_state="expanded"
+)
+
 # ------------------------------------------
 # 다국어 지원을 위한 번역 사전 및 헬퍼 함수
 # ------------------------------------------
@@ -273,7 +280,6 @@ translations = {
     }
 }
 
-# 번역 헬퍼 함수: 선택된 언어에 따라 키 값을 반환합니다.
 def t(key):
     lang = st.session_state.get("selected_language", "en")
     return translations[lang].get(key, key)
@@ -305,17 +311,9 @@ with st.sidebar:
     st.session_state.selected_language = lang_map.get(language, "en")
 
 # ------------------------------------------
-# 페이지 설정 (번역 적용)
+# 이후 Streamlit UI 코드 시작
 # ------------------------------------------
 
-
-st.set_page_config(
-    page_title="Globetrotter AI: Your AI Agent for Travelling",
-    page_icon="✈️",
-    layout="wide",
-    initial_sidebar_state="expanded"
-)
-
 # Modern CSS with refined color scheme and sleek animations
 st.markdown("""
 <style>
@@ -539,10 +537,10 @@ def display_modern_progress(current_step, total_steps=6):
     if 'progress_steps' not in st.session_state:
         st.session_state.progress_steps = {
             0: {'status': 'pending', 'name': t("trip_details")},
-            1: {'status': 'pending', 'name': t("about")},  # Accommodation
-            2: {'status': 'pending', 'name': t("travel_style")},  # Transportation
-            3: {'status': 'pending', 'name': t("live_agent_outputs")},  # Activities
-            4: {'status': 'pending', 'name': t("download_share")},  # Dining
+            1: {'status': 'pending', 'name': t("about")},  # 예: 숙소 관련
+            2: {'status': 'pending', 'name': t("travel_style")},  # 예: 교통편 관련
+            3: {'status': 'pending', 'name': t("live_agent_outputs")},  # 예: 액티비티 관련
+            4: {'status': 'pending', 'name': t("download_share")},  # 예: 식사 관련
             5: {'status': 'pending', 'name': t("full_itinerary")}
         }
     
@@ -689,7 +687,9 @@ def run_task_with_logs(task, input_text, log_container, output_container, result
     
     return result
 
-# Initialize session state
+# ------------------------------------------
+# Session state 초기화
+# ------------------------------------------
 if 'generated_itinerary' not in st.session_state:
     st.session_state.generated_itinerary = None
 if 'generation_complete' not in st.session_state:
@@ -781,6 +781,7 @@ with st.sidebar:
         st.markdown(f"**{name}**")
         st.markdown(f"<small>{desc}</small>", unsafe_allow_html=True)
     st.markdown('</div>', unsafe_allow_html=True)
+
 # Main content area
 if not st.session_state.generation_complete:
     # Sleek form with minimal design
@@ -821,13 +822,13 @@ if not st.session_state.generation_complete:
                                                 "Nature", "Shopping", "Nightlife", "Family-friendly"],
                                        default=["Culture", "Food & Dining"])
         
-        # Simplified expander
+        # Simplified expander for additional preferences
         with st.expander("Additional Preferences", expanded=False):
             preferences = st.text_area("Interests", placeholder="History museums, local cuisine, hiking, art...")
             special_requirements = st.text_area("Special Requirements", 
                                             placeholder="Dietary restrictions, accessibility needs...")
         
-        # Submit button with enhanced styling
+        # Submit button
         submit_button = st.form_submit_button(t("submit"))
     
     st.markdown('</div>', unsafe_allow_html=True)
@@ -850,7 +851,7 @@ if not st.session_state.generation_complete:
                 "special_requirements": special_requirements
             }
             
-            # Format the user input for tasks with enhanced details
+            # Format the user input for tasks
             input_context = f"""Travel Request Details:
 Origin: {user_input['origin']}
 Destination: {user_input['destination']}
@@ -863,7 +864,7 @@ Preferences/Interests: {user_input['preferences']}
 Special Requirements: {user_input['special_requirements']}
 """
             
-            # Display a minimal, sleek processing animation
+            # Display processing animation
             st.markdown("""
             <div class="sleek-processing-container">
               <div class="pulse-container">
@@ -922,13 +923,10 @@ Special Requirements: {user_input['special_requirements']}
             </style>
             """, unsafe_allow_html=True)
             
-            # Create modern containers for progress, logs, and live output
+            # Create containers for progress, logs, and live output
             st.markdown('<div class="modern-card">', unsafe_allow_html=True)
-            
-            # Create tabs for better organization
             progress_tab, logs_tab, details_tab = st.tabs(["📊 Progress", "🔄 Live Activity", "📋 Request Details"])
             
-            # Show request details in the details tab
             with details_tab:
                 st.markdown("#### " + t("request_details"))
                 st.markdown(f"**{t('destination')}:** {user_input['destination']}")
@@ -942,7 +940,6 @@ Special Requirements: {user_input['special_requirements']}
                     st.markdown(f"**Special Requirements:** {user_input['special_requirements']}")
             
             with progress_tab:
-                # Create a persistent placeholder for progress display to avoid duplication
                 if 'progress_placeholder' not in st.session_state:
                     st.session_state.progress_placeholder = st.empty()
                 with st.session_state.progress_placeholder.container():
@@ -952,8 +949,7 @@ Special Requirements: {user_input['special_requirements']}
                 log_container = st.container()
                 st.session_state.log_messages = []
             
-            # Create a container for output that spans the full width
-            st.markdown('</div>', unsafe_allow_html=True) # Close the progress card
+            st.markdown('</div>', unsafe_allow_html=True)
             output_container = st.container()
             with output_container:
                 st.markdown('<div class="modern-card">', unsafe_allow_html=True)
@@ -961,13 +957,10 @@ Special Requirements: {user_input['special_requirements']}
                 st.info("Our AI agents will show their work here as they create your itinerary")
                 st.markdown('</div>', unsafe_allow_html=True)
             
-            # Initialize progress tracking
             st.session_state.current_step = 0
             
-            # Display initial progress - using the session state placeholder
-            
             # Step 1: Destination Research
-            update_step_status(0, 'active')  # Mark step 1 as active
+            update_step_status(0, 'active')
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
             
@@ -982,10 +975,9 @@ Special Requirements: {user_input['special_requirements']}
                 "destination_info"
             )
             
-            # Update progress after step 1 completes
-            update_step_status(0, 'complete')  # Mark step 1 as complete
+            update_step_status(0, 'complete')
             st.session_state.current_step = 1
-            update_step_status(1, 'active')  # Mark step 2 as active
+            update_step_status(1, 'active')
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
             
@@ -1002,10 +994,9 @@ Special Requirements: {user_input['special_requirements']}
                 "accommodation_info"
             )
             
-            # Update progress after step 2 completes
-            update_step_status(1, 'complete')  # Mark step 2 as complete
+            update_step_status(1, 'complete')
             st.session_state.current_step = 2
-            update_step_status(2, 'active')  # Mark step 3 as active
+            update_step_status(2, 'active')
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
             
@@ -1021,10 +1012,9 @@ Special Requirements: {user_input['special_requirements']}
                 "transportation_info"
             )
             
-            # Update progress after step 3 completes
-            update_step_status(2, 'complete')  # Mark step 3 as complete
+            update_step_status(2, 'complete')
             st.session_state.current_step = 3
-            update_step_status(3, 'active')  # Mark step 4 as active
+            update_step_status(3, 'active')
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
             
@@ -1040,10 +1030,9 @@ Special Requirements: {user_input['special_requirements']}
                 "activities_info"
             )
             
-            # Update progress after step 4 completes
-            update_step_status(3, 'complete')  # Mark step 4 as complete
+            update_step_status(3, 'complete')
             st.session_state.current_step = 4
-            update_step_status(4, 'active')  # Mark step 5 as active
+            update_step_status(4, 'active')
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
             
@@ -1059,10 +1048,9 @@ Special Requirements: {user_input['special_requirements']}
                 "dining_info"
             )
             
-            # Update progress after step 5 completes
-            update_step_status(4, 'complete')  # Mark step 5 as complete
+            update_step_status(4, 'complete')
             st.session_state.current_step = 5
-            update_step_status(5, 'active')  # Mark step 6 as active
+            update_step_status(5, 'active')
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
             
@@ -1097,25 +1085,19 @@ Dining Recommendations:
                 "itinerary"
             )
             
-            # Update final step status to complete
-            update_step_status(5, 'complete')  # Mark final step as complete
+            update_step_status(5, 'complete')
             st.session_state.current_step = 6
             with st.session_state.progress_placeholder.container():
                 display_modern_progress(st.session_state.current_step)
                 
-            # Save the generated itinerary to session state
             st.session_state.generated_itinerary = itinerary
             st.session_state.generation_complete = True
             
-            # Create a filename based on the destination and date
             date_str = datetime.now().strftime("%Y-%m-%d")
             st.session_state.filename = f"{user_input['destination'].replace(' ', '_')}_{date_str}_itinerary.txt"
-            
-            # No need to rerun, we'll update the UI directly
 
 # Display results if generation is complete
 if st.session_state.generation_complete:
-    # Success animation and header
     st.markdown("""
     <div class="modern-card animate-in">
       <div style="display: flex; justify-content: center; margin-bottom: 20px;">
@@ -1183,18 +1165,13 @@ if st.session_state.generation_complete:
     </style>
     """, unsafe_allow_html=True)
     
-    # Main container for itinerary content
     st.markdown('<div class="modern-card">', unsafe_allow_html=True)
-    
-    # Modern tabs for different views
     itinerary_tab, details_tab, download_tab = st.tabs([f"🗒️ {t('full_itinerary')}", f"💼 {t('details')}", f"💾 {t('download_share')}"])
     
     with itinerary_tab:
-        # Preview the itinerary as text
         st.text_area("Your Itinerary", st.session_state.generated_itinerary, height=600)
     
     with details_tab:
-        # More detailed view with agent outputs in a nested tab structure
         agent_tabs = st.tabs([
             "🌎 Destination", "🏨 Accommodation", "🚗 Transportation", 
             "🎭 Activities", "🍽️ Dining"
@@ -1226,36 +1203,26 @@ if st.session_state.generation_complete:
         with col1:
             st.markdown("### " + t("save_itinerary"))
             st.markdown("Download your personalized travel plan to access it offline or share with your travel companions.")
-            
-            # Display stylized download button
             st.markdown("""
             <div style="background-color: #f8f9fa; padding: 15px; border-radius: 10px; margin-top: 20px;">
                 <h4 style="margin-top: 0;">Your Itinerary File</h4>
                 <p style="font-size: 0.9rem; color: #6c757d;">Text format - Can be opened in any text editor</p>
             """, unsafe_allow_html=True)
-            
-            # Enhanced download link
             st.markdown(
                 f"""<div style="margin: 10px 0;">{ get_download_link(st.session_state.generated_itinerary, st.session_state.filename) }</div>""",
                 unsafe_allow_html=True
             )
-            
             st.markdown("</div>", unsafe_allow_html=True)
-            
-            # Share options
             st.markdown("### " + t("share_itinerary"))
             st.markdown("*Coming soon: Email your itinerary or share via social media.*")
         
         with col2:
-            # QR code placeholder for future implementation
             st.markdown("### " + t("save_for_mobile"))
             st.markdown("*Coming soon: QR code for easy access on your phone*")
     
     st.markdown('</div>', unsafe_allow_html=True)
     
-    # Add a reset button to create a new itinerary
     if st.button(t("plan_another_trip"), key="reset_button"):
-        # Reset the session state
         st.session_state.generated_itinerary = None
         st.session_state.generation_complete = False
         st.session_state.current_step = 0
@@ -1263,16 +1230,12 @@ if st.session_state.generation_complete:
         st.session_state.results = {}
         st.experimental_rerun()
     
-    # Footer for the app
     st.markdown(f"""
     <div style="margin-top: 50px; text-align: center; padding: 20px; color: #6c757d; font-size: 0.8rem;">
         <p>{t("built_with")}</p>
     </div>
     """, unsafe_allow_html=True)
-    
-    # End of app
 
-# Modern footer with more information
 st.markdown("""
 <div style="margin-top: 50px; border-top: 1px solid #e9ecef; padding: 30px 0; color: #6c757d;">
     <div style="display: flex; justify-content: space-between; flex-wrap: wrap; max-width: 1200px; margin: 0 auto; padding: 0 20px;">
@@ -1293,7 +1256,6 @@ st.markdown("""
             <h4 style="color: #4361ee; margin-bottom: 15px;">Connect</h4>
             <p>Stay updated with our latest travel guides and features.</p>
             <div style="margin-top: 15px;">
-                <!-- Social media icons would go here -->
                 <span>🐰 🐿 🐸 🦉</span>
             </div>
         </div>