Update app.py

app.py

@@ -64,11 +64,9 @@ async def process_message(message, history, model_name):
     if "thinking" in model_name:
         async for response, thinking in respond_thinking(message, history, model_name):
             yield response
-            yield gr.update(value=thinking)
     else:
         async for response in respond(message, history, model_name):
             yield response
-            yield gr.update(value="")
 
 def clear_thinking():
     return ""
@@ -95,7 +93,12 @@ with gr.Blocks() as demo:
         clear_button = gr.Button("Очистить")
 
     def change_chatbot(model_name):
-        return gr.ChatInterface.update() # No need to change the processing function here
+        return gr.ChatInterface.update()
+
+    async def update_thinking(history):
+        if history and history[-1][0] == "assistant" and len(history[-1]) > 1:
+            return history[-1][1]
+        return ""
 
     model_selection.change(
         change_chatbot,
@@ -103,6 +106,12 @@ with gr.Blocks() as demo:
         outputs=[chatbot],
     )
 
+    chatbot.on_message(
+        update_thinking,
+        inputs=[chatbot],
+        outputs=[thinking_output]
+    )
+
     clear_button.click(lambda: None, None, chatbot, queue=False)
     clear_button.click(clear_thinking, outputs=[thinking_output], queue=False)