Update app.py

app.py

@@ -153,6 +153,11 @@ def run_chain(user_query):
     else:
       print("Invalid inputs.")  
 
+def button_callback():
+    atexit.register(exit_handler)
+
+button = gr.inputs.Button(label="Clear Data", callback=button_callback, help="Click me to clear data before exiting this program.")
+
 iface = gr.Interface(fn=run_chain, inputs="text", outputs="text", title="AI Response")
-iface.launch()
-atexit.register(exit_handler)
+iface.inputs = [button]  # 将按钮添加到输入组件列表中
+iface.launch()