Update app.py

app.py

@@ -50,34 +50,40 @@ def main_interface():
         with gr.Row():
             username = gr.Textbox(label="用戶名", placeholder="輸入用戶名")
             password = gr.Textbox(label="密碼", type="password", placeholder="輸入密碼")
-            login_btn = gr.Button("登錄")
+            login_btn = gr.Button(value="登錄")
             login_msg = gr.Textbox(label="登錄消息", interactive=False)
 
         # 隱藏的主界面
         main_content = gr.Group(visible=False)
         with main_content:
             with gr.Row():
-                with gr.Column():
-                    gr.Markdown("### 輸入活動類型")
-                    activity_input = gr.Dropdown(
-                        choices=["拉屎", "尿尿", "全吃了", "吃了一半", "沒怎麼吃"], 
-                        label="活動"
-                    )
-                    activity_btn = gr.Button("記錄活動")
-                    output_msg = gr.Textbox(label="系統消息", interactive=False)
-                with gr.Column():
-                    gr.Markdown("### 活動記錄")
-                    data_table = gr.Dataframe(headers=["活動", "時間"], datatype=["str", "str"], interactive=False)
-                    refresh_btn = gr.Button("刷新數據")
+                gr.Markdown("### 點擊按鈕記錄活動")
+            with gr.Row():
+                buttons = ["拉屎", "尿尿", "全吃了", "吃了一半", "沒怎麼吃"]
+                activity_buttons = [
+                    gr.Button(value=btn) for btn in buttons
+                ]
+            with gr.Row():
+                gr.Markdown("### 活動記錄")
+                data_table = gr.Dataframe(headers=["活動", "時間"], datatype=["str", "str"], interactive=False)
 
         # 功能邏輯
         def handle_login(username, password):
             success, msg = login(username, password)
             return gr.update(visible=success), gr.update(value=msg)
 
+        def handle_activity(activity):
+            save_activity(activity)
+            return get_activity_data(), f"記錄成功：{activity}"
+
         login_btn.click(handle_login, inputs=[username, password], outputs=[main_content, login_msg])
-        activity_btn.click(lambda x: save_activity(x), inputs=activity_input, outputs=output_msg)
-        refresh_btn.click(get_activity_data, outputs=data_table)
+
+        for btn, activity in zip(activity_buttons, buttons):
+            btn.click(
+                lambda a=activity: handle_activity(a),
+                inputs=[],
+                outputs=[data_table, gr.Textbox(interactive=False, visible=False)]
+            )
 
     return app