Update app.py

app.py

@@ -128,6 +128,34 @@ def update_vectors(files, parser):
 
     return f"Vector store updated successfully. Processed {total_chunks} chunks from {len(files)} files using {parser}.", display_documents()
 
+def delete_documents(selected_docs):
+    global uploaded_documents
+    
+    if not selected_docs:
+        return "No documents selected for deletion.", display_documents()
+    
+    embed = get_embeddings()
+    database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
+    
+    deleted_docs = []
+    for doc_name in selected_docs:
+        # Remove from uploaded_documents list
+        uploaded_documents = [doc for doc in uploaded_documents if doc["name"] != doc_name]
+        
+        # Remove from FAISS database
+        docs_to_keep = [doc for doc in database.docstore._dict.values() if doc.metadata["source"] != doc_name]
+        database = FAISS.from_documents(docs_to_keep, embed)
+        
+        deleted_docs.append(doc_name)
+    
+    # Save updated FAISS database
+    database.save_local("faiss_database")
+    
+    # Save updated uploaded_documents list
+    save_documents(uploaded_documents)
+    
+    return f"Deleted documents: {', '.join(deleted_docs)}", display_documents()
+
 def generate_chunked_response(prompt, model, max_tokens=10000, num_calls=3, temperature=0.2, should_stop=False):
     print(f"Starting generate_chunked_response with {num_calls} calls")
     full_response = ""
@@ -484,7 +512,7 @@ def display_documents():
     return gr.CheckboxGroup(
         choices=[doc["name"] for doc in uploaded_documents],
         value=[doc["name"] for doc in uploaded_documents if doc["selected"]],
-        label="Select documents to query"
+        label="Select documents to query or delete"
     )
 
 def initial_conversation():
@@ -558,7 +586,7 @@ demo = gr.ChatInterface(
 
 # Add file upload functionality
 with demo:
-    gr.Markdown("## Upload PDF Documents")
+    gr.Markdown("## Upload and Manage PDF Documents")
 
     with gr.Row():
         file_input = gr.Files(label="Upload your PDF documents", file_types=[".pdf"])
@@ -567,6 +595,8 @@ with demo:
         refresh_button = gr.Button("Refresh Document List")
     
     update_output = gr.Textbox(label="Update Status")
+    document_selector = display_documents()
+    delete_button = gr.Button("Delete Selected Documents")
     
     # Update both the output text and the document selector
     update_button.click(update_vectors, 
@@ -577,6 +607,11 @@ with demo:
     refresh_button.click(refresh_documents, 
                          inputs=[], 
                          outputs=[document_selector])
+    
+    # Add the delete button functionality
+    delete_button.click(delete_documents,
+                        inputs=[document_selector],
+                        outputs=[update_output, document_selector])
 
     gr.Markdown(
     """