DL-Broadcast-Tool/utils/graphics_engine.py

import os
import math
from PIL import Image, ImageDraw, ImageFont

# --- PATHING ---
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
ASSETS_DIR = os.path.join(BASE_DIR, "assets")
EXPORT_DIR = os.path.join(BASE_DIR, "..", "stats", "obs_exports")

# Hex to RGB conversion
COLOR_BLUE = (13, 134, 244) # #0d86f4
COLOR_RED = (244, 13, 53)   # #f40d35

SPOTLIGHT_PATH = os.path.join(EXPORT_DIR, "spotlight_card.png")
TAPE_PATH = os.path.join(EXPORT_DIR, "tale_of_the_tape.png")
DNA_PATH = os.path.join(EXPORT_DIR, "team_dna.png")
WIN_PATH = os.path.join(EXPORT_DIR, "win_probability.png")

def get_font(size):
    try: return ImageFont.truetype("arialbd.ttf", size)
    except: 
        try: return ImageFont.truetype("arial.ttf", size)
        except: return ImageFont.load_default()

def load_bg(screen_name):
    width, height = 3840, 2160
    paths = [os.path.join(ASSETS_DIR, f"{screen_name}_bg.png"), os.path.join(ASSETS_DIR, "background.png")]
    for p in paths:
        if os.path.exists(p): return Image.open(p).resize((width, height))
    return Image.new('RGB', (width, height), color=(10, 10, 15))

def draw_radar_chart(draw, center, radius, params, color):
    points = []
    for i, (label, value) in enumerate(params):
        angle = (i * 2 * math.pi / 5) - (math.pi / 2)
        dist = (max(5, value) / 100) * radius
        points.append((center[0] + dist * math.cos(angle), center[1] + dist * math.sin(angle)))
    
    for r_step in [0.25, 0.5, 0.75, 1.0]:
        grid_points = []
        for i in range(5):
            angle = (i * 2 * math.pi / 5) - (math.pi / 2)
            grid_points.append((center[0] + (radius * r_step) * math.cos(angle), center[1] + (radius * r_step) * math.sin(angle)))
        draw.polygon(grid_points, outline=(255, 255, 255, 30))
        if r_step == 1.0:
            for i, (label, _) in enumerate(params):
                angle = (i * 2 * math.pi / 5) - (math.pi / 2)
                draw.text((center[0] + (radius + 80) * math.cos(angle), center[1] + (radius + 40) * math.sin(angle)), label, font=get_font(50), fill=(255, 255, 255), anchor="mm")
    draw.polygon(points, fill=color + (80,), outline=color + (255,))

def generate_player_card(player_data, team_side="Blue", export_dir=None):
    img = load_bg("spotlight")
    draw = ImageDraw.Draw(img, 'RGBA')
    accent_color = COLOR_BLUE if team_side == "Blue" else COLOR_RED

    name = player_data.get("name", "Unknown").upper()
    draw.text((200, 150), name, font=get_font(220), fill=(255, 255, 255))

    has_career = bool(player_data.get("career")) and not player_data.get("stats_estimated") and not player_data.get("is_season_only")
    is_season_only = bool(player_data.get("is_season_only")) and not player_data.get("stats_estimated")

    if has_career:
        matches = player_data.get("matches", [])
        career_kd = player_data.get("career", {}).get("KD", 1.0)
        last_5 = matches[-5:] if len(matches) >= 5 else matches
        recent_kd = (sum(m.get("Kills",0) for m in last_5)/sum(m.get("Deaths",1) for m in last_5)) if last_5 else career_kd
        diff = recent_kd - career_kd
        draw.rounded_rectangle([200, 400, 950, 500], radius=20, fill=(50, 255, 50) if diff >= 0 else (255, 50, 50))
        draw.text((230, 415), f"{'HOT FORM' if diff >=0 else 'COOLING'} ({'+' if diff >=0 else ''}{diff:.2f} KD)", font=get_font(45), fill=(0,0,0))

        career = player_data.get("career", {})
        stats = [("TOTAL KILLS", f"{int(career.get('kills', 0)):,}"), ("TOTAL SCORE", f"{int(career.get('score', 0)):,}"), ("MAPS PLAYED", f"{int(career.get('maps', 0))}"), ("CAREER KD", f"{career.get('KD', 0):.2f}")]
    elif is_season_only:
        # Real numbers, just not a full multi-season career yet —
        # distinct from "no data at all" below. No HOT FORM/COOLING
        # badge since that needs a real match-by-match history we
        # don't have for these players.
        draw.rounded_rectangle([200, 400, 950, 500], radius=20, fill=(120, 200, 255))
        draw.text((230, 415), "NEW THIS SEASON", font=get_font(45), fill=(0,0,0))

        career = player_data.get("career", {})
        stats = [("SEASON KILLS", f"{int(career.get('kills', 0)):,}"), ("SEASON SCORE", f"{int(career.get('score', 0)):,}"), ("MAPS PLAYED", f"{int(career.get('maps', 0))}"), ("SEASON KD", f"{career.get('KD', 0):.2f}")]
    else:
        draw.rounded_rectangle([200, 400, 950, 500], radius=20, fill=(255, 200, 50))
        draw.text((230, 415), "NEW PLAYER", font=get_font(45), fill=(0,0,0))
        stats = [("TOTAL KILLS", "—"), ("TOTAL SCORE", "—"), ("MAPS PLAYED", "—"), ("CAREER KD", "—")]

    y = 650
    for label, value in stats:
        draw.text((200, y), label, font=get_font(70), fill=(180, 180, 180))
        draw.text((200, y + 80), value, font=get_font(130), fill=(255, 255, 255))
        y += 280

    radar_params = [("SLY", player_data.get('slayer', {}).get('pct') or 0), ("AIM", player_data.get('sharpshooter', {}).get('pct') or 0), ("STA", player_data.get('consistency', {}).get('pct') or 0), ("CLU", player_data.get('clutch', {}).get('pct') or 0), ("ANC", player_data.get('anchor', {}).get('pct') or 0)]
    draw_radar_chart(draw, (1850, 1150), 450, radar_params, accent_color)

    draw.text((2700, 550), "SKILL PROFILE", font=get_font(70), fill=accent_color)
    by = 680
    for label_long, key in [("SLAYING", "slayer"), ("MARKSMAN", "sharpshooter"), ("STABILITY", "consistency"), ("PRESSURE", "clutch"), ("DEFENSIVE", "anchor")]:
        pct = player_data.get(key, {}).get("pct") or 0
        draw.text((2700, by), label_long, font=get_font(45), fill=(200, 200, 200))
        draw.rectangle([2700, by+60, 3600, by+90], fill=(40, 40, 45))
        draw.rectangle([2700, by+60, 2700 + (900 * (pct/100)), by+90], fill=accent_color)
        draw.text((3610, by+45), f"{int(pct)}%", font=get_font(40), fill=(255,255,255), anchor="lm")
        by += 180

    if not has_career:
        if is_season_only:
            intel = "NEW THIS SEASON — percentiles above are based on this season's real stats only, not a full career. One to watch!"
        else:
            intel = "NEW PLAYER — no career stats recorded yet. One to watch!"
    else:
        best_tag = max(["slayer", "sharpshooter", "consistency", "clutch", "anchor"], key=lambda k: player_data.get(k, {}).get("pct") or 0)
        intel = player_data.get(best_tag, {}).get("summary", "Reliable team player.")
    draw.rectangle([150, 1850, 3690, 2080], fill=(20, 20, 25))
    draw.text((200, 1880), "SCOUTING REPORT", font=get_font(70), fill=accent_color)
    draw.text((200, 1960), intel, font=get_font(60), fill=(230, 230, 230))
    save_path = os.path.join(export_dir, "spotlight_card.png") if export_dir else SPOTLIGHT_PATH
    os.makedirs(os.path.dirname(save_path), exist_ok=True)
    img.save(save_path)

def generate_tale_of_the_tape(rosterA, rosterB, nameA, nameB, export_dir=None):
    img = load_bg("tape")
    draw = ImageDraw.Draw(img, 'RGBA')
    rosterA.sort(key=lambda p: p.get('career', {}).get('KD', 0), reverse=True)
    rosterB.sort(key=lambda p: p.get('career', {}).get('KD', 0), reverse=True)

    draw.text((960, 150), nameA, font=get_font(150), fill=COLOR_BLUE, anchor="mm")
    draw.text((2880, 150), nameB, font=get_font(150), fill=COLOR_RED, anchor="mm")
    draw.text((1920, 150), "PVP MATCHUPS", font=get_font(80), fill=(255, 255, 255), anchor="mm")

    y = 350
    # Added ANCHOR (Stability is Consistency)
    tags = [("SLAYER", "slayer"), ("AIM", "sharpshooter"), ("STABLE", "consistency"), ("CLUTCH", "clutch"), ("ANCHOR", "anchor")]
    
    for i in range(5):
        pA = rosterA[i] if i < len(rosterA) else {"name": "---"}
        pB = rosterB[i] if i < len(rosterB) else {"name": "---"}
        nameA_display = pA['name'].upper() + (" (NEW)" if pA.get("stats_estimated") else "")
        nameB_display = pB['name'].upper() + (" (NEW)" if pB.get("stats_estimated") else "")
        draw.text((250, y+100), nameA_display, font=get_font(90), fill=(255,255,255))
        draw.text((3590, y+100), nameB_display, font=get_font(90), fill=(255,255,255), anchor="ra")
        
        sub_y = y + 10
        for label, key in tags:
            pctA, pctB = pA.get(key, {}).get("pct") or 0, pB.get(key, {}).get("pct") or 0
            draw.text((1920, sub_y), label, font=get_font(25), fill=(150,150,150), anchor="mm")
            draw.rectangle([1450, sub_y-10, 1850, sub_y+10], fill=(30,30,35))
            draw.rectangle([1850 - (400 * (pctA/100)), sub_y-10, 1850, sub_y+10], fill=COLOR_BLUE)
            draw.rectangle([1990, sub_y-10, 2390, sub_y+10], fill=(30,30,35))
            draw.rectangle([1990, sub_y-10, 1990 + (400 * (pctB/100)), sub_y+10], fill=COLOR_RED)
            sub_y += 48

        draw.line([200, y+280, 3640, y+280], fill=(255,255,255, 15), width=2)
        y += 310

    avgA = sum((p.get('slayer', {}).get('pct') or 0) for p in rosterA) / 5
    avgB = sum((p.get('slayer', {}).get('pct') or 0) for p in rosterB) / 5
    draw.rectangle([0, 1950, 3840, 2160], fill=(15, 15, 20))
    draw.text((400, 2055), f"SQUAD SLAYER POWER: {avgA:.1f}%", font=get_font(60), fill=COLOR_BLUE, anchor="lm")
    draw.text((3440, 2055), f"SQUAD SLAYER POWER: {avgB:.1f}%", font=get_font(60), fill=COLOR_RED, anchor="rm")
    save_path = os.path.join(export_dir, "tale_of_the_tape.png") if export_dir else TAPE_PATH
    os.makedirs(os.path.dirname(save_path), exist_ok=True)
    img.save(save_path)

def generate_team_dna(rosterA, rosterB, nameA, nameB, export_dir=None):
    img = load_bg("dna")
    draw = ImageDraw.Draw(img, 'RGBA')
    draw.text((200, 100), nameA, font=get_font(140), fill=COLOR_BLUE)
    draw.text((3640, 100), nameB, font=get_font(140), fill=COLOR_RED, anchor="ra")
    draw.text((1920, 150), "STRATEGY DNA", font=get_font(100), fill=(255,255,255), anchor="mm")

    # Flag rosters that include a player with no real data at all (not
    # even a verified current-season fetch or a teammate average to
    # fall back on) — otherwise their silent zero drags the team-wide
    # averages below down with no visual explanation, since this
    # graphic (unlike Tale of the Tape) only shows aggregates, not
    # individual player rows where a "(NEW)" tag could be shown.
    unreliableA = sum(1 for p in rosterA if p.get("stats_estimated") and not p.get("career", {}).get("kills"))
    unreliableB = sum(1 for p in rosterB if p.get("stats_estimated") and not p.get("career", {}).get("kills"))
    if unreliableA:
        draw.text((200, 260), f"* {unreliableA} player(s) with no data — averages affected", font=get_font(35), fill=(255, 200, 50))
    if unreliableB:
        draw.text((3640, 260), f"* {unreliableB} player(s) with no data — averages affected", font=get_font(35), fill=(255, 200, 50), anchor="ra")

    tags = ["slayer", "sharpshooter", "consistency", "clutch", "objective_payload", "objective_domination", "anchor"]
    y = 450
    for tag in tags:
        avgA = sum((p.get(tag, {}).get("pct") or 0) for p in rosterA) / max(1, len(rosterA))
        avgB = sum((p.get(tag, {}).get("pct") or 0) for p in rosterB) / max(1, len(rosterB))
        draw.text((1920, y - 40), tag.replace("_", " ").upper(), font=get_font(55), fill=(160,160,160), anchor="mm")
        draw.rectangle([600, y, 1850, y+70], fill=(30, 30, 35))
        draw.rectangle([1850 - (1250 * (avgA/100)), y, 1850, y+70], fill=COLOR_BLUE)
        draw.rectangle([1990, y, 3240, y+70], fill=(30, 30, 35))
        draw.rectangle([1990, y, 1990 + (1250 * (avgB/100)), y+70], fill=COLOR_RED)
        y += 200

    draw.rectangle([0, 1850, 3840, 2160], fill=(15, 15, 20)) 
    vitals = [("TOTAL KILLS", lambda r: sum((p.get('career', {}).get('kills') or 0) for p in r)), ("AVG KD", lambda r: sum((p.get('career', {}).get('KD') or 0) for p in r) / len(r)), ("TOTAL MAPS", lambda r: sum((p.get('career', {}).get('maps') or 0) for p in r))]
    vx_pos = [450, 1100, 1750] # Increased spacing to fix overlap
    for i, (label, func) in enumerate(vitals):
        valA, valB = func(rosterA), func(rosterB)
        draw.text((vx_pos[i], 1920), label, font=get_font(45), fill=(150, 150, 150), anchor="mm")
        draw.text((vx_pos[i], 2020), f"{valA:,.1f}" if isinstance(valA, float) else f"{valA:,}", font=get_font(90), fill=COLOR_BLUE, anchor="mm")
        draw.text((3840 - vx_pos[i], 1920), label, font=get_font(45), fill=(150, 150, 150), anchor="mm")
        draw.text((3840 - vx_pos[i], 2020), f"{valB:,.1f}" if isinstance(valB, float) else f"{valB:,}", font=get_font(90), fill=COLOR_RED, anchor="mm")
    save_path = os.path.join(export_dir, "team_dna.png") if export_dir else DNA_PATH
    os.makedirs(os.path.dirname(save_path), exist_ok=True)
    img.save(save_path)

def draw_vs_donut(draw, center, radius, valA, valB, colorA, colorB, label):
    """Draws a split donut showing relative strength, matching screen sides perfectly."""
    cx, cy = center
    width = 35
    total = valA + valB
    if total <= 0: # Prevent div by zero
        valA, valB = 50, 50
        total = 100
        
    pctA = valA / total
    
    # Draw background circle
    draw.ellipse([cx-radius, cy-radius, cx+radius, cy+radius], outline=(40, 40, 45), width=width)
    
    # --- SWAP CORRECTION ---
    # We start at 90 degrees (6 o'clock) and draw clockwise. 
    # This aligns Team A (Blue, Left) with the Left Side of the donut, 
    # and Team B (Red, Right) with the Right Side.
    startA = 90
    endA = startA + (360 * pctA)
    
    # Blue Arc (Team A - Left side)
    draw.arc([cx-radius, cy-radius, cx+radius, cy+radius], start=startA, end=endA, fill=colorA, width=width)
    # Red Arc (Team B - Right side)
    draw.arc([cx-radius, cy-radius, cx+radius, cy+radius], start=endA, end=startA + 360, fill=colorB, width=width)
    
    # Center Text
    pctA_display = int((valA/total)*100)
    pctB_display = 100 - pctA_display
    draw.text((cx, cy), f"{pctA_display}% | {pctB_display}%", font=get_font(45), fill=(255, 255, 255), anchor="mm")
    draw.text((cx, cy + radius + 60), label, font=get_font(40), fill=(180, 180, 180), anchor="mm")

def generate_win_probability(rosterA, rosterB, nameA, nameB, export_dir=None):
    img = load_bg("win")
    draw = ImageDraw.Draw(img, 'RGBA')
    
    # 1. DATA CALCULATION
    from analysis.team_identity_v2 import summarize_team_tags
    sumA, sumB = summarize_team_tags(rosterA), summarize_team_tags(rosterB)

    # Calculate Strengths (Tags >= 60th percentile)
    def get_strengths(summary):
        s_list = []
        for tag, data in summary.items():
            if data["avg_pct"] >= 60:
                s_list.append(f"{tag.replace('_',' ').title()} Specialists")
        return s_list if s_list else ["Versatile Roster"]

    strengthsA = get_strengths(sumA)
    strengthsB = get_strengths(sumB)

    # 2. WIN PROBABILITY MATH
    weights = {"slayer": 0.35, "consistency": 0.25, "clutch": 0.15, "objective_payload": 0.15, "sharpshooter": 0.10}
    powerA = sum(sumA[tag]["avg_pct"] * weights[tag] for tag in weights)
    powerB = sum(sumB[tag]["avg_pct"] * weights[tag] for tag in weights)
    
    probA = (powerA / (powerA + powerB)) * 100 if (powerA + powerB) > 0 else 50
    probB = 100 - probA

    # 3. HEADER
    draw.text((1920, 150), "MATCH PREVIEW & PROBABILITY", font=get_font(120), fill=(255,255,255), anchor="mm")
    
    # Big Win Numbers
    draw.text((960, 450), nameA.upper(), font=get_font(100), fill=COLOR_BLUE, anchor="mm")
    draw.text((960, 650), f"{probA:.1f}%", font=get_font(300), fill=(255,255,255), anchor="mm")
    
    draw.text((2880, 450), nameB.upper(), font=get_font(100), fill=COLOR_RED, anchor="mm")
    draw.text((2880, 650), f"{probB:.1f}%", font=get_font(300), fill=(255,255,255), anchor="mm")

    # 4. VS DONUT GAUGES (6 Total)
    gauge_tags = [
        ("SLAYING", "slayer"), ("AIM", "sharpshooter"), 
        ("STABILITY", "consistency"), ("PRESSURE", "clutch"), ("TACTICS", "anchor")
    ]
    gx = 450
    for label, key in gauge_tags:
        draw_vs_donut(draw, (gx, 1150), 120, sumA[key]["avg_pct"], sumB[key]["avg_pct"], COLOR_BLUE, COLOR_RED, label)
        gx += 600

    # 6th Gauge: EXPERIENCE (Maps Played)
    mapsA = sum((p.get('career', {}).get('maps') or 0) for p in rosterA)
    mapsB = sum((p.get('career', {}).get('maps') or 0) for p in rosterB)
    draw_vs_donut(draw, (gx, 1150), 120, mapsA, mapsB, COLOR_BLUE, COLOR_RED, "EXPERIENCE")

    # 5. MATCH KEY FACTOR
    diffs = {tag: sumA[tag]["avg_pct"] - sumB[tag]["avg_pct"] for tag in weights}
    best_tag = max(diffs, key=lambda k: abs(diffs[k]))
    favored = nameA if diffs[best_tag] > 0 else nameB
    
    hooks = {
        "slayer": f"The '{favored}' Firepower: Superior elimination pressure likely to dictate map pace.",
        "consistency": f"The '{favored}' Floor: Higher stability makes them significantly harder to upset.",
        "clutch": f"The '{favored}' Resolve: Statistical advantage in high-pressure closing moments.",
        "objective_payload": f"The '{favored}' Efficiency: Superior objective focus provides a clear win-condition.",
        "sharpshooter": f"The '{favored}' Precision: Lethal accuracy in duels could negate enemy numbers."
    }
    key_hook = hooks.get(best_tag, f"The {best_tag.title()} Edge: {favored} hold the strategic advantage.")

    draw.rectangle([400, 1450, 3440, 1700], fill=(20, 20, 25))
    draw.text((1920, 1510), "PRODUCER'S KEY FACTOR", font=get_font(50), fill=COLOR_BLUE if favored == nameA else COLOR_RED, anchor="mm")
    draw.text((1920, 1610), key_hook, font=get_font(70), fill=(255, 255, 255), anchor="mm")

    # 6. TEAM INTEL (Strengths Only)
    def draw_intel(draw, x, name, strengths, color):
        draw.text((x, 1800), f"{name} STRENGTHS", font=get_font(55), fill=color, anchor="mm")
        y_off = 1880
        for s in strengths[:3]: 
            draw.text((x, y_off), f"• {s}", font=get_font(45), fill=(220, 220, 220), anchor="mm")
            y_off += 65

    draw_intel(draw, 960, nameA, strengthsA, COLOR_BLUE)
    draw_intel(draw, 2880, nameB, strengthsB, COLOR_RED)

    save_path = os.path.join(export_dir, "win_probability.png") if export_dir else WIN_PATH
    os.makedirs(os.path.dirname(save_path), exist_ok=True)
    img.save(save_path)