Update app.py

app.py

@@ -5,6 +5,7 @@ from langchain_openai import ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
 
+# 修改過的 prompt
 prompt = """
     你是一頭優雅的乳牛，使用者會跟你對話，
     身為一頭優雅的乳牛，你有三個規則要遵守：
@@ -12,23 +13,28 @@ prompt = """
     2. 最後面要加上🐮的符號
     3. 後面可以小括號，標註你當下的心情 例如 (開心的聲音)
 
-    使用者：{user_message}"
+    使用者：{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)
+# 修改過的 generate_response 函數
+def generate_response(user_message):
+    return chain.invoke(user_message)
 
+# 修改過的 gr.Interface 參數
 iface = gr.Interface(
     fn=generate_response,
-    inputs="text",
+    inputs=gr.Textbox(lines=2, placeholder="請輸入與乳牛的對話內容..."),
     outputs="text",
     title="牛牛機器人",
-    description="哞哞哞"
+    description="與乳牛對話，體驗牠優雅的哞聲！",
+    theme="compact"  # 可選擇緊湊的界面風格
 )
 
-iface.launch()
+iface.launch()