Update app.py

app.py

@@ -139,11 +139,16 @@ def delete_files(files_to_delete):
             if database.docstore._dict[doc_id].metadata["source"] not in deleted_files:
                 new_docs.append(database.docstore._dict[doc_id])
 
-        # 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")
+        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(
@@ -152,7 +157,11 @@ def delete_files(files_to_delete):
         label="Select documents to query"
     )
 
-    return f"Deleted files: {', '.join(deleted_files)}", updated_selector
+    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")
@@ -573,6 +582,12 @@ with demo:
                         inputs=[file_input, parser_dropdown], 
                         outputs=[update_output, document_selector])
 
+    document_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"
+    )
+    
     # Add delete button
     delete_button = gr.Button("Delete Selected Files")
     delete_output = gr.Textbox(label="Delete Status")