#!/usr/bin/env python3
"""用Playwright将HTML图表截图"""
from playwright.sync_api import sync_playwright
import os

output_dir = '/root/.openclaw/workspace/ppt_charts_pyecharts'

def screenshot_html(html_file, output_png, width=1400, height=700):
    with sync_playwright() as p:
        browser = p.chromium.launch()
        page = browser.new_page(viewport={'width': width, 'height': height})
        page.goto(f"file://{html_file}", wait_until='networkidle')
        page.screenshot(path=output_png, full_page=True)
        browser.close()
        print(f"截图完成: {output_png}")

# 截图3个图表
screenshot_html(f"{output_dir}/kline_chart.html", f"{output_dir}/kline_chart.png")
screenshot_html(f"{output_dir}/volume_chart.html", f"{output_dir}/volume_chart.png")
screenshot_html(f"{output_dir}/correlation_chart.html", f"{output_dir}/correlation_chart.png")