Update app.py

app.py

@@ -1,74 +1,88 @@
 import gradio as gr
 
-# 模擬資料，方便執行
+# 全域變數儲存小隊資料
 groups = {}
 
 def initialize_groups(group_info, health):
+    """初始化小隊資料"""
     global groups
-    groups = {}
-    group_lines = group_info.split("\n")
+    groups = {}  # 清空現有小隊資料
+    group_lines = group_info.strip().split("\n")  # 逐行解析輸入
     for idx, line in enumerate(group_lines, 1):
+        students = [name.strip() for name in line.split(",")]
         groups[f"小隊{idx}"] = {
-            "students": line.split(","),
+            "students": students,
             "health": health,
             "current_health": health
         }
-    return f"已建立 {len(groups)} 個小隊，每隊初始血量 {health}。"
+    team_names = list(groups.keys())  # 小隊名稱清單
+    return f"已成功建立 {len(groups)} 個小隊，每隊初始血量 {health}。", team_names
 
-def attack_team(team_name, damage):
+def attack_team(team_name, attack_value):
+    """對指定小隊進行攻擊"""
     if team_name not in groups:
-        return "錯誤：小隊不存在！"
-    groups[team_name]["current_health"] -= damage
-    if groups[team_name]["current_health"] < 0:
-        groups[team_name]["current_health"] = 0
-    return f"{team_name} 受到 {damage} 點攻擊！剩餘血量：{groups[team_name]['current_health']}"
+        return "請先選擇有效的小隊名稱！", get_team_health_status()
+    
+    # 計算血量變化
+    groups[team_name]["current_health"] = max(
+        groups[team_name]["current_health"] - attack_value, 0
+    )
+    result = f"{team_name} 受到 {attack_value} 點傷害，剩餘血量 {groups[team_name]['current_health']}/{groups[team_name]['health']}。"
+    return result, get_team_health_status()
 
-def display_teams():
-    result = ""
-    for team_name, data in groups.items():
-        result += f"{team_name}：{data['current_health']}/{data['health']}\n"
-    return result.strip()
+def get_team_health_status():
+    """回傳所有小隊的血量狀態"""
+    status = []
+    for team, data in groups.items():
+        current = data["current_health"]
+        total = data["health"]
+        status.append(f"{team}: {current}/{total}")
+    return "\n".join(status)
 
-# Gradio 介面
+# Gradio 介面設置
 with gr.Blocks() as app:
-    gr.Markdown("## 雞排王小隊對戰系統")
+    gr.Markdown("## 小隊攻擊系統")
     
-    # 初始化組別與最大血量
+    # 初始化小隊部分
     with gr.Row():
-        group_input = gr.Textbox(label="輸入組別資訊（每組學生用逗號分隔，換行分組）", placeholder="範例：小明,小華\n小美,小強")
-        health_input = gr.Number(label="設定雞排王最大血量", value=500)
-        init_button = gr.Button("初始化組別")
-    output_init = gr.Textbox(label="系統訊息")
+        group_input = gr.Textbox(
+            placeholder="請輸入小隊資訊，例如：1號, 2號, 3號, 4號\n5號, 6號, 7號, 8號",
+            label="輸入小隊成員 (每行一個小隊)"
+        )
+        health_input = gr.Number(value=600, label="初始血量")
+    init_button = gr.Button("初始化")
+    output_init = gr.Textbox(label="初始化結果")
     
-    # 攻擊輸入區
-    with gr.Row():
-        team_dropdown = gr.Dropdown(label="選擇攻擊小隊", choices=[], interactive=True)
-        damage_input = gr.Number(label="輸入攻擊值", value=50)
-        attack_button = gr.Button("攻擊")
-    attack_output = gr.Textbox(label="攻擊結果")
+    # 攻擊小隊部分
+    team_dropdown = gr.Dropdown(choices=[], label="選擇攻擊小隊")
+    attack_input = gr.Number(value=200, label="輸入攻擊值")
+    attack_button = gr.Button("攻擊")
+    output_attack = gr.Textbox(label="攻擊結果")
     
-    # 更新小隊血量顯示
-    display_button = gr.Button("顯示目前血量")
-    teams_display = gr.Textbox(label="小隊血量狀態", lines=10)
-
-    # 初始化組別並更新 Dropdown 選項
-    def initialize_groups_and_update_dropdown(group_info, health):
-        message = initialize_groups(group_info, health)  # 呼叫初始化函式
-        dropdown_choices = list(groups.keys())  # 取得小隊名稱列表
-        return message, dropdown_choices
-
-    # 綁定事件：初始化按鈕
+    # 小隊血量顯示
+    team_status = gr.Textbox(label="小隊血量狀態", interactive=False)
+    
+    # 初始化按鈕事件
     init_button.click(
-        initialize_groups_and_update_dropdown,
-        inputs=[group_input, health_input],
+        initialize_groups, 
+        inputs=[group_input, health_input], 
         outputs=[output_init, team_dropdown]
     )
+    
+    # 攻擊按鈕事件
+    attack_button.click(
+        attack_team,
+        inputs=[team_dropdown, attack_input],
+        outputs=[output_attack, team_status]
+    )
 
-    # 攻擊按鈕綁定
-    attack_button.click(attack_team, inputs=[team_dropdown, damage_input], outputs=attack_output)
-    display_button.click(display_teams, outputs=teams_display)
+    # 初始顯示所有小隊血量狀態
+    init_button.click(
+        get_team_health_status,
+        inputs=None,
+        outputs=team_status
+    )
 
-# 啟動應用程式
 app.launch()