Update app.py

app.py

@@ -56,14 +56,14 @@ async def respond_thinking(message, history, selected_model):
                 else:
                     model_response_text += (part.text or "")
 
-        yield {"role": "assistant", "content": model_response_text}, thinking_process_text
+        yield ({"role": "assistant", "content": model_response_text}, thinking_process_text)
     except Exception as e:
-        yield {"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:
-        async for response, thinking in 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, ""
@@ -86,7 +86,8 @@ with gr.Blocks() as demo:
         additional_inputs=[model_selection],
         title="Gemini Chat",
         description="Общайтесь с моделями Gemini от Google.",
-        type="messages"
+        type="messages",
+        outputs=[gr.Chatbot(), thinking_output] # Явное указание выходов
     )
 
     with gr.Row():
@@ -95,25 +96,12 @@ with gr.Blocks() as demo:
     def change_chatbot(model_name):
         return gr.ChatInterface.update() # No need to change the processing function here
 
-    async def update_thinking_output(history):
-        if history:
-            last_turn = history[-1]
-            if len(last_turn) > 1:
-                return last_turn[1]
-        return ""
-
     model_selection.change(
         change_chatbot,
         inputs=[model_selection],
         outputs=[chatbot],
     )
 
-    chatbot.on_change(
-        update_thinking_output,
-        inputs=[chatbot],
-        outputs=[thinking_output]
-    )
-
     clear_button.click(lambda: None, None, chatbot, queue=False)
     clear_button.click(clear_thinking, outputs=[thinking_output], queue=False)