import os 
import gradio as gr

from langchain.chat_models import ChatOpenAI
from langchain.agents import load_tools, initialize_agent
from langchain.agents import AgentType
from langchain.tools import AIPluginTool



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(prompt)

title="""
<div style="text-align:center;">
<h1>ChatGPT Plugins playground</h1>
</div>
"""
with gr.Blocks() as demo:
    with gr.Column():
        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")
        run_btn = gr.Button("Run")
        response = gr.Textbox(label="Response")
    run_btn.click(fn=run, 
                 inputs=[prompt, plugin],
                 outputs=[response]
                 )
demo.launch()