From 2064c38ec9a8a7dc392513ae9910b0f08a9b6bc6 Mon Sep 17 00:00:00 2001 From: Archie Date: Wed, 9 Apr 2025 17:24:20 +0800 Subject: [PATCH] Update selenium_converter.py --- dataframe_image/converter/browser/selenium_converter.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/dataframe_image/converter/browser/selenium_converter.py b/dataframe_image/converter/browser/selenium_converter.py index 0f6effc..8d35871 100644 --- a/dataframe_image/converter/browser/selenium_converter.py +++ b/dataframe_image/converter/browser/selenium_converter.py @@ -54,6 +54,13 @@ def screenshot(self, html: str) -> Image: # temp_img will be deleted after context exit img = Image.open(temp_img) + + # Retry with larger height if the first screenshot is too short + if img.height < required_height: + shortfall = required_height - img.height +200 + driver.set_window_size(required_width + 150, required_height + 90 + shortfall) + driver.save_screenshot(str(temp_img)) + img = Image.open(temp_img) try: temp_dir_obj.cleanup() except OSError: