import gradio as gr
from langchain.llms import OpenAI
from langchain.llms import PromptLayerOpenAI


def chatOpenAI(input):
    llm = OpenAI(temperature=0.9)
    return llm(input)


def promptLayer(input):
    llm = PromptLayerOpenAI(pl_tags=["LangChainGo"])
    return llm(input)


with gr.Blocks() as demo:
    gr.Markdown("# LangChain Test，LLM跑步上车。")
    gr.Interface(fn=promptLayer, inputs="text", outputs="text")
demo.launch()