ui and clear button

app.py

@@ -136,19 +136,10 @@ if __name__ == "__main__":
             conversation_history.append(msg)
             yield "", conversation_history
 
-
     def clear_message(chat_history: list):
+        agent.memory.reset()
         return chat_history.clear(), ""
 
-
-    def stop_gen():
-        agent = CodeAgent(
-            tools=[retriever_tool],
-            model=model,
-            max_steps=10,
-            verbosity_level=10,
-        )
-
     with gr.Blocks() as b:
         gr.Markdown("# Demo agentic rag on some general knowledge.")
         chatbot = gr.Chatbot(type="messages", height=1000)
@@ -159,8 +150,9 @@ if __name__ == "__main__":
             stop_generating_button = gr.Button("stop generating")
             enter_button = gr.Button("enter")
         reply_button_click_event = enter_button.click(enter_message, [textbox, chatbot], [textbox, chatbot])
-        textbox.submit(enter_message, [textbox, chatbot], [textbox, chatbot])
-        clear_messages_button.click(fn=clear_message, inputs=chatbot, outputs=[chatbot, textbox], )
-        stop_generating_button.click(fn=stop_gen(), cancels=[reply_button_click_event])
+        submit_event = textbox.submit(enter_message, [textbox, chatbot], [textbox, chatbot])
+        clear_messages_button.click(fn=clear_message, inputs=chatbot, outputs=[chatbot, textbox],
+                                    cancels=[reply_button_click_event, submit_event])
+        stop_generating_button.click(cancels=[reply_button_click_event, submit_event])
 
     b.launch()