ui and clear button

app.py

@@ -147,15 +147,9 @@ if __name__ == "__main__":
 
 
     def clear_message(chat_history: list):
+        agent.memory.reset()
         return chat_history.clear(), ""
 
-    def stop_gen():
-        agent = CodeAgent(
-            tools=[sql_engine_tool],
-            model=model,
-            max_steps=10,
-            verbosity_level=10,
-        )
     with gr.Blocks() as b:
         gr.Markdown('''# Demo text to sql on paying customers' receipts
         a self correcting text to sql ai agent using smolagents, gradio, HF Spaces, sqlalchemy improved from a smolagents guide
@@ -167,7 +161,8 @@ if __name__ == "__main__":
             clear_messages_button = gr.ClearButton([message_box, chatbot])
             enter_button = gr.Button("enter")
         reply_button_click_event = enter_button.click(enter_message, [message_box, chatbot], [message_box, chatbot])
-        message_box.submit(enter_message, [message_box, chatbot], [message_box, chatbot])
-        stop_generating_button.click(fn= stop_gen,cancels=[reply_button_click_event])
+        message_submit = message_box.submit(enter_message, [message_box, chatbot], [message_box, chatbot])
+        stop_generating_button.click(fn= stop_gen,cancels=[reply_button_click_event,message_submit])
+        clear_messages_button.click(clear_message,outputs=[chatbot,message_box])
 
     b.launch()