Update app.py

app.py

@@ -41,7 +41,6 @@ from beschreibungen import *
 #Validieren des PW
 ANTI_BOT_PW = os.getenv("VALIDATE_PW")
 PREPROCESSED_SPLITS=None 
-SPLIT_TO_ORIGINAL_MAPPING = None
 
 ###############################
 #HF Authentifizierung
@@ -69,7 +68,8 @@ vektordatenbank = load_vectorstore()
 if vektordatenbank is None:
     print("Vektorstore neu.......................")
     # Initialisieren Sie den Vektorstore bei Bedarf
-    vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING = create_vectorstore()
+    #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
+    create_vectorstore()
 else:
     SPLIT_TO_ORIGINAL_MAPPING = load_splits_and_metadata()
 
@@ -213,6 +213,19 @@ def cancel_outputing():
 def reset_textbox():
     return gr.update(value=""),""
 
+################################################
+# Erstellen des Vektorstores
+################################################
+def create_vectorstore():
+    global vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING
+    # Splits zu allen Dokumenten in den Verzeichnissen erstellen
+    PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING = document_loading_splitting()
+    if PREPROCESSED_SPLITS:
+        # Vektordatenbank zu den Splits erstellen
+        vektordatenbank = document_storage_chroma(PREPROCESSED_SPLITS)
+        # Speichern der Splits und Metadaten
+        save_splits_and_metadata((PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING))
+
 
 
 ####################################################
@@ -295,7 +308,8 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
         if vektordatenbank is None:
             print("db neu aufbauen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1")
             #Splits zu allen Dokumenten in den Verzeichnissen erstellen
-            vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING = create_vectorstore()
+            #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
+            create_vectorstore()
             
         if vektordatenbank:
             #Retriever erstellen, um die relevanten Slpits zu einem Prompt zu suchen.... (retrieven)
@@ -362,6 +376,7 @@ async def upload_pdf(file):
         await loop.run_in_executor(pool, upload_file_to_huggingface, file.name, upload_path)
     
     # Asynchrones Erstellen des Vektorstores
+    #vektordatenbank, SPLIT_TO_ORIGINAL_MAPPING werden aktualisiert
     await loop.run_in_executor(pool, create_vectorstore)
     
     return f"PDF '{filename}' erfolgreich hochgeladen."