Update app.py

app.py

@@ -1,89 +1,79 @@
 import gradio as gr
 
-# 全域變數儲存小隊資料
-groups = {}
+# 全局變數設定
+groups = {}  # 儲存每組學生和血量
+team_health = {}  # 儲存每組的雞排王當前血量
+max_health = 500  # 預設雞排王的最大血量
 
+# 初始化組別
 def initialize_groups(group_info, health):
-    """初始化小隊資料"""
-    global groups
-    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": students,
-            "health": health,
-            "current_health": health
-        }
-    team_names = list(groups.keys())  # 小隊名稱清單
-    return f"已成功建立 {len(groups)} 個小隊，每隊初始血量 {health}。", team_names
-
-def attack_team(team_name, attack_value):
-    """對指定小隊進行攻擊"""
-    if team_name not in groups:
-        return "請先選擇有效的小隊名稱！", get_team_health_status()
+    global groups, team_health, max_health
+    max_health = int(health)
+    groups.clear()
+    team_health.clear()
     
-    # 計算血量變化
-    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()
+    group_data = group_info.split("\n")
+    for idx, group in enumerate(group_data, start=1):
+        students = group.split(",")  # 每組學生用逗號分隔
+        group_name = f"小隊{idx}"
+        groups[group_name] = students
+        team_health[group_name] = max_health
+        
+    return f"已建立 {len(groups)} 組，每組血量為 {max_health}。"
+
+# 輸入攻擊值並扣血
+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 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)
+# 顯示目前所有小隊的血量狀態
+def display_teams():
+    display = ""
+    for team, students in groups.items():
+        health = team_health.get(team, 0)
+        bar_length = int((health / max_health) * 20)
+        health_bar = "█" * bar_length + " " * (20 - bar_length)
+        display += f"{team} ({', '.join(students)}): [{health_bar}] {health}/{max_health}\n"
+    return display.strip()
 
-# Gradio 介面設置
+# Gradio 介面
 with gr.Blocks() as app:
-    gr.Markdown("## 小隊攻擊系統")
+    gr.Markdown("## 雞排王小隊對戰系統")
     
-    # 初始化小隊部分
+    # 初始化組別與最大血量
     with gr.Row():
-        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="初始化結果")
+        group_input = gr.Textbox(label="輸入組別資訊（每組學生用逗號分隔，換行分組）", placeholder="範例：小明,小華\n小美,小強")
+        health_input = gr.Number(label="設定雞排王最大血量", value=500)
+        init_button = gr.Button("初始化組別")
+    output_init = gr.Textbox(label="系統訊息")
+    
+    init_button.click(initialize_groups, inputs=[group_input, health_input], outputs=output_init)
     
-    # 攻擊小隊部分
-    team_dropdown = gr.Dropdown(choices=[], label="選擇攻擊小隊")
-    attack_input = gr.Number(value=200, label="輸入攻擊值")
-    attack_button = gr.Button("攻擊")
-    output_attack = 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_status = gr.Textbox(label="小隊血量狀態", interactive=False)
+    # 更新小隊血量顯示
+    display_button = gr.Button("顯示目前血量")
+    teams_display = gr.Textbox(label="小隊血量狀態", lines=10)
     
-    # 初始化按鈕事件
-    init_button.click(
-        initialize_groups, 
-        inputs=[group_input, health_input], 
-        outputs=[output_init, team_dropdown]
-    )
+    # 事件綁定
+    attack_button.click(attack_team, inputs=[team_dropdown, damage_input], outputs=attack_output)
+    display_button.click(display_teams, outputs=teams_display)
     
-    # 攻擊按鈕事件
-    attack_button.click(
-        attack_team,
-        inputs=[team_dropdown, attack_input],
-        outputs=[output_attack, team_status]
-    )
-
-    # 初始顯示所有小隊血量狀態
-    init_button.click(
-        get_team_health_status,
-        inputs=None,
-        outputs=team_status
-    )
+    # 更新選單小隊名稱
+    def update_team_dropdown():
+        return list(groups.keys())
+    init_button.click(update_team_dropdown, outputs=team_dropdown)
 
+# 啟動應用程式
 app.launch()
 
 
 
+