Update app.py

app.py

@@ -131,11 +131,26 @@ print("#1")
 
 # Step 3: Prepare documents (unchanged)
 def prepare_documents_for_chroma(embeddings):
-    print("Preparing documents for chroma")
-    return [
-        Document2(page_content=entry["chunk"], metadata={"chunk_index": i})
-        for i, entry in enumerate(embeddings, start=1)
-    ]
+    status_text = st.empty()
+    progress_bar = st.progress(0)
+    documents = []
+    
+    total_entries = len(embeddings)
+    
+    for i, entry in enumerate(embeddings, start=1):
+        doc = Document2(
+            page_content=entry["chunk"],
+            metadata={"chunk_index": i}
+        )
+        documents.append(doc)
+        
+        progress = i / total_entries
+        progress_bar.progress(progress)
+        status_text.text(f"Processing document {i}/{total_entries} ({progress:.0%})")
+    
+    progress_bar.progress(1.0)
+    status_text.text(f"✅ Successfully prepared {total_entries} documents")
+    return documents
 
 
 print("#2")