Update app.py

app.py

@@ -1,5 +1,6 @@
-import atexit
+#import atexit
 import gradio as gr
+from gradio import Button
 #from langchain.document_loaders import UnstructuredPDFLoader
 from langchain.document_loaders import PyPDFLoader
 from langchain.text_splitter import RecursiveCharacterTextSplitter
@@ -85,11 +86,6 @@ def generate_random_string(length):
     return ''.join(random.choice(letters) for i in range(length))      
 random_string = generate_random_string(10)
 
-def exit_handler():
-    pinecone.init(api_key=PINECONE_API_KEY, environment=PINECONE_ENVIRONMENT)
-    index_namespace_to_delete = pinecone.Index(index_name=index_name)
-    index_namespace_to_delete.delete(delete_all=True, namespace=namespace)   
-
 pinecone.init(api_key=PINECONE_API_KEY, environment=PINECONE_ENVIRONMENT)
 index_name = PINECONE_INDEX_NAME
 #index_name = pinecone.Index(index_name)
@@ -153,11 +149,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.")
+def delete_index_namespace():
+    pinecone.init(api_key=PINECONE_API_KEY, environment=PINECONE_ENVIRONMENT)
+    index_namespace_to_delete = pinecone.Index(index_name=index_name)
+    index_namespace_to_delete.delete(delete_all=True, namespace=namespace)
 
-iface = gr.Interface(fn=run_chain, inputs="text", outputs="text", title="AI Response")
-iface.inputs = [button]  # 将按钮添加到输入组件列表中
+delete_button = Button("Delete Index Namespace", delete_index_namespace)
+iface = gr.Interface(fn=run_chain, inputs=[delete_button, "text"], outputs="text", title="AI Response")
 iface.launch()