Update chat.py

chat.py

@@ -199,6 +199,13 @@ def process_normal_query(prompt, api_key):
             if code_blocks and len(code_blocks) > 0:
                 st.session_state.last_generated_code = code_blocks[0]
             
+            # 明示的に表示を更新するために変数を設定
+            st.session_state.ui_needs_update = True
+            
+            # モバイルでは自動的にUI更新のためのフラグを設定
+            if st.session_state.mobile_mode:
+                st.rerun()  # モバイルでは即座に画面を更新
+            
         except Exception as e:
             st.error(f"エラーが発生しました: {str(e)}")
     else: