Update app.py

app.py

@@ -16,6 +16,8 @@ from langchain.vectorstores import MongoDBAtlasVectorSearch
 
 from pymongo import MongoClient
 
+from wandb.integration.openai import autolog
+
 from dotenv import load_dotenv, find_dotenv
 _ = load_dotenv(find_dotenv())
 
@@ -122,6 +124,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.")
+    autolog()
     completion = ""
     try:
         llm = ChatOpenAI(model_name = config["model"], 
@@ -145,7 +148,8 @@ def invoke(openai_api_key, rag_option, prompt):
         completion = e
         raise gr.Error(e)
     finally:
-        wandb.finish()
+        #wandb.finish()
+        autolog.disable()
     return completion
 
 description = """<strong>Overview:</strong> Context-aware multimodal reasoning application using a <strong>large language model (LLM)</strong> with