Update utils.py

utils.py

@@ -396,8 +396,7 @@ def document_storage_chroma(splits):
     return vectorstore
 
 ########################################################
-#Vektorstore speichern - bzw. laden
-"""
+#Splits für den Vektorstore speichern - bzw. laden
 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):
@@ -420,7 +419,8 @@ def load_splits_and_metadata(directory="chroma/kkg", filename="splits_and_metada
             return pickle.load(f)
     return None
 
-        
+########################################################
+#Vektorstore speichern - bzw. laden       
 #Laden des Vektorstores - aus den gespeicherten splits
 def load_vectorstore():
     splits_and_metadata = load_splits_and_metadata()
@@ -428,8 +428,9 @@ def load_vectorstore():
         PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING = splits_and_metadata
         return document_storage_chroma(PREPROCESSED_SPLITS)
     return None
-"""
 
+#################################
+#das Mapping der orginal-Splits und der preprocessed Splits speichern
 def save_split_to_original_mapping(mapping, directory="chroma/kkg", mapping_filename="mapping.pkl"):
     # Erstellen des Verzeichnisses, falls es nicht existiert
     if not os.path.exists(directory):
@@ -443,6 +444,7 @@ def save_split_to_original_mapping(mapping, directory="chroma/kkg", mapping_file
     # Hochladen der Mapping-Datei zum Hugging Face Space
     upload_file_to_huggingface(mapping_filepath, f"{directory}/{mapping_filename}")
 
+
 def load_split_to_original_mapping(directory="chroma/kkg", mapping_filename="mapping.pkl"):
     # Laden des Mappings
     mapping_filepath = os.path.join(directory, mapping_filename)
@@ -452,6 +454,15 @@ def load_split_to_original_mapping(directory="chroma/kkg", mapping_filename="map
             return pickle.load(f)
     return None
 
+
+
+
+
+
+    
+
+"""
+
 def save_vectorstore(vectorstore, directory="chroma/kkg", splits_filename="splits_and_metadata.pkl", vectorstore_filename="vectorstore.pkl"):
     # Erstellen des Verzeichnisses, falls es nicht existiert
     if not os.path.exists(directory):
@@ -485,7 +496,7 @@ def load_vectorstore(directory="chroma/kkg", splits_filename="splits_and_metadat
             vectorstore.documents = documents  # Setze die Dokumente im Vektorstore
             return vectorstore, SPLIT_TO_ORIGINAL_MAPPING
     return None, None
-
+"""