如何使用Java和Selenium将HTML页面转换为图片并解决内容丢失问题？

import org.openqa.selenium.*;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class HtmlToImage_Selenium {

    public static void main(String[] args) {

        // 路径设置 (请替换为您的实际路径)
        String chromePath = "D:\htmlToImg\Selenium\chromedriver-win64\chromedriver.exe";
        String mergePath = "D:\htmlToImg\Selenium\output\merge.png";
        String tempPath = "D:\htmlToImg\Selenium\output\screenshot_";

        System.setProperty("webdriver.chrome.driver", chromePath);

        ChromeOptions options = new ChromeOptions();
        options.addArguments("--headless"); // 无头模式
        options.addArguments("--window-size=2160,1440"); // 窗口大小

        WebDriver driver = new ChromeDriver(options);

        try {
            driver.get("file:///D:/htmlToImg/Test.html");

            WebDriverWait wait = new WebDriverWait(driver, 15);
            wait.until(ExpectedConditions.visibilityOfElementLocated(By.tagName("body")));

            JavascriptExecutor js = (JavascriptExecutor) driver;
            long pageHeight = (long) js.executeScript("return document.body.scrollHeight;");
            System.out.println("页面高度为" + pageHeight);

            int captureHeight = 5000; // 每次截取的高度
            int overlapHeight = 2000; // 重叠高度
            int numberOfScreenshots = (int) Math.ceil((double) pageHeight / (captureHeight - overlapHeight));

            List<BufferedImage> images = new ArrayList<>();
            for (int i = 0; i < numberOfScreenshots; i++) {
                long scrollPosition = i * (captureHeight - overlapHeight);
                js.executeScript("window.scrollTo(0," + scrollPosition + ")");

                // 添加等待，确保页面滚动完成
                wait.until(ExpectedConditions.stalenessOf(driver.findElement(By.tagName("body"))));

                File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
                BufferedImage image = ImageIO.read(screenshot);
                images.add(image);
            }

            // (此处需要添加图像拼接代码，根据实际情况选择合适的图像拼接库)
            // ...  图像拼接逻辑 ...

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            driver.quit();
        }
    }
}