Update chat.py

chat.py

@@ -12,26 +12,32 @@ def setup_chat(api_key, mobile=False):
     # 過去のメッセージを表示
     display_message_history()
     
-    # 思考プロセスの表示（開閉可能なエクスパンダー）
-    if st.session_state.thinking_process:
-        with st.expander("思考プロセスを表示"):
-            for i, thought in enumerate(st.session_state.thinking_process):
-                st.subheader(thought["title"])
-                st.markdown(thought["content"])
-                st.divider()
-    
-    # 最後に生成されたコードを保存する機能
-    if st.session_state.last_generated_code and st.button("AIが生成したコードをエディタに保存", key="save_ai_code"):
-        with st.spinner("コードを保存中..."):
-            file_name = save_generated_code_to_new_file(st.session_state.last_generated_code)
-            st.success(f"コードを '{file_name}' として保存しました")
-            st.session_state.last_generated_code = None  # 保存したのでクリア
-            
-            # モバイルモードならエディタタブに切り替え
-            if mobile:
-                st.session_state.active_tab = "エディタ"
-            
-            st.rerun()
+    # 明示的に更新する
+    if st.session_state.get("ui_needs_update", False) or True:  # Trueを追加して常に表示
+        # 思考プロセスの表示（開閉可能なエクスパンダー）
+        if st.session_state.thinking_process:
+            with st.expander("思考プロセスを表示"):
+                for i, thought in enumerate(st.session_state.thinking_process):
+                    st.subheader(thought["title"])
+                    st.markdown(thought["content"])
+                    st.divider()
+        
+        # 最後に生成されたコードを保存する機能
+        if st.session_state.last_generated_code:
+            if st.button("AIが生成したコードをエディタに保存", key="save_ai_code"):
+                with st.spinner("コードを保存中..."):
+                    file_name = save_generated_code_to_new_file(st.session_state.last_generated_code)
+                    st.success(f"コードを '{file_name}' として保存しました")
+                    st.session_state.last_generated_code = None  # 保存したのでクリア
+                    
+                    # モバイルモードならエディタタブに切り替え
+                    if mobile:
+                        st.session_state.active_tab = "エディタ"
+                    
+                    st.rerun()
+        
+        # フラグをリセット
+        st.session_state.ui_needs_update = False
     
     # チャット入力
     if prompt := st.chat_input("質問を入力してください"):