Update app.py

app.py

@@ -8,7 +8,8 @@ from langchain.tools import AIPluginTool
 
 
 
-def run(prompt, plugin_json):
+def run(prompt, plugin_json, openai_api_key):
+    os.environ["OPENAI_API_KEY"] = openai_api_key
     tool = AIPluginTool.from_plugin_url(plugin_json)
     llm = ChatOpenAI(temperature=0)
     tools = load_tools(["requests_all"] )
@@ -30,10 +31,11 @@ with gr.Blocks() as demo:
         gr.HTML(title)
         prompt = gr.Textbox(label="Prompt", value="what t shirts are available in klarna?")
         plugin = gr.Textbox(label="Plugin json", info="You need the .json plugin file of the plugin you want to use", value="https://www.klarna.com/.well-known/ai-plugin.json")
+        openai_api_key = gr.Textbox(label="OpenAI API Key", type="password")
         run_btn = gr.Button("Run")
         response = gr.Textbox(label="Response")
     run_btn.click(fn=run, 
-                 inputs=[prompt, plugin],
+                 inputs=[prompt, plugin, openai_api_key],
                  outputs=[response]
                  )
 demo.launch()