Update app.py

app.py

@@ -96,6 +96,10 @@ def document_storage_mongodb(splits):
                                             index_name = MONGODB_INDEX_NAME)
 
 def document_retrieval_chroma(llm, prompt):
+    # Erstelle das Verzeichnis, wenn es nicht existiert
+    if not os.path.exists(CHROMA_DIR):
+        os.makedirs(CHROMA_DIR)
+        
     db = Chroma(embedding_function = OpenAIEmbeddings(),
                 persist_directory = CHROMA_DIR)
     return db
@@ -132,8 +136,8 @@ def invoke(openai_api_key, rag_option, prompt):
                          openai_api_key = openai_api_key, 
                          temperature = 0)
         if (rag_option == "Chroma"):
-            #splits = document_loading_splitting()
-            #document_storage_chroma(splits)
+            splits = document_loading_splitting()
+            document_storage_chroma(splits)
             db = document_retrieval_chroma(llm, prompt)
             result = rag_chain(llm, prompt, db)
         elif (rag_option == "MongoDB"):