Update app.py

app.py

@@ -40,7 +40,6 @@ from beschreibungen import *
 #Konstanten
 #Validieren des PW
 ANTI_BOT_PW = os.getenv("VALIDATE_PW")
-PREPROCESSED_SPLITS=None 
 
 ###############################
 #HF Authentifizierung
@@ -103,7 +102,7 @@ file_path_download = ""
 # Erstellen des Vektorstores
 ################################################
 def create_vectorstore():
-    global vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING
+    global vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING, ORIGINAL_SPLITS, PREPROCESSED_SPLITS
     # Splits zu allen Dokumenten in den Verzeichnissen erstellen
     PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING, ORIGINAL_SPLITS = document_loading_splitting()
     if PREPROCESSED_SPLITS:
@@ -113,20 +112,31 @@ def create_vectorstore():
         save_splits(PREPROCESSED_SPLITS, ORIGINAL_SPLITS)
         save_split_to_original_mapping(SPLIT_TO_ORIGINAL_MAPPING)
 
+#falls Vektorstore vorhanden: neu laden!!!!!!!!!!!!!!!!!
+def load_vectorstore_and_mapping():
+    global vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING, ORIGINAL_SPLITS, PREPROCESSED_SPLITS
+    preprocessed_splits, original_splits = load_splits()
+    mapping = load_split_to_original_mapping()
+    if preprocessed_splits is not None and original_splits is not None and mapping is not None:
+        # Vektordatenbank zu den Splits erstellen
+        vektordatenbank = document_storage_chroma(PREPROCESSED_SPLITS)
+        SPLIT_TO_ORIGINAL_MAPPING = mapping
+        ORIGINAL_SPLITS = original_splits
+        PREPROCESSED_SPLITS = preprocessed_splits
+    else:
+        print("Vektorstore neu.......................")
+        #fehler beim laden -> Vektorstore neu zusammensetzen
+        create_and_save_vectorstore()
+
 
 
 ###########################################
 # Beim Start der Anwendung - Vektorstore laden!!!!
 ###########################################
 print("Vektorstore laden.........................")
-vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING = load_vectorstore()
-if vektordatenbank is None:
-    print("Vektorstore neu.......................")
-    # Initialisieren Sie den Vektorstore bei Bedarf
-    #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
-    create_vectorstore()
-else:
-    SPLIT_TO_ORIGINAL_MAPPING = load_splits_and_metadata()
+#die Variablen: vektordatenbank, PREPROCESSED_SPLITS, ORGINAL_SPLITS und das Mapping werden neu gesetzt global!!!!
+load_vectorstore_and_mapping()
+
 
 #################################################
 #################################################