Update app.py

app.py

@@ -477,14 +477,25 @@ def process_answer(question, full_text):
     # Load embeddings
     embeddings = SentenceTransformerEmbeddings(model_name="all-MiniLM-L6-v2")
 
-    # Create a temporary directory for ChromaDB
+
+    import tempfile
+    import os
+    import shutil
+    
+    # Use a safe, writable temp directory
     chroma_dir = os.path.join(tempfile.gettempdir(), "chroma_db")
+    
+    # Clean up old data and recreate
     if os.path.exists(chroma_dir):
         shutil.rmtree(chroma_dir)
-
+    os.makedirs(chroma_dir, exist_ok=True)
+    
+    # Create vectorstore and retriever
     db = Chroma.from_documents(splits, embeddings, persist_directory=chroma_dir)
+    db.persist()  # Optional: Saves data to disk for reuse within the same session
     retriever = db.as_retriever()
 
+
     # Set up the model
     llm = load_model()