gradio

app.py

@@ -144,7 +144,13 @@ if __name__ == "__main__":
     def clear_message(chat_history: list):
         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:
         chatbot = gr.Chatbot(type="messages", height=1000)
         textbox = gr.Textbox(lines=3, label="")
@@ -155,6 +161,6 @@ if __name__ == "__main__":
         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])
-        stop_generating_button.click(cancels=[reply_button_click_event])
+        stop_generating_button.click(fn= stop_gen,cancels=[reply_button_click_event])
 
     b.launch()