Update app.py

app.py

@@ -37,6 +37,8 @@ from beschreibungen import *
 #Konstanten
 #Validieren des PW
 ANTI_BOT_PW = os.getenv("VALIDATE_PW")
+PREPROCESSED_SPLITS=None 
+SPLIT_TO_ORIGINAL_MAPPING = None
 
 ###############################
 #HF Authentifizierung
@@ -54,7 +56,7 @@ CHROMA_DIR  = "/chroma/kkg"
 CHROMA_PDF = './chroma/kkg/pdf'
 CHROMA_WORD = './chroma/kkg/word'
 CHROMA_EXCEL = './chroma/kkg/excel'
-DOCS_DIR = "chroma/kkg/pdf"
+DOCS_DIR = "chroma/kkg"
 
 ###########################################
 # Alternativen, um HF Modelle in der rAG Chain einzusetzen
@@ -287,11 +289,11 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
         
         #muss nur einmal ausgeführt werden... 
 		#?????????????????????????????????????????????? Nicht passend zum Promt???????????????????????????
-        if vektordatenbank == None:
+        if vektordatenbank == None or SPLIT_TO_ORIGINAL_MAPPING = None:
             print("db neu aufbauen!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!!1")
-            preprocessed_splits, split_to_original_mapping = document_loading_splitting()
-            if preprocessed_splits:
-                vektordatenbank, retriever = document_storage_chroma(preprocessed_splits)
+            PREPROCESSED_SPLITS, SPLIT_TO_ORIGINAL_MAPPING = document_loading_splitting()
+            if PREPROCESSED_SPLITS:
+                vektordatenbank, retriever = document_storage_chroma(PREPROCESSED_SPLITS)
 
         #kein Bild hochgeladen -> auf Text antworten...
         status = "Antwort der Vektordatenbank"
@@ -302,7 +304,7 @@ def generate_auswahl(prompt_in, file, file_history, chatbot, history, anzahl_doc
         for result in results['relevant_docs']:
             split_id = result.get("metadata", {}).get("split_id")
             if split_id:
-                original_split = split_to_original_mapping[split_id]
+                original_split = SPLIT_TO_ORIGINAL_MAPPING[split_id]
                 relevant_docs_org.append(original_split)
 
         relevant_docs = extract_document_info(relevant_docs_org)