fix chat fragment warnings

Author: dmitrkozlovsk

src/ui/chat/chat.py

@@ -59,22 +59,24 @@ def render() -> None:
             is_stream_output=assistant_cfg.enable_streaming,
         )
         agent_placeholder = history_container.agent_placeholder
+        if chat_state.future_result:
+            agent_placeholder.show_status()
+
         # TODO: feature - select mode for agent (auto, create, multipurpose)
         UserInputField.render(controller)  # if user input: save to chat state and rerun
 
         # Handle input and response logic
-        run_every = 0.5 if chat_state.future_result else None
+        run_every = 2 if chat_state.future_result else None
 
         @st.fragment(run_every=run_every)
-        def handle_future_response(placeholder):
+        def handle_future_response():
             state = ChatStateManager.get_or_create_state()
             if not state.future_result:
                 return
             if state.future_result.done():
                 try:
                     response = state.future_result.result()  # get result
                     state.future_result = None
-                    agent_placeholder.handle_response(response)
                     ChatStateManager.update_state(response)
                 except Exception as e:
                     ChatStateManager.add_message(
@@ -83,10 +85,10 @@ def handle_future_response(placeholder):
                         content=str(e),
                         thinking=None,
                     )
+                    chat_scroll()
                 finally:  # prevent eternal loop
                     st.rerun()
             else:
-                placeholder.show_status()
-                chat_scroll()
+                return
 
-        handle_future_response(agent_placeholder)
+        handle_future_response()

src/ui/chat/elements.py

@@ -174,7 +174,7 @@ def handle_response(self, response: ChatResponse) -> None:
 
     def show_status(self) -> None:
         """Show thinking status message."""
-        self.placeholder.status("Thinking... Please, don't change the page. It may take a while.")
+        self.placeholder.status("Thinking... ", expanded=False)
 
     def show_error(self, error_msg: str | None) -> None:
         """Show error message.