Update app.py

app.py

@@ -22,7 +22,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"]
+WANDB_API_KEY = os.environ["WANDB_API_KEY"]
 
 os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
 os.environ["WANDB_PROJECT"] = "openai-llm-rag"
@@ -157,9 +157,9 @@ def invoke(openai_api_key, rag_option, prompt):
     except Exception as e:
         completion = e
         raise gr.Error(e)
-    #finally:
+    finally:
         #wandb_log(prompt, completion, rag_option)
-        #wandb.finish()
+        wandb.finish()
     return completion
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application using a <strong>large language model (LLM)</strong> with