Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
218 |
collection_name = create_collection_name(list_file_path[0])
|
219 |
-
progress(0.25, desc="
|
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="
|
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 |
|