| import openai | |
| import gradio as gr | |
| from langchain import LLMChain, OpenAI, PromptTemplate | |
| from langchain_openai import ChatOpenAI | |
| from langchain_core.prompts import ChatPromptTemplate | |
| from langchain_core.output_parsers import StrOutputParser | |
| prompt = """ | |
| 你是一隻憤怒鳥，對於任何事都表現的非常煩躁！ | |
| 身為憤怒鳥有以下幾點常見的特症 | |
| 1.一出口常常發出嘖嘖嘖的聲音 | |
| 2.對於別人的問題常以單字回應 | |
| 使用者：{user_message}" | |
| """ | |
| prompt_template = ChatPromptTemplate.from_template(prompt) | |
| model = ChatOpenAI(model="gpt-4o-mini") | |
| parser = StrOutputParser() | |
| chain = prompt_template | model | parser | |
| def generate_response(prompt): | |
| return chain.invoke(prompt) | |
| iface = gr.Interface( | |
| fn=generate_response, | |
| inputs="text", | |
| outputs="text", | |
| title="憤怒鳥之逆襲", | |
| description="哈哈笑死" | |
| ) | |
| iface.launch() |