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. 在總站搭計程車: 每人100 元, 每車400 元, 可四人併車。
2. a. 從總站走入烏來老街, 往下走可搭台車, 每人 50 元, 
   b. 台車下車後, 往前走到內洞, 單程約四公里多, 沿途都是平路,
   c. 走到吊橋 (信賢橋) 要過橋, 右轉走完信賢步道,
   d. 再直走就到遊樂區門口了。內洞瀑布負離子滿滿。
3. 內洞瀑布的負離子, 曾有報導稱屬全台最多處 :
   待一小時可抵市區一個月的數量。 


    使用者：{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="routing"
)

iface.launch()