Update app.py

app.py

@@ -19,8 +19,6 @@ from pymongo import MongoClient
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
-WANDB_API_KEY = os.environ["WANDB_API_KEY"]
-
 MONGODB_URI = os.environ["MONGODB_ATLAS_CLUSTER_URI"]
 client = MongoClient(MONGODB_URI)
 MONGODB_DB_NAME = "langchain_db"
@@ -120,6 +118,7 @@ def invoke(openai_api_key, rag_option, prompt):
         raise gr.Error("Retrieval Augmented Generation is required.")
     if (prompt == ""):
         raise gr.Error("Prompt is required.")
+    WANDB_API_KEY = os.environ["WANDB_API_KEY"]
     os.environ["LANGCHAIN_WANDB_TRACING"] = "true"
     os.environ["WANDB_PROJECT"] = "openai-llm-rag"
     completion = ""