Update app.py

app.py

@@ -3,70 +3,76 @@ import json
 import os
 from datetime import datetime
 
-# JSON文件路径
-json_file_path = "1.json"
+# 定义JSON文件路径
+JSON_FILE = "1.json"
 
-# 初始化JSON文件，如果文件不存在
-if not os.path.exists(json_file_path):
-    with open(json_file_path, "w") as f:
-        json.dump({"records": []}, f)
+# 初始化JSON文件
+def initialize_json():
+    if not os.path.exists(JSON_FILE):
+        with open(JSON_FILE, 'w') as f:
+            json.dump([], f)
 
-# 读取JSON文件内容
+# 读取JSON文件
 def read_json():
-    with open(json_file_path, "r") as f:
+    with open(JSON_FILE, 'r') as f:
         data = json.load(f)
-    return data["records"]
+    return data
 
-# 更新JSON文件
-def update_json(records):
-    with open(json_file_path, "w") as f:
-        json.dump({"records": records}, f)
+# 写入JSON文件
+def write_json(data):
+    with open(JSON_FILE, 'w') as f:
+        json.dump(data, f, indent=4)
 
-# 按钮点击后记录数据
-def record_activity(activity):
-    # 获取当前时间
-    time_stamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
-    records = read_json()
-    
-    # 添加新的记录
-    records.append({"activity": activity, "time": time_stamp})
-    update_json(records)
-    
-    return records
+# 记录事件
+def record_event(event_type):
+    data = read_json()
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    data.append({"timestamp": timestamp, "event": event_type})
+    write_json(data)
+    return data
 
-# 删除错误记录
+# 删除记录
 def delete_record(index):
-    records = read_json()
-    if 0 <= index < len(records):
-        del records[index]
-        update_json(records)
-    return records
+    data = read_json()
+    if 0 <= index < len(data):
+        data.pop(index)
+        write_json(data)
+    return data
 
-# 设置Gradio界面
-with gr.Blocks() as demo:
-    # 显示当前记录的表格
-    output = gr.Textbox(label="Current Records", interactive=False)
-    
-    # 设置按钮
-    with gr.Row():
-        gr.Button("拉屎").click(record_activity, inputs="拉屎", outputs=output)
-        gr.Button("尿尿").click(record_activity, inputs="尿尿", outputs=output)
-        gr.Button("全吃了").click(record_activity, inputs="全吃了", outputs=output)
-        gr.Button("吃了一半").click(record_activity, inputs="吃了一半", outputs=output)
-        gr.Button("没吃").click(record_activity, inputs="没吃", outputs=output)
-    
-    # 删除记录按钮
-    delete_index = gr.Number(label="删除记录索引", interactive=True)
-    delete_button = gr.Button("删除记录")
-    delete_button.click(delete_record, inputs=delete_index, outputs=output)
+# 显示记录
+def display_records():
+    data = read_json()
+    return "\n".join([f"{i}: {item['timestamp']} - {item['event']}" for i, item in enumerate(data)])
 
-    # 更新显示的记录
-    def update_output():
-        records = read_json()
-        return "\n".join([f"{record['time']} - {record['activity']}" for record in records])
+# Gradio界面
+def main():
+    initialize_json()
 
-    # 设置定时器自动更新显示
-    output.update(update_output)
+    with gr.Blocks() as demo:
+        gr.Markdown("## 狗狗日常记录")
+        
+        with gr.Row():
+            poop_btn = gr.Button("拉屎", variant="primary")
+            pee_btn = gr.Button("尿尿", variant="secondary")
+            eat_all_btn = gr.Button("全吃了", variant="success")
+            eat_half_btn = gr.Button("吃了一半", variant="warning")
+            eat_none_btn = gr.Button("没吃", variant="danger")
+        
+        output = gr.Textbox(label="记录", interactive=False)
+        delete_index = gr.Number(label="删除记录的索引", precision=0)
+        delete_btn = gr.Button("删除记录")
 
-# 启动Gradio应用
-demo.launch()
+        poop_btn.click(lambda: record_event("拉屎"), None, output, queue=False)
+        pee_btn.click(lambda: record_event("尿尿"), None, output, queue=False)
+        eat_all_btn.click(lambda: record_event("全吃了"), None, output, queue=False)
+        eat_half_btn.click(lambda: record_event("吃了一半"), None, output, queue=False)
+        eat_none_btn.click(lambda: record_event("没吃"), None, output, queue=False)
+        
+        delete_btn.click(lambda idx: delete_record(int(idx)), delete_index, output, queue=False)
+        
+        output.update(display_records())
+
+    demo.launch()
+
+if __name__ == "__main__":
+    main()