Update app.py

app.py

@@ -60,12 +60,6 @@ async def respond_thinking(message, history, selected_model):
     except Exception as e:
         yield {"role": "assistant", "content": f"Error during API call: {e}"}, f"Error during API call: {e}"
 
-def update_chatbot_function(model_name):
-    if "thinking" in model_name:
-        return respond_thinking
-    else:
-        return respond
-
 async def process_message(message, history, model_name):
     if "thinking" in model_name:
         generator = respond_thinking(message, history, model_name)
@@ -77,6 +71,9 @@ async def process_message(message, history, model_name):
         async for response, _ in respond(message, history, model_name):
             yield response, ""
 
+def clear_thinking():
+    return ""
+
 with gr.Blocks() as demo:
     gr.Markdown("# Gemini Chatbot с режимом размышления")
 
@@ -85,25 +82,19 @@ with gr.Blocks() as demo:
             AVAILABLE_MODELS, value="gemini-1.5-flash", label="Выберите модель Gemini"
         )
 
+    thinking_output = gr.Code(label="Процесс размышления (для моделей с размышлением)")
+
     chatbot = gr.ChatInterface(
         process_message,  # Функция обработки сообщений передается сюда
         additional_inputs=[model_selection],
         title="Gemini Chat",
         description="Общайтесь с моделями Gemini от Google.",
-        type="messages"
+        type="messages",
+        clear=lambda: ("", "") # Функция для очистки chatbot и thinking_output
     )
 
-    thinking_output = gr.Code(label="Процесс размышления (для моделей с размышлением)")
-
-    def change_function(model_name):
-        # Обновляем функцию обработки сообщений при смене модели
-        if "thinking" in model_name:
-            return respond_thinking
-        else:
-            return respond
-
     def change_chatbot(model_name):
-        return gr.ChatInterface.update(fn=process_message)
+        return gr.ChatInterface.update() # No need to change the processing function here
 
     model_selection.change(
         change_chatbot,
@@ -111,10 +102,5 @@ with gr.Blocks() as demo:
         outputs=[chatbot],
     )
 
-    def clear_thinking():
-        return ""
-
-    chatbot.clear(fn=clear_thinking, outputs=[thinking_output])
-
 if __name__ == "__main__":
     demo.launch()