Update app.py

app.py

@@ -1,62 +1,81 @@
 import gradio as gr
-import pandas as pd
-from datetime import datetime
-
-# 初始化CSV文件路径
-csv_file = "dog_records.csv"
-
-# 如果CSV文件不存在，则创建文件并添加列名
-def init_csv():
-    try:
-        df = pd.read_csv(csv_file)
-    except FileNotFoundError:
-        df = pd.DataFrame(columns=["time", "action", "details"])
-        df.to_csv(csv_file, index=False)
-
-# 记录数据到CSV
-def record_data(action, details):
-    time = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
-    new_record = pd.DataFrame([[time, action, details]], columns=["time", "action", "details"])
-    
-    # 读取现有的CSV并追加新记录
-    df = pd.read_csv(csv_file)
-    df = pd.concat([df, new_record], ignore_index=True)
-    df.to_csv(csv_file, index=False)
+import json
+import os
+import datetime
 
-    return "记录已保存！"
+# 确保 1.json 文件存在，如果不存在则创建
+json_file_path = '1.json'
+if not os.path.exists(json_file_path):
+    with open(json_file_path, 'w') as f:
+        json.dump([], f)
 
-# 删除错误记录
-def delete_record(time_to_delete):
-    df = pd.read_csv(csv_file)
-    df = df[df['time'] != time_to_delete]  # 删除匹配的记录
-    df.to_csv(csv_file, index=False)
-    return f"记录 {time_to_delete} 已删除！"
+# 加载现有记录
+def load_records():
+    with open(json_file_path, 'r') as f:
+        records = json.load(f)
+    return records
 
-# 查看最近的记录
-def show_records():
-    df = pd.read_csv(csv_file)
-    return df.tail(10)  # 显示最近的10条记录
+# 更新记录到文件
+def save_records(records):
+    with open(json_file_path, 'w') as f:
+        json.dump(records, f)
+
+# 记录操作
+def record_action(action):
+    records = load_records()
+    timestamp = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+    record = {'action': action, 'timestamp': timestamp}
+    records.append(record)
+    save_records(records)
+    return records
 
-# 初始化CSV文件
-init_csv()
+# 删除记录
+def delete_record(index):
+    records = load_records()
+    if 0 <= index < len(records):
+        del records[index]
+        save_records(records)
+    return records
 
-# 创建Gradio界面
+# 显示当前记录
+def show_records():
+    records = load_records()
+    return '\n'.join([f"{rec['timestamp']} - {rec['action']}" for rec in records])
+
+# Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("### 记录狗狗的活动时间（拉屎、尿尿、吃东西）")
-    
+    gr.Markdown("## 狗狗记录应用")
+
+    # 按钮定义
     with gr.Row():
-        action_input = gr.Dropdown(choices=["拉屎", "尿尿", "吃东西"], label="选择活动")
-        details_input = gr.Dropdown(choices=["全吃了", "吃了一半", "没吃"], label="吃东西情况")
+        record_buttons = [
+            gr.Button(value="拉屎", elem_id="defecation", color="brown"),
+            gr.Button(value="尿尿", elem_id="urination", color="blue"),
+            gr.Button(value="吃东西全吃了", elem_id="eaten_all", color="green"),
+            gr.Button(value="吃了一半", elem_id="eaten_half", color="yellow"),
+            gr.Button(value="没吃", elem_id="eaten_none", color="red")
+        ]
+    
+    # 显示当前记录的文本框
+    output = gr.Textbox(label="记录", interactive=False, lines=10)
     
-    record_button = gr.Button("记录")
-    delete_button = gr.Button("删除记录")
-    show_button = gr.Button("查看最近记录")
+    # 删除记录输入框和按钮
+    with gr.Row():
+        delete_index_input = gr.Number(label="删除记录索引（从0开始）", min=0)
+        delete_button = gr.Button("删除记录")
     
-    output_text = gr.Textbox(label="反馈")
-    output_table = gr.DataFrame()
+    # 显示记录
+    show_button = gr.Button("显示记录")
+
+    # 动作按钮事件处理
+    for button in record_buttons:
+        button.click(record_action, inputs=[button], outputs=[output])
+
+    # 删除记录处理
+    delete_button.click(delete_record, inputs=[delete_index_input], outputs=[output])
 
-    record_button.click(record_data, inputs=[action_input, details_input], outputs=output_text)
-    delete_button.click(delete_record, inputs=gr.Textbox(), outputs=output_text)
-    show_button.click(show_records, outputs=output_table)
+    # 显示记录按钮事件
+    show_button.click(show_records, outputs=[output])
 
+# 启动 Gradio 应用
 demo.launch()