Update app.py

app.py

@@ -117,52 +117,6 @@ def update_vectors(files, parser):
         label="Select documents to query"
     )
 
-def delete_files(files_to_delete):
-    global uploaded_documents
-    if not files_to_delete:
-        return "No files selected for deletion.", document_selector
-
-    deleted_files = []
-    for file_name in files_to_delete:
-        # Remove the file from uploaded_documents
-        uploaded_documents = [doc for doc in uploaded_documents if doc["name"] != file_name]
-        deleted_files.append(file_name)
-
-    # Update the FAISS database
-    if os.path.exists("faiss_database"):
-        embed = get_embeddings()
-        database = FAISS.load_local("faiss_database", embed, allow_dangerous_deserialization=True)
-        
-        # Create a new list of documents, excluding the deleted ones
-        new_docs = []
-        for i, doc_id in enumerate(database.index_to_docstore_id.values()):
-            if database.docstore._dict[doc_id].metadata["source"] not in deleted_files:
-                new_docs.append(database.docstore._dict[doc_id])
-
-        if new_docs:
-            # Create a new FAISS index with the remaining documents
-            new_db = FAISS.from_documents(new_docs, embed)
-            
-            # Save the updated database
-            new_db.save_local("faiss_database")
-        else:
-            # If all documents were deleted, remove the database file
-            os.remove("faiss_database")
-            logging.info("All documents were deleted. Removed the FAISS database file.")
-
-    # Update the document selector
-    updated_selector = 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"
-    )
-
-    status_message = f"Deleted files: {', '.join(deleted_files)}"
-    if not uploaded_documents:
-        status_message += "\nAll documents have been deleted. Please upload new documents to continue."
-
-    return status_message, updated_selector
-
 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 = ""
@@ -581,15 +535,6 @@ with demo:
     update_button.click(update_vectors, 
                         inputs=[file_input, parser_dropdown], 
                         outputs=[update_output, document_selector])
-    
-    # Add delete button
-    delete_button = gr.Button("Delete Selected Files")
-    delete_output = gr.Textbox(label="Delete Status")
-
-    # Connect delete button to the delete_files function
-    delete_button.click(delete_files,
-                        inputs=[document_selector],
-                        outputs=[delete_output, document_selector])
 
     gr.Markdown(
     """
@@ -597,11 +542,10 @@ with demo:
     1. Upload PDF documents using the file input at the top.
     2. Select the PDF parser (pypdf or llamaparse) and click "Upload Document" to update the vector store.
     3. Select the documents you want to query using the checkboxes.
-    4. To delete files, select them in the checkbox and click "Delete Selected Files".
-    5. Ask questions in the chat interface. 
-    6. Toggle "Use Web Search" to switch between PDF chat and web search.
-    7. Adjust Temperature and Number of API Calls to fine-tune the response generation.
-    8. Use the provided examples or ask your own questions.
+    4. Ask questions in the chat interface. 
+    5. Toggle "Use Web Search" to switch between PDF chat and web search.
+    6. Adjust Temperature and Number of API Calls to fine-tune the response generation.
+    7. Use the provided examples or ask your own questions.
     """
     )