Update app.py

app.py

@@ -34,6 +34,9 @@ config = {
     "temperature": 0,
 }
 
+wandb.login(key = wandb_api_key)
+wandb.init(project = "openai-llm-rag", config = 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. """
 
@@ -111,11 +114,6 @@ def rag_chain(llm, prompt, db):
     completion = rag_chain({"query": prompt})
     return completion["result"]
 
-def log_wandb(prompt, completion, rag_option):
-    wandb.login(key = wandb_api_key)
-    wandb.init(project = "openai-llm-rag", config = config)
-    wandb.log({"prompt": prompt, "completion": completion, "rag_option": rag_option})
-
 def invoke(openai_api_key, rag_option, prompt):
     if (openai_api_key == ""):
         raise gr.Error("OpenAI API Key is required.")
@@ -141,7 +139,7 @@ def invoke(openai_api_key, rag_option, prompt):
             completion = llm_chain(llm, prompt)
     except Exception as e:
         raise gr.Error(e)
-    log_wandb(prompt, completion, rag_option)
+    wandb.log({"prompt": prompt, "completion": completion, "rag_option": rag_option})
     return completion
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application that demonstrates a <strong>large language model (LLM)</strong> with