Update app.py

app.py

@@ -125,6 +125,8 @@ def invoke(openai_api_key, rag_option, prompt):
         raise gr.Error("Retrieval Augmented Generation is required.")
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
+    config.rag_option = rag_option
+    config.update(allow_val_change = True)
     try:
         llm = ChatOpenAI(model_name = config.model, 
                          openai_api_key = openai_api_key, 
@@ -132,13 +134,11 @@ 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: