Update utils/vector_store.py

utils/vector_store.py

@@ -1,3 +1,4 @@
+
 # utils/vector_store.py
 import faiss
 import numpy as np
@@ -7,18 +8,28 @@ import os
 from pathlib import Path
 
 class VectorStore:
-    def __init__(self, persist_directory: str = "./data/faiss"):
-        self.persist_directory = persist_directory
+    def __init__(self):
+        # Use absolute path for HF Spaces
+        self.persist_directory = "/data/faiss"
         self.index = None
         self.documents = []
         self.metadata = []
         
-        # Create directory if it doesn't exist
-        Path(persist_directory).mkdir(parents=True, exist_ok=True)
+        # Ensure directories exist
+        self._create_data_directories()
         
         # Try to load existing index and data
         self._load_or_create_index()
 
+    def _create_data_directories(self):
+        """Create necessary data directories"""
+        # Create main data directory
+        Path("/data").mkdir(parents=True, exist_ok=True)
+        # Create FAISS specific directory
+        Path(self.persist_directory).mkdir(parents=True, exist_ok=True)
+        # Create uploads directory
+        Path("/data/uploads").mkdir(parents=True, exist_ok=True)
+
     def _load_or_create_index(self):
         """Load existing index or create new one"""
         index_path = os.path.join(self.persist_directory, "faiss.index")
@@ -26,6 +37,7 @@ class VectorStore:
         
         try:
             if os.path.exists(index_path) and os.path.exists(data_path):
+                print(f"Loading existing index from {index_path}")
                 # Load existing index
                 self.index = faiss.read_index(index_path)
                 
@@ -34,7 +46,9 @@ class VectorStore:
                     data = pickle.load(f)
                     self.documents = data['documents']
                     self.metadata = data['metadata']
+                print(f"Loaded {len(self.documents)} documents from existing index")
             else:
+                print("No existing index found, creating new one")
                 # Create new index
                 self.index = None  # Will be created when first vectors are added
                 self.documents = []
@@ -45,6 +59,7 @@ class VectorStore:
             self.documents = []
             self.metadata = []
 
+    
     def _save_index(self):
         """Save index and data to disk"""
         if self.index is not None: