Update app.py

app.py

@@ -126,39 +126,28 @@ def load_vector_index(role: str):
             input_type="search_document"
         )
         Settings.embed_model = embed_model
-
-        # ✅ Use /tmp for Vercel/read-only filesystems
-        if os.getenv("VERCEL") or os.access(".", os.W_OK) is False:
-            persist_dir = os.path.join("/tmp", "chroma_db", role)
-        else:
-            persist_dir = os.path.join("chroma_db", role)
-
-        os.makedirs(persist_dir, exist_ok=True)
-
+        
         # Initialize Chroma client
+        persist_dir = f"./chroma_db/{role}"
         chroma_client = chromadb.PersistentClient(path=persist_dir)
-
-        # ✅ Get or create collection
-        try:
-            chroma_collection = chroma_client.get_collection("documents")
-        except chromadb.errors.NotFoundError:
-            chroma_collection = chroma_client.create_collection("documents")
-
+        
+        # Get the collection
+        chroma_collection = chroma_client.get_collection("documents")
+        
         # Create vector store
         vector_store = ChromaVectorStore(chroma_collection=chroma_collection)
-
+        
         # Create storage context
         storage_context = StorageContext.from_defaults(vector_store=vector_store)
-
+        
         # Load the index
         index = VectorStoreIndex.from_vector_store(
             vector_store=vector_store,
             storage_context=storage_context,
             embed_model=embed_model
         )
-
+        
         return index
-
     except Exception as e:
         st.error(f"Error loading vector index: {str(e)}")
         st.stop()
@@ -329,4 +318,4 @@ def main():
         chat_interface()
 
 if __name__ == "__main__":
-    main()
+    main()