Update app.py

app.py

@@ -45,16 +45,19 @@ def create_collection_name(filepath):
     return collection_name
 
 # Initialize database
-def initialize_database(file_obj, chunk_size, chunk_overlap, progress=gr.Progress()):
-    file_path = file_obj[0].name  # Assuming single file processing
+# Modify the function signature to accept a gr.Progress parameter
+def initialize_database(file_obj, chunk_size, chunk_overlap, progress):
+    file_path = file_obj.name  # Adjusted for direct access assuming a single file
+    progress.update(0.1)  # Example progress update
     collection_name = create_collection_name(file_path)
-    progress(0.1, desc="Creating collection name...")
+    progress.update(0.3)  # Updating progress
     doc_splits = load_doc(file_path, chunk_size, chunk_overlap)
-    progress(0.5, desc="Generating vector database...")
+    progress.update(0.6)  # More updates
     vector_db = create_db(doc_splits, collection_name)
-    progress(0.9, desc="Done!")
+    progress.update(1.0)  # Finalize progress
     return vector_db, collection_name, "Complete!"
 
+
 def demo():
     with gr.Blocks(theme="base") as demo:
         vector_db = gr.State()