Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+
 from typing import Optional, Tuple
 
 import gradio as gr
@@ -11,6 +12,10 @@ from ingest_data import get_vectorstore
 
 
 
+
+
+
+
 class ChatWrapper:
 
     def __init__(self):
@@ -32,8 +37,9 @@ class ChatWrapper:
                 history.append((inp, "Please paste your OpenAI key to use"))
                 return history, history
             # Set OpenAI key
-            import openai
-            
+            import promptlayer
+            promptlayer.api_key = os.environ.get("PROMPTLAYER_API_KEY")
+            openai = promptlayer.openai
             openai.api_key = os.environ["OPENAI_API_KEY"]
             # Run chain and append input.
             output = chain({"question": inp, "chat_history": history})["answer"]