Update app.py

app.py

@@ -853,6 +853,7 @@ if not st.session_state.generation_complete:
                 "preferences": preferences,
                 "special_requirements": special_requirements
             }
+            # 기존의 여행 요청 프롬프트
             input_context = f"""Travel Request Details:
 Origin: {user_input['origin']}
 Destination: {user_input['destination']}
@@ -864,6 +865,21 @@ Travel Style: {user_input['travel_style']}
 Preferences/Interests: {user_input['preferences']}
 Special Requirements: {user_input['special_requirements']}
 """
+            # LLM에 전달할 프롬프트에 언어 지시문 추가
+            llm_language_instructions = {
+                "en": "Please output the response in English.",
+                "ko": "한국어로 출력해 주세요.",
+                "ja": "日本語で出力してください。",
+                "zh": "请用中文输出。",
+                "es": "Por favor, responda en español.",
+                "fr": "Veuillez répondre en français.",
+                "de": "Bitte antworten Sie auf Deutsch.",
+                "ar": "يرجى الرد باللغة العربية."
+            }
+            selected_lang = st.session_state.get("selected_language", "en")
+            language_instruction = llm_language_instructions.get(selected_lang, "Please output the response in English.")
+            modified_input_context = language_instruction + "\n" + input_context
+            
             st.markdown("""
             <div class="sleek-processing-container">
               <div class="pulse-container">
@@ -948,7 +964,7 @@ Special Requirements: {user_input['special_requirements']}
                 display_modern_progress(st.session_state.current_step)
             destination_info = run_task_with_logs(
                 destination_research_task, 
-                input_context.format(destination=user_input['destination'], preferences=user_input['preferences']),
+                modified_input_context.format(destination=user_input['destination'], preferences=user_input['preferences']),
                 log_container,
                 output_container,
                 "destination_info"
@@ -960,7 +976,7 @@ Special Requirements: {user_input['special_requirements']}
                 display_modern_progress(st.session_state.current_step)
             accommodation_info = run_task_with_logs(
                 accommodation_task, 
-                input_context.format(destination=user_input['destination'], budget=user_input['budget'], preferences=user_input['preferences']),
+                modified_input_context.format(destination=user_input['destination'], budget=user_input['budget'], preferences=user_input['preferences']),
                 log_container,
                 output_container,
                 "accommodation_info"
@@ -972,7 +988,7 @@ Special Requirements: {user_input['special_requirements']}
                 display_modern_progress(st.session_state.current_step)
             transportation_info = run_task_with_logs(
                 transportation_task, 
-                input_context.format(origin=user_input['origin'], destination=user_input['destination']),
+                modified_input_context.format(origin=user_input['origin'], destination=user_input['destination']),
                 log_container,
                 output_container,
                 "transportation_info"
@@ -984,7 +1000,7 @@ Special Requirements: {user_input['special_requirements']}
                 display_modern_progress(st.session_state.current_step)
             activities_info = run_task_with_logs(
                 activities_task, 
-                input_context.format(destination=user_input['destination'], preferences=user_input['preferences']),
+                modified_input_context.format(destination=user_input['destination'], preferences=user_input['preferences']),
                 log_container,
                 output_container,
                 "activities_info"
@@ -996,7 +1012,7 @@ Special Requirements: {user_input['special_requirements']}
                 display_modern_progress(st.session_state.current_step)
             dining_info = run_task_with_logs(
                 dining_task, 
-                input_context.format(destination=user_input['destination'], preferences=user_input['preferences']),
+                modified_input_context.format(destination=user_input['destination'], preferences=user_input['preferences']),
                 log_container,
                 output_container,
                 "dining_info"
@@ -1023,6 +1039,7 @@ Recommended Activities:
 Dining Recommendations:
 {dining_info}
 """
+            # 여기서는 language instruction은 이미 포함되지 않은 원본 input_context을 사용(필요에 따라 수정 가능)
             itinerary = run_task_with_logs(
                 itinerary_task, 
                 combined_info.format(duration=user_input['duration'], origin=user_input['origin'], destination=user_input['destination']),