Update app.py

app.py

@@ -2,41 +2,64 @@ import gradio as gr
 import datetime
 import pandas as pd
 import json
+import os
 
-# 模擬數據存儲
+# 模擬用戶數據
+users = {"admin": "password"}
+
+# 數據存儲文件
 data_file = "dog_activity_data.json"
-users = {"admin": "password"}  # 模擬用戶數據
 
 # 初始化數據文件
 def init_data_file():
-    try:
-        with open(data_file, "r") as file:
-            json.load(file)
-    except FileNotFoundError:
+    if not os.path.exists(data_file):
         with open(data_file, "w") as file:
             json.dump([], file)
 
-# 保存活動數據
+# 加載活動記錄
+def load_data():
+    try:
+        with open(data_file, "r") as file:
+            return json.load(file)
+    except:
+        return []
+
+# 保存活動記錄
 def save_activity(activity):
     timestamp = datetime.datetime.now()
     record = {"活動": activity, "時間": timestamp.strftime("%Y-%m-%d %H:%M:%S")}
-    with open(data_file, "r") as file:
-        data = json.load(file)
+    data = load_data()
     data.append(record)
     with open(data_file, "w") as file:
         json.dump(data, file)
-    return f"記錄成功：{activity} - {record['時間']}", pd.DataFrame(data)
+    return update_activity_display()
 
-# 獲取活動數據
-def get_activity_data():
-    try:
-        with open(data_file, "r") as file:
-            data = json.load(file)
-        return pd.DataFrame(data)
-    except:
+# 更新顯示活動記錄 (僅顯示最近 5 條)
+def update_activity_display(show_all=False):
+    data = load_data()
+    df = pd.DataFrame(data)
+    if df.empty:
         return pd.DataFrame(columns=["活動", "時間"])
+    if not show_all:
+        df = df.tail(5)
+    return df
 
-# 驗證用戶登錄
+# 刪除記錄功能
+def delete_activity(row_index):
+    data = load_data()
+    try:
+        row_index = int(row_index)
+        if 0 <= row_index < len(data):
+            del data[row_index]
+            with open(data_file, "w") as file:
+                json.dump(data, file)
+            return "記錄刪除成功", update_activity_display()
+        else:
+            return "錯誤: 行號超出範圍", update_activity_display()
+    except ValueError:
+        return "錯誤: 請輸入有效的數字行號", update_activity_display()
+
+# 登錄功能
 def login(username, password):
     if username in users and users[username] == password:
         return gr.update(visible=False), gr.update(visible=True), "登錄成功"
@@ -61,13 +84,34 @@ def main_interface():
         main_section = gr.Column(visible=False)
         with main_section:
             gr.Markdown("### 🚀 點擊按鈕記錄活動")
-            activity_buttons = gr.Row()
-            buttons = ["拉屎", "尿尿", "全吃了", "吃了一半", "沒怎麼吃"]
-            activity_btns = [gr.Button(value=btn) for btn in buttons]
+            # 記錄活動按鈕 (帶顏色和 icon)
+            with gr.Row():
+                activity_buttons = {
+                    "💩 拉屎": "#FF6347",
+                    "💦 尿尿": "#4682B4",
+                    "🍽️ 全吃了": "#32CD32",
+                    "🍛 吃了一半": "#FFD700",
+                    "🥄 沒怎麼吃": "#D3D3D3"
+                }
+                button_list = []
+                for name, color in activity_buttons.items():
+                    button_list.append(
+                        gr.Button(name, elem_id=name, interactive=True, color=color)
+                    )
 
             # 活動記錄表格
-            gr.Markdown("### 📊 活動記錄")
-            data_table = gr.Dataframe(value=get_activity_data(), interactive=False)
+            gr.Markdown("### 📊 最近活動記錄 (僅顯示最近5條)")
+            data_table = gr.Dataframe(update_activity_display(), interactive=False)
+
+            # 顯示更多按鈕
+            show_more_btn = gr.Button("顯示更多記錄")
+            show_less_btn = gr.Button("顯示最近5條", visible=False)
+
+            # 刪除記錄功能
+            gr.Markdown("### 🗑️ 刪除記錄")
+            row_to_delete = gr.Textbox(label="輸入要刪除的行號 (從 0 開始)")
+            delete_btn = gr.Button("刪除記錄")
+            delete_msg = gr.Textbox(label="刪除狀態", interactive=False)
 
         # 登錄邏輯
         login_btn.click(
@@ -76,13 +120,28 @@ def main_interface():
             outputs=[login_section, main_section, login_msg]
         )
 
-        # 綁定按鈕事件
-        for btn, activity in zip(activity_btns, buttons):
-            btn.click(
-                lambda a=activity: save_activity(a),
-                inputs=[],
-                outputs=[login_msg, data_table]
-            )
+        # 綁定活動按鈕邏輯
+        for button, activity in zip(button_list, activity_buttons.keys()):
+            button.click(save_activity, inputs=[], outputs=[data_table])
+
+        # 綁定顯示更多邏輯
+        show_more_btn.click(
+            lambda: (update_activity_display(show_all=True), gr.update(visible=False), gr.update(visible=True)),
+            inputs=[],
+            outputs=[data_table, show_more_btn, show_less_btn]
+        )
+        show_less_btn.click(
+            lambda: (update_activity_display(show_all=False), gr.update(visible=True), gr.update(visible=False)),
+            inputs=[],
+            outputs=[data_table, show_more_btn, show_less_btn]
+        )
+
+        # 刪除記錄邏輯
+        delete_btn.click(
+            delete_activity, 
+            inputs=[row_to_delete], 
+            outputs=[delete_msg, data_table]
+        )
 
     return app