farmax commited on
Commit
414dc63
·
verified ·
1 Parent(s): e413310

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -208,22 +208,19 @@ def create_collection_name(filepath):
208
 
209
  # Initialize database
210
  def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Progress()):
211
- # Delete the existing database
212
- chromadb.delete()
213
- print("Existing database deleted successfully.")
214
  # Create list of documents (when valid)
215
  list_file_path = [x.name for x in list_file_obj if x is not None]
216
  # Create collection_name for vector database
217
- progress(0.1, desc="Creating collection name...")
218
  collection_name = create_collection_name(list_file_path[0])
219
- progress(0.25, desc="Loading document...")
220
  # Load document and create splits
221
  doc_splits = load_doc(list_file_path, chunk_size, chunk_overlap)
222
  # Create or load vector database
223
  progress(0.5, desc="Generazione vector database...")
224
  # global vector_db
225
  vector_db = create_db(doc_splits, collection_name)
226
- progress(0.9, desc="Done!")
227
  return vector_db, collection_name, "Completato!"
228
 
229
 
 
208
 
209
  # Initialize database
210
  def initialize_database(list_file_obj, chunk_size, chunk_overlap, progress=gr.Progress()):
 
 
 
211
  # Create list of documents (when valid)
212
  list_file_path = [x.name for x in list_file_obj if x is not None]
213
  # Create collection_name for vector database
214
+ progress(0.1, desc="Creazione collezione...")
215
  collection_name = create_collection_name(list_file_path[0])
216
+ progress(0.25, desc="Caricamento documenti..")
217
  # Load document and create splits
218
  doc_splits = load_doc(list_file_path, chunk_size, chunk_overlap)
219
  # Create or load vector database
220
  progress(0.5, desc="Generazione vector database...")
221
  # global vector_db
222
  vector_db = create_db(doc_splits, collection_name)
223
+ progress(0.9, desc="Fatto!")
224
  return vector_db, collection_name, "Completato!"
225
 
226