farmax commited on
Commit
bf8283d
·
verified ·
1 Parent(s): eb99b99

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -27
app.py CHANGED
@@ -207,7 +207,7 @@ def create_collection_name(filepath):
207
  return collection_name
208
 
209
  # Initialize database
210
- def initialize_database(vector_db, 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
  print(list_file_path)
@@ -218,32 +218,6 @@ def initialize_database(vector_db, list_file_obj, chunk_size, chunk_overlap, pro
218
  # Load document and create splits
219
  doc_splits = load_doc(list_file_path, chunk_size, chunk_overlap)
220
 
221
- # Create or load vector database
222
- #progress(0.5, desc="Generazione vector database...")
223
- # global vector_db
224
- #vector_db = create_db(doc_splits, collection_name)
225
- #progress(0.9, desc="Fatto!")
226
- #return vector_db, collection_name, "Completato!"
227
-
228
- # Controlla se il database esiste già
229
- if vector_db is not None:
230
- print("Database esistente trovato. Stiamo per eliminarlo.")
231
-
232
- # Eliminare il database esistente
233
- vector_db.client.close()
234
- db_path = vector_db.client.path
235
- if os.path.exists(db_path):
236
- os.remove(db_path)
237
- print(f"Rimosso file database principale: {db_path}")
238
-
239
- # Rimuovi eventuali raccolte
240
- collections = vector_db.collections
241
- for collection in collections:
242
- print(f"Cancellando raccolta: {collection}")
243
- vector_db.delete_collection(collection)
244
-
245
- print("Tutte le raccolte e il database sono stati completamente eliminati.")
246
-
247
  # Creare o caricare il nuovo database
248
  progress(0.5, desc="Generazione vector database...")
249
  vector_db = create_db(doc_splits, collection_name)
 
207
  return collection_name
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
  print(list_file_path)
 
218
  # Load document and create splits
219
  doc_splits = load_doc(list_file_path, chunk_size, chunk_overlap)
220
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
221
  # Creare o caricare il nuovo database
222
  progress(0.5, desc="Generazione vector database...")
223
  vector_db = create_db(doc_splits, collection_name)