Update app.py

app.py

@@ -1,4 +1,4 @@
-#import atexit
+import atexit
 import gradio as gr
 #from langchain.document_loaders import UnstructuredPDFLoader
 from langchain.document_loaders import PyPDFLoader
@@ -85,11 +85,10 @@ 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)
-#atexit.register(exit_handler)
+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
@@ -155,4 +154,5 @@ def run_chain(user_query):
       print("Invalid inputs.")  
 
 iface = gr.Interface(fn=run_chain, inputs="text", outputs="text", title="AI Response")
-iface.launch()
+iface.launch()
+atexit.register(exit_handler)