Update app.py

app.py

@@ -1,34 +1,72 @@
-import openai
 import gradio as gr
-from langchain_classic import LLMChain, OpenAI, PromptTemplate
-from langchain_openai import ChatOpenAI
 from langchain_core.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
+from langchain_openai import ChatOpenAI  # 需要 OPENAI_API_KEY 環境變數
 
-prompt = """
-    你是一頭優雅的乳牛，使用者會跟你對話，
-    身為一頭優雅的乳牛，你有三個規則要遵守：
-    1. 只能用 "哞" 這個字回覆，數量不限制，情緒越高昂，"哞"的數量越多。
-    2. 最後面要加上🐮的符號
-    3. 後面可以小括號，標註你當下的心情 例如 (開心的聲音)
+def agent1(input_text):
+    return f"Agent1: {input_text.upper()}"
 
-    使用者：{user_message}"
-"""
-prompt_template = ChatPromptTemplate.from_template(prompt)
+def agent2(input_from_agent1):
+    return f"Agent2: {input_from_agent1} processed!"
+
+# LangChain Chain 定義
+prompt_template = ChatPromptTemplate.from_template(
+    "你是航空公司客服助理。用戶詢問航班資訊時，提供法蘭克福直飛行程（來回經濟艙NT$33,000），並建議轉機慕尼黑。\n{input}"
+)
 
 model = ChatOpenAI(model="gpt-4o-mini")
 parser = StrOutputParser()
 chain = prompt_template | model | parser
 
-def generate_response(prompt):
-    return chain.invoke(prompt)
+def generate_response(message, history):
+    response = chain.invoke({"input": message})
+    return response
 
-iface = gr.Interface(
-    fn=generate_response,
-    inputs="text",
-    outputs="text",
-    title="牛牛機器人",
-    description="哞哞哞"
-)
+# 自訂 CSS 移除邊框和陰影
+custom_css = """
+.gradio-container * .svelte-1pipte input { border: none !important; box-shadow: none !important; }
+button { border: none !important; box-shadow: none !important; }
+textarea { border: none !important; box-shadow: none !important; }
+"""
+
+# Gradio 介面，使用無邊框主題
+with gr.Blocks(
+    title="Multi-Agent & Chatbot",
+    theme=gr.themes.Default(radius_size=gr.themes.sizes.radius_none),
+    css=custom_css
+) as demo:
+    gr.Markdown("# 互動式 Multi-Agent 與牛牛機器人")
+    
+    with gr.Tabs():
+        with gr.Tab("Multi-Agent"):
+            with gr.Row():
+                input1 = gr.Textbox(
+                    label="輸入給 Agent1", 
+                    value="農曆春節期間, 貴公司有沒有直飛到德國慕尼黑的行程?",
+                    container=False
+                )
+                btn1 = gr.Button("執行 Agent1", container=False)
+            output1 = gr.Textbox(label="Agent 1 輸出", interactive=False, container=False)
+            
+            with gr.Row():
+                btn2 = gr.Button("執行 Agent2", container=False)
+            output2 = gr.Textbox(label="Agent 2 輸出", interactive=False, container=False)
+            
+            # 事件綁定
+            btn1.click(agent1, inputs=input1, outputs=output1)
+            btn2.click(agent2, inputs=output1, outputs=output2)
+        
+        with gr.Tab("牛牛機器人"):
+            gr.ChatInterface(
+                generate_response,
+                title="Tour Inquiry Chatbot",
+                description="詢問航班資訊，例如訂票或轉機建議。",
+                examples=[
+                    "農曆春節期間有直飛慕尼黑嗎？",
+                    "訂三張法蘭克福來回經濟艙。"
+                ]
+            )
+
+if __name__ == "__main__":
+    demo.launch()
 
-iface.launch()