Update app.py

app.py

@@ -20,6 +20,7 @@ from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
 #openai.api_key = os.environ["OPENAI_API_KEY"]
+wandb_api_key = os.environ["WANDB_API_KEY"]
 
 MONGODB_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
 client = MongoClient(MONGODB_URI)
@@ -33,11 +34,6 @@ config = {
     "temperature": 0,
 }
 
-wandb_api_key = os.environ["WANDB_API_KEY"]
-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. """
 
@@ -140,7 +136,11 @@ def invoke(openai_api_key, rag_option, prompt):
             completion = llm_chain(llm, prompt)
     except Exception as e:
         raise gr.Error(e)
+    wandb.login(key = wandb_api_key)
+    wandb.config[model] = model
+    wandb.config[temperature] = temperature
     wandb.config[rag_option] = rag_option
+    wandb.init(project = "openai-llm-rag")
     wandb.log({"prompt": prompt, "completion": completion})
     return completion