Update app.py

app.py

@@ -1,4 +1,3 @@
-import atexit
 import gradio as gr
 #from langchain.document_loaders import UnstructuredPDFLoader
 from langchain.document_loaders import PyPDFLoader
@@ -153,14 +152,20 @@ def run_chain(user_query):
     else:
       print("Invalid inputs.")  
 
-#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.launch()
-
-delete_button = gr.Button("Exit & Clear Data")
-if delete_button:
-    atexit.register(exit_handler)
+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)
+    print("Pinecone Index Namespace: "+namespace+" has been deleted!")
+        
+with gr.Blocks() as demo:
+    gr.Markdown("Start typing below and then click **Run** to see the output.")
+    with gr.Row():
+        user_query = gr.Textbox(placeholder="Enter your query here.")
+        ai_response = gr.Textbox()
+    btn_1 = gr.Button("Get AI Response")
+    btn_2 = gr.Button("Exit & Clear Data")
+    btn_1.click(fn=run_chain, inputs=user_query, outputs=ai_response)
+    btn_2.click(fn=delete_index_namespace)
+
+demo.launch()