""" HTML Export generator for Hermes sessions. Generates a standalone, beautiful HTML file with all messages embedded. Supports single and multi-session exports with a professional sidebar. No remote dependencies. Enhanced with UI-UX-PRO-MAX design intelligence. """ import json import datetime import secrets from typing import Any, Dict, List from urllib.parse import quote # --- Icons (Lucide-style SVGs) --- ICON_USER = '' ICON_BOT = '' ICON_TERMINAL = '' ICON_WRENCH = '' ICON_SPARKLES = '' ICON_CHEVRON_RIGHT = '' ICON_SEARCH = '' ICON_SHIELD = '' ICON_HERMES = '' HTML_TEMPLATE = """ {page_title}

{sidebar_html}

{sessions_html}

""" def _escape_html(text: str) -> str: if not isinstance(text, str): text = str(text) return text.replace("&", "&").replace("<", "<").replace(">", ">").replace('"', """).replace("'", "'") def _format_timestamp(ts: float) -> str: if not ts: return "N/A" return datetime.datetime.fromtimestamp(ts).strftime("%Y-%m-%d %H:%M:%S") def _generate_messages_html(messages: List[Dict[str, Any]]) -> str: html_list = [] for i, msg in enumerate(messages): role = msg.get("role", "unknown") # Skip internal metadata messages if role == "session_meta": continue content = msg.get("content") or "" timestamp = _format_timestamp(msg.get("timestamp", 0)) # Icon selection role_icon = ICON_TERMINAL if role == "user": role_icon = ICON_USER elif role == "assistant": role_icon = ICON_BOT elif role == "system": role_icon = ICON_SHIELD # Handle multimodal or complex content if isinstance(content, list): content_parts = [] for part in content: if isinstance(part, dict): if part.get("type") == "text": content_parts.append(part.get("text", "")) elif part.get("type") == "image_url": content_parts.append("[Image Attachment]") else: content_parts.append(str(part)) content = "\n".join(content_parts) # Build message HTML. The role feeds two sinks and for tool/MCP messages # is externally influenced, so treat each sink on its own terms: # - display text: HTML-escape (prevents markup/JS injection). # - class attribute: reduce to a single safe CSS token (alnum/-/_), # so a crafted role can neither break out of the attribute nor split # into several unintended classes. Real roles (user/assistant/system/ # tool) are unchanged, so the `.message-` rules still match. safe_role = _escape_html(role) role_class = "".join(c if c.isalnum() or c in "-_" else "-" for c in str(role).lower()) msg_class = f"message message-{role_class} active" # Delay animation for initial items delay_style = f' style="animation-delay: {min(i * 0.05, 1.0)}s"' if i < 10 else "" chevron_html = ICON_CHEVRON_RIGHT.replace('class="', 'class="chevron ') html = f'

' html += f'

{chevron_html} {role_icon} {safe_role}

' html += f'

{timestamp}

' html += '

' # Tool Calls tool_calls = msg.get("tool_calls") if tool_calls: for tc in tool_calls: fn_name = tc.get("function", {}).get("name", "unknown") args = tc.get("function", {}).get("arguments", "{}") html += f'''

{ICON_CHEVRON_RIGHT.replace('class="', 'class="chevron ')} {ICON_WRENCH} Tool Call: {_escape_html(fn_name)}

{_escape_html(args)}

''' # Content if content: if role == "tool": html += f'

{_escape_html(content)}

' else: html += f'

{_escape_html(content)}

' # Reasoning reasoning = msg.get("reasoning") or msg.get("reasoning_content") if reasoning: html += f'''

{ICON_CHEVRON_RIGHT.replace('class="', 'class="chevron ')} {ICON_SPARKLES} Reasoning

{_escape_html(reasoning)}

''' html += '

' html += '

' html_list.append(html) return "\n".join(html_list) def generate_multi_session_html_export(sessions: List[Dict[str, Any]]) -> str: if not sessions: return "

No sessions to export.

" is_multi = len(sessions) > 1 generated_at = datetime.datetime.now().strftime("%Y-%m-%d %H:%M:%S") # Sidebar sidebar_html = "" if is_multi: sidebar_items = [] for s in sessions: sid = str(s.get("id", "N/A")) escaped_sid = _escape_html(sid) title = s.get("title") or s.get("preview") or "Untitled Session" if len(title) > 50: title = title[:47] + "..." date = _format_timestamp(s.get("started_at", 0)).split(" ")[0] item = f'''

{_escape_html(title)}

{_escape_html(sid[:8])} {date}

''' sidebar_items.append(item) sidebar_html = f''' ''' # Main Content sessions_html_list = [] for s in sessions: sid = str(s.get("id", "N/A")) escaped_sid = _escape_html(sid) title = s.get("title") or "Hermes Session" model = s.get("model", "Unknown") started_at = _format_timestamp(s.get("started_at", 0)) messages = s.get("messages", []) messages_html = _generate_messages_html(messages) view_class = "session-view" if not is_multi: view_class += " active" session_view_id = f"view-{escaped_sid}" system_prompt = s.get("system_prompt") system_html = "" if system_prompt: system_html = f'''

{ICON_CHEVRON_RIGHT.replace('class="', 'class="chevron ')} {ICON_SHIELD} System Prompt (Persona)

{_escape_html(system_prompt)}

''' session_html = f'''

{_escape_html(title)}

ID: {escaped_sid}

Model: {_escape_html(model)}

Started: {started_at}

{system_html}

{messages_html}

''' sessions_html_list.append(session_html) script_nonce = secrets.token_urlsafe(16) return HTML_TEMPLATE.format( page_title="Hermes Session Export" if is_multi else _escape_html(sessions[0].get("title", "Hermes Session")), sidebar_html=sidebar_html, sessions_html="\n".join(sessions_html_list), main_margin="var(--sidebar-width)" if is_multi else "0", layout_class="layout-multi" if is_multi else "layout-single", generated_at=generated_at, script_nonce=script_nonce, ) def generate_html_export(session_data: Dict[str, Any]) -> str: """Legacy wrapper for single session export.""" return generate_multi_session_html_export([session_data])