Update app.py

app.py

@@ -2,40 +2,18 @@ import gradio as gr
 import time
 import csv
 from datetime import datetime
+import os
+import pandas as pd
 
 # ====== 菜单数据结构（英文 + 编号） ======
 MENU_DATA = {
-    "Appetizers": [
-        {"id": "A1", "name": "Spring Rolls", "desc": "Crispy rolls with vegetables", "price": 4.5, "img": "https://via.placeholder.com/100"},
-        {"id": "A2", "name": "Dumplings", "desc": "Steamed pork dumplings", "price": 5.0, "img": "https://via.placeholder.com/100"},
-        {"id": "A3", "name": "Salad", "desc": "Fresh vegetable salad", "price": 3.5, "img": "https://via.placeholder.com/100"},
-        {"id": "A4", "name": "Tofu Skewers", "desc": "Grilled tofu with sauce", "price": 4.0, "img": "https://via.placeholder.com/100"},
-        {"id": "A5", "name": "Edamame", "desc": "Boiled soybeans with salt", "price": 3.0, "img": "https://via.placeholder.com/100"},
-    ],
-    "Main Dishes": [
-        {"id": "B1", "name": "Beef Pho", "desc": "Traditional Vietnamese noodle soup", "price": 9.5, "img": "https://via.placeholder.com/100"},
-        {"id": "B2", "name": "Chicken Curry", "desc": "Spicy chicken curry with rice", "price": 10.0, "img": "https://via.placeholder.com/100"},
-        {"id": "B3", "name": "Fried Rice", "desc": "Fried rice with egg and vegetables", "price": 8.0, "img": "https://via.placeholder.com/100"},
-        {"id": "B4", "name": "Grilled Fish", "desc": "Grilled fish with herbs", "price": 11.0, "img": "https://via.placeholder.com/100"},
-        {"id": "B5", "name": "Veggie Stir Fry", "desc": "Mixed vegetables stir-fried with sauce", "price": 8.5, "img": "https://via.placeholder.com/100"},
-    ],
-    "Desserts": [
-        {"id": "C1", "name": "Mango Sticky Rice", "desc": "Sweet coconut rice with mango", "price": 5.0, "img": "https://via.placeholder.com/100"},
-        {"id": "C2", "name": "Taro Cake", "desc": "Steamed taro cake", "price": 4.5, "img": "https://via.placeholder.com/100"},
-        {"id": "C3", "name": "Coconut Jelly", "desc": "Chilled coconut milk jelly", "price": 4.0, "img": "https://via.placeholder.com/100"},
-        {"id": "C4", "name": "Sweet Bean Soup", "desc": "Warm red bean dessert soup", "price": 3.5, "img": "https://via.placeholder.com/100"},
-        {"id": "C5", "name": "Sesame Balls", "desc": "Fried sesame rice balls", "price": 3.5, "img": "https://via.placeholder.com/100"},
-    ],
-    "Drinks": [
-        {"id": "D1", "name": "Lime Soda", "desc": "Refreshing soda with lime", "price": 2.5, "img": "https://via.placeholder.com/100"},
-        {"id": "D2", "name": "Iced Tea", "desc": "Chilled black tea with lemon", "price": 2.0, "img": "https://via.placeholder.com/100"},
-        {"id": "D3", "name": "Coconut Water", "desc": "Natural coconut juice", "price": 3.0, "img": "https://via.placeholder.com/100"},
-        {"id": "D4", "name": "Soy Milk", "desc": "Sweetened soy milk", "price": 2.0, "img": "https://via.placeholder.com/100"},
-        {"id": "D5", "name": "Bottled Water", "desc": "Still mineral water", "price": 1.5, "img": "https://via.placeholder.com/100"},
-    ],
+    "Appetizers": [...],  # （略，保留原结构）
+    "Main Dishes": [...],
+    "Desserts": [...],
+    "Drinks": [...],
 }
 
-# ========== Gradio 主界面 ==========
+# ========== 功能函数 ==========
 def record_time(start_time):
     duration = round(time.time() - start_time, 2)
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -44,6 +22,18 @@ def record_time(start_time):
         writer.writerow([timestamp, duration])
     return f"✅ Thank you! Your decision time was {duration} seconds.\nPlease proceed to the next page and enter your selected item IDs."
 
+def show_csv():
+    if not os.path.exists("decision_times.csv"):
+        return "No data recorded yet."
+    df = pd.read_csv("decision_times.csv", header=None, names=["Timestamp", "Duration (s)"])
+    return df.to_markdown(index=False)
+
+def clear_csv():
+    if os.path.exists("decision_times.csv"):
+        os.remove("decision_times.csv")
+    return "🧹 decision_times.csv has been cleared."
+
+# ========== Gradio 主界面 ==========
 with gr.Blocks() as demo:
     start_time = time.time()
     gr.Markdown("# 📋 Digital Menu\nPlease browse the menu and decide what you'd like to order. When ready, click the button at the bottom.")
@@ -53,11 +43,20 @@ with gr.Blocks() as demo:
             with gr.Tab(label=category):
                 for item in items:
                     with gr.Row():
-                        gr.Image(value=item["img"], shape=(100, 100))
+                        gr.Image(value=item["img"], width=100, height=100)
                         gr.Markdown(f"**{item['id']} - {item['name']}**\n\n{item['desc']}\n\n💰 €{item['price']:.2f}")
 
     submit_btn = gr.Button("✅ I have made my selection")
     output_box = gr.Textbox(label="", lines=3, interactive=False)
     submit_btn.click(fn=record_time, inputs=[gr.State(value=start_time)], outputs=output_box)
 
+    with gr.Accordion("📊 Admin Tools (optional)", open=False):
+        gr.Markdown("### View or Clear Recorded CSV Data")
+        show_btn = gr.Button("📥 Show CSV Data")
+        show_output = gr.Textbox(label="decision_times.csv content", lines=10)
+        show_btn.click(fn=show_csv, outputs=show_output)
+
+        clear_btn = gr.Button("🧹 Clear CSV File")
+        clear_btn.click(fn=clear_csv, outputs=show_output)
+
 demo.launch()