Update app.py

app.py

@@ -30,6 +30,7 @@ MONGODB_INDEX_NAME = "default"
 
 config = {
     "model": "gpt-4",
+    "rag_option": "Off"
     "temperature": 0,
 }
 
@@ -131,11 +132,13 @@ def invoke(openai_api_key, rag_option, prompt):
         if (rag_option == "Chroma"):
             #splits = document_loading_splitting()
             #document_storage_chroma(splits)
+            config.rag_option = "Chroma"
             db = document_retrieval_chroma(llm, prompt)
             result = rag_chain(llm, prompt, db)
         elif (rag_option == "MongoDB"):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
+            config.rag_option = "MongoDB"
             db = document_retrieval_mongodb(llm, prompt)
             result = rag_chain(llm, prompt, db)
         else: