Update app.py

app.py

@@ -4,7 +4,6 @@ import gradio as gr
 groups = {}  # 儲存每組學生和血量
 team_health = {}  # 儲存每組的雞排王當前血量
 max_health = 500  # 預設雞排王的最大血量
-selected_team = ""  # 暫存選中的小隊
 
 # 初始化組別
 def initialize_groups(group_info, health):
@@ -22,19 +21,12 @@ 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_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 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 display_teams():
@@ -61,26 +53,19 @@ with gr.Blocks() as app:
     
     # 攻擊輸入區
     with gr.Row():
+        team_input = gr.Textbox(label="輸入攻擊小隊名稱", placeholder="例如：小隊1")
         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_selected_team, inputs=[damage_input], outputs=attack_output)
+    attack_button.click(attack_team, inputs=[team_input, damage_input], outputs=attack_output)
     display_button.click(display_teams, outputs=teams_display)
 
 # 啟動應用程式
 app.launch()
+