Update app.py

app.py

@@ -8,21 +8,22 @@ from langchain.tools import AIPluginTool
 
 
 
-def run(prompt):
-    tool = AIPluginTool.from_plugin_url("https://www.klarna.com/.well-known/ai-plugin.json")
+def run(prompt, plugin_json):
+    tool = AIPluginTool.from_plugin_url(plugin_json)
     llm = ChatOpenAI(temperature=0)
     tools = load_tools(["requests_all"] )
     tools += [tool]
     agent_chain = initialize_agent( tools, llm, agent=AgentType.ZERO_SHOT_REACT_DESCRIPTION, verbose=True)
-    return agent_chain.run("what t shirts are available in klarna?")
+    return agent_chain.run(prompt)
 
 with gr.Blocks() as demo:
     with gr.Column():
-        prompt = gr.Textbox()
+        prompt = gr.Textbox(label="Prompt", value="what t shirts are available in klarna?")
+        plugin = gr.Textbox(label="Plugin", value="https://www.klarna.com/.well-known/ai-plugin.json")
         run_btn = gr.Button("Run")
-        response = gr.Textbox()
+        response = gr.Textbox(label="Response")
     run_btn.click(fn=run, 
-                 inputs=[prompt],
+                 inputs=[prompt, plugin],
                  outputs=[response]
                  )
 demo.launch()