Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -207,7 +207,7 @@ def create_collection_name(filepath):
|
|
207 |
return collection_name
|
208 |
|
209 |
# Initialize database
|
210 |
-
def initialize_database(
|
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)
|