Update utils.py

utils.py

@@ -138,6 +138,7 @@ YOUTUBE_DIR = "/youtube"
 HISTORY_PFAD = "/data/history"
 DOCS_DIR_PDF = "chroma/kkg/pdf"
 DOCS_DIR_WORD = "chroma/kkg/word"
+VECTORSTORE_DIR="chroma/kkg"
 
 ###############################################
 #URLs zu Dokumenten oder andere Inhalte, die einbezogen werden sollen
@@ -396,13 +397,25 @@ def document_storage_chroma(splits):
 
 ########################################################
 #Vektorstore speichern - bzw. laden
-def save_splits_and_metadata(splits, filename="splits_and_metadata.pkl"):
-    with open(filename, "wb") as f:
+def save_splits_and_metadata(splits, directory="chroma/kkg", filename="splits_and_metadata.pkl"):
+    # Erstellen des Verzeichnisses, falls es nicht existiert
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+    
+    # Vollständigen Pfad zur Datei erstellen
+    filepath = os.path.join(directory, filename)
+    
+    # Speichern der Splits und Metadaten in der Datei
+    with open(filepath, "wb") as f:
         pickle.dump(splits, f)
 
-def load_splits_and_metadata(filename="splits_and_metadata.pkl"):
-    if os.path.exists(filename):
-        with open(filename, "rb") as f:
+def load_splits_and_metadata(directory="chroma/kkg", filename="splits_and_metadata.pkl"):
+    # Vollständigen Pfad zur Datei erstellen
+    filepath = os.path.join(directory, filename)
+    
+    # Laden der Splits und Metadaten aus der Datei
+    if os.path.exists(filepath):
+        with open(filepath, "rb") as f:
             return pickle.load(f)
     return None