Update app.py

app.py

@@ -34,10 +34,6 @@ config = {
     "temperature": 0,
 }
 
-wandb.login(key = wandb_api_key)
-wandb.init(project = "openai-llm-rag", config = config)
-config = wandb.config
-
 template = """If you don't know the answer, just say that you don't know, don't try to make up an answer. Keep the answer as concise as possible. Always say 
               "🧠 Thanks for using the app - Bernd" at the end of the answer. """
 
@@ -115,6 +111,18 @@ def rag_chain(llm, prompt, db):
     completion = rag_chain({"query": prompt})
     return completion["result"]
 
+def evaluate(prompt, completion, rag_option):
+    wandb.login(key = wandb_api_key)
+    wandb.config["model"] = config.model
+    wandb.config["temperature"] = config.temperature
+    wandb.config["rag_option"] = rag_option
+    wandb.config["prompt"] = prompt
+    wandb.config["completion"] = completion
+    wandb.init(project = "openai-llm-rag", config = config)
+    #config = wandb.config
+    wandb.log({"prompt": prompt, "completion": completion})
+    wandb.logout()
+
 def invoke(openai_api_key, rag_option, prompt):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
@@ -140,7 +148,7 @@ def invoke(openai_api_key, rag_option, prompt):
             completion = llm_chain(llm, prompt)
     except Exception as e:
         raise gr.Error(e)
-    wandb.log({"prompt": prompt, "completion": completion})
+    evaluate(prompt, completion, rag_option)
     return completion
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application that demonstrates a <strong>large language model (LLM)</strong> with