Update app.py

app.py

@@ -769,8 +769,12 @@ def region_ui(travel_df, external_score_df, festival_df, weather_df, package_df,
             weather_df,
             chat_container=chat_container
         )
-        st.session_state[step_key] = "companion"
-        st.rerun()        
+        if st.session_state.get("package_rendered", False):
+            st.session_state[step_key] = "package_end"
+        else:
+            st.session_state[step_key] = "companion"
+        
+        st.rerun()
         return
 
 
@@ -1135,8 +1139,12 @@ def intent_ui(travel_df, external_score_df, festival_df, weather_df, package_df,
             weather_df,
             chat_container=chat_container
         )
-        st.session_state[step_key] = "companion"
-        st.rerun()        
+        if st.session_state.get("package_rendered", False):
+            st.session_state[step_key] = "package_end"
+        else:
+            st.session_state[step_key] = "companion"
+        
+        st.rerun()
         return
     
     # ────────────────── 5) 동행·연령 받기 단계
@@ -1571,9 +1579,13 @@ def emotion_ui(travel_df, external_score_df, festival_df, weather_df, package_df
             weather_df,
             chat_container=chat_container
         )
-        st.session_state[step_key] = "companion"
-        st.rerun()        
-        return   
+        if st.session_state.get("package_rendered", False):
+            st.session_state[step_key] = "package_end"
+        else:
+            st.session_state[step_key] = "companion"
+        
+        st.rerun()
+        return 
       
     # ────────────────── 5) 동행·연령 받기 단계
     elif st.session_state[step_key] == "companion":