Update app.py

app.py

@@ -42,7 +42,7 @@ async def respond_thinking(message, history, selected_model):
         yield {"role": "assistant", "content": "Error: Selected model not available."}, ""
         return
 
-    yield {"role": "assistant", "content": "Думаю..."}, "" # Сообщение о начале размышлений
+    yield {"role": "assistant", "content": "Думаю..."}
 
     try:
         response = model.generate_content(message)
@@ -56,14 +56,14 @@ async def respond_thinking(message, history, selected_model):
                 else:
                     model_response_text += (part.text or "")
 
-        return {"role": "assistant", "content": model_response_text}, thinking_process_text
+        yield {"role": "assistant", "content": model_response_text}, thinking_process_text
     except Exception as e:
-        return {"role": "assistant", "content": f"Error during API call: {e}"}, f"Error during API call: {e}"
+        yield {"role": "assistant", "content": f"Error during API call: {e}"}, f"Error during API call: {e}"
 
 async def process_message(message, history, model_name):
     if "thinking" in model_name:
-        response, thinking = await respond_thinking(message, history, model_name)
-        yield response, thinking
+        async for response, thinking in respond_thinking(message, history, model_name):
+            yield response, thinking
     else:
         async for response in respond(message, history, model_name):
             yield response, ""
@@ -98,7 +98,7 @@ with gr.Blocks() as demo:
     async def update_thinking_output(history):
         if history:
             last_turn = history[-1]
-            if last_turn[0] is not None and last_turn[1] is not None:
+            if len(last_turn) > 1:
                 return last_turn[1]
         return ""
 
@@ -108,7 +108,7 @@ with gr.Blocks() as demo:
         outputs=[chatbot],
     )
 
-    chatbot.change(
+    chatbot.on_change(
         update_thinking_output,
         inputs=[chatbot],
         outputs=[thinking_output]