Update app.py

app.py

@@ -22,6 +22,7 @@ import tqdm
 import accelerate
 import re
 # from chromadb.utils import get_default_config
+vector_db = ''
 
 # default_persist_directory = './chroma_HF/'
 list_llm = ["mistralai/Mistral-7B-Instruct-v0.2", "mistralai/Mixtral-8x7B-Instruct-v0.1", "mistralai/Mistral-7B-Instruct-v0.1", \
@@ -72,24 +73,6 @@ def load_db():
         embedding_function=embedding)
     return vectordb
 
-def delete_db(vectordb):
-    # Rimuovi tutte le raccolte
-    collections = vectordb.collections
-    for collection in collections:
-        print(f"Cancellando raccolta: {collection}")
-        vectordb.delete_collection(collection)
-
-    # Rimuovi il client ChromaDB
-    vectordb.client.close()
-
-    # Rimuovi il database principale
-    db_path = vectordb.client.path
-    if os.path.exists(db_path):
-        os.remove(db_path)
-        print(f"Rimosso file database principale: {db_path}")
-
-    print("Tutte le raccolte e il database sono stati completamente eliminati.")
-
 # Initialize langchain LLM chain
 def initialize_llmchain(llm_model, temperature, max_tokens, top_k, vector_db, progress=gr.Progress()):
     progress(0.1, desc="Initializing HF tokenizer...")
@@ -224,7 +207,7 @@ def create_collection_name(filepath):
     return collection_name
 
 # Initialize database
-def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Progress()):
+def initialize_database(vector_db, list_file_obj, chunk_size, chunk_overlap, progress=gr.Progress()):
     # Create list of documents (when valid)
     list_file_path = [x.name for x in list_file_obj if x is not None]
     print(list_file_path)