Update app.py

app.py

@@ -134,6 +134,36 @@ CSS = """
 }
 """
 
+
+def refresh_graph_index(index_name=index_name):
+    # Check if index exists; if not, create it
+    if index_name not in pc.list_indexes().names():
+        pc.create_index(
+            index_name,
+            dimension=dimension_embedding_model,
+            metric='cosine',
+            spec=spec
+        )
+        # Wait for index to be ready
+        while not pc.describe_index(index_name).status['ready']:
+            time.sleep(1)
+
+    # Connect to Pinecone index
+    index = pc.Index(index_name)
+    text_field = 'content'
+    
+    # Initialize Vector Store
+    vectorstore = LangchainPinecone(
+        index, embed_model.embed_query, text_field
+    )
+    return index, vectorstore
+
+def delete_index():
+    pc.delete_index(index_name)
+    index, vectorstore = refresh_graph_index(index_name)
+    return "Delete & Refresh Sucessful "
+
+
 # Title for the application
 TITLE = "<h1 style='text-align:center;'>CNI RAG AGENTIC v0.1</h1>"
 
@@ -213,17 +243,6 @@ with gr.Blocks(css=CSS) as demo:
             ],
         )
 
-    # with gr.Tab("Process PDF"):
-    #     pdf_input = gr.File(label="Upload PDF File")
-    #     pdf_output = gr.Textbox(label="PDF Result", interactive=False)
-
-    #     pdf_button = gr.Button("Process PDF")
-    #     pdf_button.click(
-    #         process_pdf,
-    #         inputs=[pdf_input],
-    #         outputs=pdf_output
-    #     )
-
     with gr.Tab("Process PDF"):
         pdf_input = gr.File(label="Upload PDF File")
         select_client_dropdown = gr.Dropdown(client_name, value="rosariarossi", label="Select or Type Client", allow_custom_value=True)
@@ -247,6 +266,11 @@ with gr.Blocks(css=CSS) as demo:
             inputs=[question_input,answer_with_rag_select_client_dropdown ],
             outputs=rag_output
         )
+    with gr.Tab(label="Manage Files"):
+        with gr.Column():
+            delete_index_button = gr.Button("Delete All Files")
+            delete_index_textout = gr.Textbox(label="Deleted Files and Refresh Result")
+            delete_index_button.click(fn=delete_index, inputs=[],outputs=[delete_index_textout])
 
 # Launch the app
 if __name__ == "__main__":