184 lines
No EOL
5.5 KiB
Python
184 lines
No EOL
5.5 KiB
Python
# storylines.py (Modernized & GUI-Compatible)
|
|
|
|
from analysis.team_identity_v2 import generate_team_identity_block
|
|
from analysis.team_identity_v2 import summarize_team_tags
|
|
|
|
__all__ = ["matchup_storyline", "generate_storyline"]
|
|
|
|
|
|
# ---------------------------------------------------------
|
|
# Team Tag Aggregation
|
|
# ---------------------------------------------------------
|
|
|
|
from utils.safe_tag import safe_pct, safe_tier, safe_avg
|
|
|
|
def summarize_team_tags(team_players):
|
|
"""
|
|
Aggregates tag tiers + percentiles for a team.
|
|
"""
|
|
|
|
tags = [
|
|
"slayer",
|
|
"objective_payload",
|
|
"objective_domination",
|
|
"sharpshooter",
|
|
"consistency",
|
|
"clutch",
|
|
"anchor",
|
|
"breaker",
|
|
]
|
|
|
|
summary = {}
|
|
|
|
for tag in tags:
|
|
pcts = []
|
|
tiers = []
|
|
|
|
for p in team_players:
|
|
tag_data = p.get(tag, {})
|
|
pcts.append(safe_pct(tag_data.get("pct")))
|
|
tiers.append(safe_tier(tag_data.get("tier")))
|
|
|
|
# -----------------------------
|
|
# FIX: safe top-tier calculation
|
|
# -----------------------------
|
|
valid_tiers = [t for t in tiers if t in "SABCD"]
|
|
top_tier = max(valid_tiers, key=lambda t: "SABCD".index(t)) if valid_tiers else "-"
|
|
|
|
summary[tag] = {
|
|
"avg_pct": safe_avg(pcts),
|
|
"top_tier": top_tier,
|
|
"count_S": tiers.count("S"),
|
|
"count_A": tiers.count("A"),
|
|
"count_B": tiers.count("B"),
|
|
"count_C": tiers.count("C"),
|
|
"count_D": tiers.count("D"),
|
|
}
|
|
|
|
return summary
|
|
|
|
|
|
# ---------------------------------------------------------
|
|
# Team Strength Comparison
|
|
# ---------------------------------------------------------
|
|
|
|
def compare_team_strengths(teamA, teamB, team_a_name, team_b_name):
|
|
"""
|
|
Compares two team tag summaries and returns storyline points.
|
|
"""
|
|
|
|
storyline = []
|
|
|
|
tags = {
|
|
"slayer": "Slayer (Elimination Pressure)",
|
|
"objective_payload": "Payload Objective",
|
|
"objective_domination": "Domination Objective",
|
|
"sharpshooter": "Sharpshooter (Accuracy)",
|
|
"consistency": "Consistency (Stability)",
|
|
"clutch": "Clutch (High-Pressure Performance)",
|
|
"anchor": "Anchor (Defensive Stability)",
|
|
"breaker": "Breaker (Offensive Disruption)",
|
|
}
|
|
|
|
for tag, label in tags.items():
|
|
a = teamA[tag]["avg_pct"]
|
|
b = teamB[tag]["avg_pct"]
|
|
|
|
diff = a - b
|
|
|
|
|
|
# Custom storyline for Anchor
|
|
if tag == "anchor":
|
|
if abs(diff) < 5:
|
|
storyline.append("Both teams are evenly matched in defensive stability.")
|
|
elif diff > 0:
|
|
storyline.append(f"{team_a_name} have stronger defensive stability, surviving longer and preventing collapses.")
|
|
else:
|
|
storyline.append(f"{team_b_name} have stronger defensive stability, giving them an edge in long fights.")
|
|
continue
|
|
|
|
# Generic storyline for all other tags
|
|
if abs(diff) < 5:
|
|
storyline.append(f"Both teams are evenly matched in {label}.")
|
|
elif diff > 0:
|
|
storyline.append(f"{team_a_name} hold an advantage in {label}.")
|
|
else:
|
|
storyline.append(f"{team_b_name} hold an advantage in {label}.")
|
|
|
|
return storyline
|
|
|
|
|
|
# ---------------------------------------------------------
|
|
# Main Storyline Generator
|
|
# ---------------------------------------------------------
|
|
|
|
def generate_storyline(teamA_players, teamB_players):
|
|
"""
|
|
Generates a full caster-ready storyline block.
|
|
"""
|
|
|
|
# Extract real team names
|
|
team_a_name = teamA_players[0].get("team", "Team A")
|
|
team_b_name = teamB_players[0].get("team", "Team B")
|
|
|
|
# 1. Summaries
|
|
A = summarize_team_tags(teamA_players)
|
|
B = summarize_team_tags(teamB_players)
|
|
|
|
# 2. Strength comparison
|
|
matchup_points = compare_team_strengths(A, B, team_a_name, team_b_name)
|
|
|
|
# 3. Identify elite players
|
|
elite_A = [p["name"] for p in teamA_players if p.get("slayer", {}).get("tier") == "S"]
|
|
elite_B = [p["name"] for p in teamB_players if p.get("slayer", {}).get("tier") == "S"]
|
|
|
|
# 4. Build storyline text
|
|
lines = []
|
|
|
|
lines.append(f"MATCH STORYLINE — {team_a_name} vs {team_b_name}\n")
|
|
|
|
# Team A identity
|
|
lines.append(f"{team_a_name} Identity:")
|
|
for tag, data in A.items():
|
|
lines.append(
|
|
f" - {tag.replace('_', ' ').title()}: "
|
|
f"{data['top_tier']} Tier (avg {data['avg_pct']:.1f} percentile)"
|
|
)
|
|
|
|
lines.append("")
|
|
|
|
# Team B identity
|
|
lines.append(f"{team_b_name} Identity:")
|
|
for tag, data in B.items():
|
|
lines.append(
|
|
f" - {tag.replace('_', ' ').title()}: "
|
|
f"{data['top_tier']} Tier (avg {data['avg_pct']:.1f} percentile)"
|
|
)
|
|
|
|
lines.append("")
|
|
|
|
# Elite players
|
|
if elite_A:
|
|
lines.append(f"{team_a_name} Elite Players: {', '.join(elite_A)}")
|
|
if elite_B:
|
|
lines.append(f"{team_b_name} Elite Players: {', '.join(elite_B)}")
|
|
|
|
lines.append("")
|
|
|
|
# Matchup points
|
|
lines.append("Matchup Breakdown:")
|
|
for point in matchup_points:
|
|
lines.append(f" - {point}")
|
|
|
|
return "\n".join(lines)
|
|
|
|
|
|
# ---------------------------------------------------------
|
|
# Legacy Wrapper (GUI expects this exact name)
|
|
# ---------------------------------------------------------
|
|
|
|
def matchup_storyline(teamA_players, teamB_players):
|
|
"""
|
|
Legacy wrapper for GUI compatibility.
|
|
"""
|
|
return generate_storyline(teamA_players, teamB_players) |