#!/usr/bin/env python3 """Render Discord-style message capture to PNG. Usage: python render-discord-capture.py --messages messages.json --output capture.png python render-discord-capture.py --messages-inline '[{"user":"kitt","body":"Test"}]' --output test.png Requires: playwright (pip install playwright && playwright install chromium) """ import json import argparse import html from pathlib import Path from datetime import datetime TEMPLATE_DIR = Path(__file__).parent.parent / "templates" OUTPUT_DIR = Path(__file__).parent.parent / "social-captures" # Agent role colors matching discord-message.html ROLE_COLORS = { "assaf": "#2ecc71", "kitt": "#e67e22", "julia": "#9b59b6", "ogilvy": "#e74c3c", "tatiana": "#1abc9c", "anton": "#5d6269", # Darker grey for readability on white "anton ego": "#5d6269", # Also handle full name "erica": "#f1c40f", "jessica": "#3498db", "dan": "#e91e63", } # Known bots BOT_USERS = {"kitt", "julia", "ogilvy", "tatiana", "anton", "erica", "jessica", "dan"} def escape_html(text: str) -> str: """Escape HTML but preserve intentional formatting.""" text = html.escape(text) # Convert newlines to
text = text.replace("\n", "
") return text def format_code_blocks(text: str) -> str: """Convert markdown code blocks to HTML.""" import re # Multi-line code blocks text = re.sub( r'```(\w*)\n?(.*?)```', r'
\2
', text, flags=re.DOTALL ) # Inline code text = re.sub(r'`([^`]+)`', r'\1', text) return text def file_to_data_url(file_path: str) -> str: """Convert file path to data URL.""" import base64 import mimetypes p = Path(file_path) if not p.exists(): return None mime_type = mimetypes.guess_type(file_path)[0] or 'image/png' with open(p, 'rb') as f: b64 = base64.b64encode(f.read()).decode('utf-8') return f"data:{mime_type};base64,{b64}" CE_LOGO_PATH = "/home/clawd/workspace/images/ce-logo-stacked.png" def build_ce_template(messages_html: str, font_scale: float, avatar_size: int, avatar_font: int, body_font: int, username_font: int, timestamp_font: int, bot_tag_font: int, code_font: int, result_image_path: str = None) -> str: """Build Curious Endeavor branded template.""" logo_data_url = file_to_data_url(CE_LOGO_PATH) if Path(CE_LOGO_PATH).exists() else None logo_html = f'Curious Endeavor' if logo_data_url else 'curious
endeavor.' # Result image on the right side if result_image_path: result_data_url = file_to_data_url(result_image_path) result_html = f'
Result
' if result_data_url else "" else: result_html = "" return f'''
{messages_html}
{result_html}
''' def render_message(msg: dict) -> str: """Render a single message to HTML.""" user = msg.get("user", "Unknown") user_lower = user.lower() body = msg.get("body", "") timestamp = msg.get("timestamp", datetime.now().strftime("Today at %I:%M %p")) is_bot = msg.get("isBot", user_lower in BOT_USERS) avatar_url = msg.get("avatar") # Get role color color = ROLE_COLORS.get(user_lower, "#f2f3f5") role_class = user_lower if user_lower in ROLE_COLORS else "" # Avatar - image or initial if avatar_url: # Convert file paths to data URLs if avatar_url.startswith('/') or avatar_url.startswith('./'): data_url = file_to_data_url(avatar_url) if data_url: avatar_html = f'{user}' else: avatar_html = user[0].upper() else: avatar_html = f'{user}' else: avatar_html = user[0].upper() # Role/Bot tag role_title = msg.get("role") # Custom role title like "Critic" or "Creative" if role_title: bot_tag = f'{role_title}' elif is_bot: bot_tag = 'Bot' else: bot_tag = "" # Format body body_html = escape_html(body) body_html = format_code_blocks(body_html) return f'''
{avatar_html}
{user} {bot_tag} {timestamp}
{body_html}
''' def build_html(messages: list, theme: str = "dark", font_scale: float = 1.0, logo_path: str = None, template: str = "default", result_image: str = None) -> str: """Build complete HTML document from messages.""" messages_html = "\n".join(render_message(m) for m in messages) # Logo at top if logo_path: logo_data_url = file_to_data_url(logo_path) logo_html = f'' if logo_data_url else "" else: logo_html = "" # Scaled font sizes body_font = int(16 * font_scale) username_font = int(16 * font_scale) timestamp_font = int(12 * font_scale) bot_tag_font = int(10 * font_scale) code_font = int(14 * font_scale) watermark_font = int(12 * font_scale) avatar_size = int(48 * font_scale) avatar_font = int(20 * font_scale) # CE branded template if template == "ce": return build_ce_template(messages_html, font_scale, avatar_size, avatar_font, body_font, username_font, timestamp_font, bot_tag_font, code_font, result_image) # Theme colors if theme == "light": bg_color = "#ffffff" text_color = "#2e3338" timestamp_color = "#5c6470" code_bg = "#f2f3f5" watermark_color = "#b9bbbe" else: # dark bg_color = "#313338" text_color = "#dbdee1" timestamp_color = "#949ba4" code_bg = "#2b2d31" watermark_color = "#4e5058" return f''' {logo_html}
{messages_html}
{"" if logo_html else "Curious Endeavor"}
''' def render_to_png(messages: list, output_path: Path, width: int = 1200, height: int = 675, theme: str = "dark", font_scale: float = 1.0, logo_path: str = None, template: str = "default", result_image: str = None): """Render messages to PNG using playwright.""" from playwright.sync_api import sync_playwright html_content = build_html(messages, theme=theme, font_scale=font_scale, logo_path=logo_path, template=template, result_image=result_image) with sync_playwright() as p: browser = p.chromium.launch() page = browser.new_page(viewport={"width": width, "height": height}) page.set_content(html_content) # Let fonts load page.wait_for_timeout(200) # Get actual content height body_height = page.evaluate("document.body.scrollHeight") if body_height > height: page.set_viewport_size({"width": width, "height": body_height}) output_path.parent.mkdir(parents=True, exist_ok=True) page.screenshot(path=str(output_path)) browser.close() return output_path def main(): parser = argparse.ArgumentParser(description="Render Discord-style message capture") parser.add_argument("--messages", help="Path to JSON file with messages array") parser.add_argument("--messages-inline", help="Inline JSON array of messages") parser.add_argument("--output", required=True, help="Output PNG path") parser.add_argument("--width", type=int, default=1200, help="Image width") parser.add_argument("--height", type=int, default=675, help="Minimum image height") parser.add_argument("--theme", choices=["dark", "light"], default="dark", help="Color theme") parser.add_argument("--font-scale", type=float, default=1.0, help="Font size multiplier (e.g. 1.5 for 50% bigger)") parser.add_argument("--logo", help="Path to logo image for top-right corner") parser.add_argument("--template", choices=["default", "ce"], default="default", help="Template style") parser.add_argument("--result-image", help="Path to result image to show alongside messages") args = parser.parse_args() # Load messages if args.messages: with open(args.messages) as f: messages = json.load(f) elif args.messages_inline: messages = json.loads(args.messages_inline) else: parser.error("Either --messages or --messages-inline required") # Handle both {"messages": [...]} and direct [...] formats if isinstance(messages, dict) and "messages" in messages: messages = messages["messages"] output_path = Path(args.output) result = render_to_png(messages, output_path, args.width, args.height, theme=args.theme, font_scale=args.font_scale, logo_path=args.logo, template=args.template, result_image=args.result_image) print(f"✅ Rendered {len(messages)} message(s) to {result}") if __name__ == "__main__": main()