Update app.py

app.py

@@ -145,7 +145,6 @@ def invoke(openai_api_key, rag_option, prompt):
         raise gr.Error("Retrieval Augmented Generation is required.")
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
-    completion = ""
     try:
         llm = ChatOpenAI(model_name = config["model"], 
                          openai_api_key = openai_api_key, 
@@ -156,19 +155,23 @@ def invoke(openai_api_key, rag_option, prompt):
             db = document_retrieval_chroma(llm, prompt)
             completion = rag_chain(llm, prompt, db)
             result = completion["result"]
+            prompt_template = rag_template
         elif (rag_option == "MongoDB"):
             #splits = document_loading_splitting()
             #document_storage_mongodb(splits)
             db = document_retrieval_mongodb(llm, prompt)
             completion = rag_chain(llm, prompt, db)
             result = completion["result"]
+            prompt_template = rag_template
         else:
             result = llm_chain(llm, prompt)
+            completion = result
+            prompt_template = llm_template
     except Exception as e:
         completion = e
         raise gr.Error(e)
     finally:
-        wandb_trace(prompt, completion, rag_option)
+        wandb_trace(prompt_template, prompt, completion, rag_option)
     return result
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application using a <strong>large language model (LLM)</strong> with