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