Update app.py

app.py

@@ -35,14 +35,19 @@ with gr.Blocks(title="K2-Think Chat") as demo:
     submit_btn = gr.Button("Submit", variant="primary")
     stop_btn = gr.Button("Stop", variant="stop")
 
-    msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+    # Store the event handler to cancel it later
+    bot_event = msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot, chatbot, chatbot
     ).then(lambda: gr.update(value=""), None, msg)
-    submit_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
+    
+    submit_event = submit_btn.click(user, [msg, chatbot], [msg, chatbot], queue=False).then(
         bot, chatbot, chatbot
     ).then(lambda: gr.update(value=""), None, msg)
+    
     clear_btn.click(lambda: None, None, chatbot, queue=False)
-    stop_btn.click(None, None, None, cancels=[bot])
+    
+    # Cancel the events instead of trying to cancel the function directly
+    stop_btn.click(None, None, None, cancels=[bot_event, submit_event])
 
 if __name__ == "__main__":
     demo.launch()