Update app.py

app.py

@@ -4,6 +4,7 @@ import gradio as gr
 groups = {}  # 儲存每組學生和血量
 team_health = {}  # 儲存每組的雞排王當前血量
 max_health = 500  # 預設雞排王的最大血量
+selected_team = ""  # 暫存選中的小隊
 
 # 初始化組別
 def initialize_groups(group_info, health):
@@ -21,12 +22,19 @@ def initialize_groups(group_info, health):
         
     return f"已建立 {len(groups)} 組，每組血量為 {max_health}。"
 
+# 選擇攻擊小隊
+def select_team(team_name):
+    global selected_team
+    selected_team = team_name
+    return f"已選擇攻擊小隊：{team_name}"
+
 # 輸入攻擊值並扣血
-def attack_team(team_name, damage):
-    if team_name in team_health:
-        team_health[team_name] = max(0, team_health[team_name] - int(damage))
-        return f"{team_name} 的血量剩餘 {team_health[team_name]}。"
-    return "小隊名稱不存在。"
+def attack_selected_team(damage):
+    global selected_team
+    if selected_team in team_health:
+        team_health[selected_team] = max(0, team_health[selected_team] - int(damage))
+        return f"{selected_team} 的血量剩餘 {team_health[selected_team]}。"
+    return "尚未選擇有效小隊或小隊不存在。"
 
 # 顯示目前所有小隊的血量狀態
 def display_teams():
@@ -38,10 +46,6 @@ def display_teams():
         display += f"{team} ({', '.join(students)}): [{health_bar}] {health}/{max_health}\n"
     return display.strip()
 
-# 更新小隊選項
-def update_team_dropdown():
-    return list(groups.keys())
-
 # Gradio 介面
 with gr.Blocks() as app:
     gr.Markdown("## 雞排王小隊對戰系統")
@@ -57,19 +61,26 @@ with gr.Blocks() as app:
     
     # 攻擊輸入區
     with gr.Row():
-        team_dropdown = gr.Dropdown(label="選擇攻擊小隊", choices=[], interactive=True)
         damage_input = gr.Number(label="輸入攻擊值", value=50)
-        attack_button = gr.Button("攻擊")
+        attack_button = gr.Button("攻擊選中小隊")
     attack_output = gr.Textbox(label="攻擊結果")
     
+    # 小隊按鈕生成
+    team_buttons = gr.Row()
+    def generate_team_buttons():
+        return [
+            gr.Button(team_name).click(select_team, inputs=[], outputs=attack_output, _js=f"() => '{team_name}'")
+            for team_name in groups.keys()
+        ]
+    init_button.click(generate_team_buttons, outputs=team_buttons)
+    
     # 更新小隊血量顯示
     display_button = gr.Button("顯示目前血量")
     teams_display = gr.Textbox(label="小隊血量狀態", lines=10)
     
     # 事件綁定
-    attack_button.click(attack_team, inputs=[team_dropdown, damage_input], outputs=attack_output)
+    attack_button.click(attack_selected_team, inputs=[damage_input], outputs=attack_output)
     display_button.click(display_teams, outputs=teams_display)
-    init_button.click(update_team_dropdown, outputs=team_dropdown)  # 初始化後更新選項
 
 # 啟動應用程式
 app.launch()