Update src/streamlit_app.py

src/streamlit_app.py

@@ -54,9 +54,14 @@ if st.sidebar.button("🔁 End Chat"):
     memory_buffer["chat_history"] = []
 
 # Extract Chroma DB ZIP (only if not already extracted)
-if not os.path.exists("chroma_db"):
-    with zipfile.ZipFile('chroma_db1.zip', 'r') as zip_ref:
-        zip_ref.extractall('chroma_db')
+
+zip_path = 'src/chroma_db1.zip'
+extract_path = 'chroma_db'
+
+if not os.path.exists(extract_path):
+    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+        zip_ref.extractall(extract_path)
+
 
 # Load embeddings & vector DB
 embedding = HuggingFaceEmbeddings(model_name='all-MiniLM-L6-v2')