DL-Broadcast-Tool/analysis/team_identity_v2.py

from utils.safe_tag import safe_pct, safe_tier, safe_avg



from utils.safe_tag import safe_pct, safe_tier, safe_avg

def summarize_team_tags(team_players):
    """
    Aggregates tag tiers + percentiles for a team.
    Returns a dict with:
        - avg_pct
        - top_tier
        - tier counts
    """

    tags = ["slayer", "objective_payload", "sharpshooter", "consistency", "clutch"]
    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")))

        summary[tag] = {
            "avg_pct": safe_avg(pcts),
            "top_tier": max(tiers, key=lambda t: "SABCD".index(t)),
            "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


def classify_team_style(summary):
    """
    Produces a high-level team style classification based on tag strengths.
    """

    slayer = summary["slayer"]["avg_pct"]
    payload = summary["objective_payload"]["avg_pct"]
    sharpshooter = summary["sharpshooter"]["avg_pct"]
    consistency = summary["consistency"]["avg_pct"]
    clutch = summary["clutch"]["avg_pct"]

    # Identify primary style
    primary = max(
        {
            "Slayer-heavy": slayer,
            "Objective-focused": payload,
            "Precision/Sharpshooter": sharpshooter,
            "Stable/Consistent": consistency,
            "High-pressure/Clutch": clutch,
        }.items(),
        key=lambda x: x[1]
    )[0]

    # Identify weaknesses
    weaknesses = []
    if slayer < 40:
        weaknesses.append("low elimination pressure")
    if payload < 40:
        weaknesses.append("weak objective presence")
    if sharpshooter < 40:
        weaknesses.append("below-average accuracy")
    if consistency < 40:
        weaknesses.append("inconsistent match-to-match output")
    if clutch < 40:
        weaknesses.append("poor high-pressure performance")

    return primary, weaknesses


def generate_team_identity_block(team_players, team_name="Team"):
    """
    Returns a structured identity block dict for a team.
    GUI or OBS formatter will convert this into text.
    """

    summary = summarize_team_tags(team_players)
    primary_style, weaknesses = classify_team_style(summary)

    # Build strengths list
    strengths = []
    for tag, data in summary.items():
        if data["avg_pct"] >= 60:
            strengths.append(
                f"Strong {tag.replace('_', ' ').title()} ({data['top_tier']} Tier)"
            )

    # Build tag breakdown
    tag_block = {}
    for tag, data in summary.items():
        tag_block[tag] = {
            "avg_pct": data["avg_pct"],
            "tier": data["top_tier"],
        }

    # Return a dict (NOT a string)
    return {
        "team_name": team_name,
        "team_style": primary_style,
        "strengths": strengths,
        "weaknesses": weaknesses,
        "tags": tag_block,
    }
    
def format_team_identity(identity_data):
    """
    Converts the identity block dict into a readable text block.
    Used by GUI and OBS exporters.
    """

    lines = []

    lines.append(f"Team: {identity_data['team_name']}")
    lines.append(f"Playstyle: {identity_data['team_style']}")
    lines.append("")

    # Strengths
    if identity_data["strengths"]:
        lines.append("Strengths:")
        for s in identity_data["strengths"]:
            lines.append(f"  - {s}")
    else:
        lines.append("Strengths:")
        lines.append("  - None identified")

    lines.append("")

    # Weaknesses
    if identity_data["weaknesses"]:
        lines.append("Weaknesses:")
        for w in identity_data["weaknesses"]:
            lines.append(f"  - {w}")
    else:
        lines.append("Weaknesses:")
        lines.append("  - None identified")

    lines.append("")
    lines.append("Tag Breakdown:")

    for tag, data in identity_data["tags"].items():
        tag_name = tag.replace("_", " ").title()
        lines.append(f"  - {tag_name}: {data['tier']} Tier (avg {data['avg_pct']:.1f} percentile)")

    return "\n".join(lines)