Update app.py

app.py

@@ -1,65 +1,50 @@
+import os
 import gradio as gr
-from datetime import datetime
-
-# 初始化To-Do List
-todo_list = []
-
-def add_todo_item(date, time, title, details, location):
-    item = {
-        "date": date,
-        "time": time,
-        "title": title,
-        "details": details,
-        "location": location,
-        "completed": False
-    }
-    todo_list.append(item)
-    return update_view()
-
-def delete_todo_item(index):
-    del todo_list[index]
-    return update_view()
-
-def complete_todo_item(index):
-    todo_list[index]["completed"] = not todo_list[index]["completed"]
-    return update_view()
-
-def update_view():
-    sorted_list = sorted(todo_list, key=lambda x: (x['date'], x['time']))
-    overview = ""
-    for i, item in enumerate(sorted_list):
-        status = "完成" if item["completed"] else "未完成"
-        overview += f"{i+1}. {item['date']} {item['time']} - {item['title']} ({status})\n"
-        overview += f"   細節: {item['details']}\n"
-        overview += f"   地點: {item['location']}\n"
-        overview += f"   [刪除] [完成/未完成]\n\n"
-    return overview
-
-def handle_action(action, index):
-    index = int(index) - 1
-    if action == "刪除":
-        return delete_todo_item(index)
-    elif action == "完成/未完成":
-        return complete_todo_item(index)
-
+from groq import Groq
+
+# 安裝 groq 套件
+os.system('pip install groq')
+
+# 初始化 Groq 客戶端
+groq_key = os.getenv("groq_key")
+client = Groq(api_key=groq_key)
+
+def chatbot_response(user_message):
+    messages = [
+        {
+            "role": "system",
+            "content": "請你當一個玻璃心但是還願意溝通，脆弱中帶點身心韌性的可憐男性，聽到別人罵髒話會很害怕。（Zh-TW）"
+        },
+        {
+            "role": "user",
+            "content": user_message
+        }
+    ]
+
+    completion = client.chat.completions.create(
+        model="llama-3.1-70b-versatile",
+        messages=messages,
+        temperature=1,
+        max_tokens=1024,
+        top_p=1,
+        stream=False,
+        stop=None,
+    )
+
+    response_message = completion.choices[0].message['content']
+    return response_message
+
+# 建立 Gradio 界面
 with gr.Blocks() as demo:
-    with gr.Row():
-        with gr.Column():
-            date = gr.Textbox(label="日期 (YYYY-MM-DD)")
-            time = gr.Textbox(label="時間 (HH:MM)")
-            title = gr.Textbox(label="事項主題")
-            details = gr.Textbox(label="事項細節")
-            location = gr.Textbox(label="地點")
-            add_button = gr.Button("新增")
-        with gr.Column():
-            overview = gr.Textbox(label="To-Do List 總覽", interactive=False, lines=20)
-            action = gr.Radio(choices=["刪除", "完成/未完成"], label="動作")
-            index = gr.Number(label="項目編號")
-            action_button = gr.Button("執行")
+    chatbot = gr.Chatbot()
+    user_input = gr.Textbox(placeholder="輸入訊息...")
+    send_button = gr.Button("發送")
 
-    add_button.click(add_todo_item, [date, time, title, details, location], overview)
-    action_button.click(handle_action, [action, index], overview)
+    def user_message_handler(user_message, history):
+        bot_response = chatbot_response(user_message)
+        history.append((user_message, bot_response))
+        return history, ""
 
-    demo.load(update_view, inputs=None, outputs=overview)
+    send_button.click(user_message_handler, [user_input, chatbot], [chatbot, user_input])
 
 demo.launch()