Upload main_app.py

main_app.py

@@ -510,7 +510,6 @@ def initialize_session_state(managers, force_reset_override=False):
         
         # 重要: 初期メッセージの復元は特別な場合のみ実行
         restore_needed = False
-        st.session_state.chat_initialized = True
         # チュートリアル保護フラグがある場合のみ初期メッセージを復元
         if st.session_state.get('preserve_initial_message', False):
             if 'chat' in st.session_state and 'messages' in st.session_state.chat:
@@ -1237,9 +1236,10 @@ def render_chat_tab_content(managers):
                 st.success("🔓 セーフティ解除モードに切り替えました！")
             else:
                 st.info("🔒 セーフティ有効モードに戻しました。")
-
+            # 状態変更後に即座にUIを更新
+            st.session_state.chat_initialized = True
             st.rerun()
-            
+
         # 好感度に応じた現在の色を計算する関数（スコープ問題回避のため外部定義）
         def get_affection_color(affection_val):
             if affection_val < 20: