Update app.py

app.py

@@ -10,6 +10,21 @@ import tempfile
 import time
 import os
 
+def get_full_dimensions(driver):
+    """
+    body.scrollWidth / scrollHeight と
+    documentElement.scrollWidth / scrollHeight を比較して
+    ページ全体の幅・高さを返すユーティリティ関数
+    """
+    body_w = driver.execute_script("return document.body.scrollWidth")
+    body_h = driver.execute_script("return document.body.scrollHeight")
+    html_w = driver.execute_script("return document.documentElement.scrollWidth")
+    html_h = driver.execute_script("return document.documentElement.scrollHeight")
+
+    full_width = max(body_w, html_w)
+    full_height = max(body_h, html_h)
+    return full_width, full_height
+
 def render_fullpage_screenshot(html_code):
     # HTMLコードを一時ファイルに保存
     tmp_file = tempfile.NamedTemporaryFile(suffix=".html", delete=False)
@@ -27,7 +42,7 @@ def render_fullpage_screenshot(html_code):
     try:
         driver = webdriver.Chrome(options=options)
 
-        # まずは適当なウィンドウサイズでページを読み込む
+        # 1) とりあえず適当なウィンドウサイズでページを読み込む
         driver.set_window_size(1200, 800)
         driver.get("file://" + tmp_path)
 
@@ -35,30 +50,30 @@ def render_fullpage_screenshot(html_code):
         WebDriverWait(driver, 10).until(
             EC.presence_of_element_located((By.TAG_NAME, "body"))
         )
-        # フォントや外部リソース読み込みの安定化のため少し待機
-        time.sleep(2)
+        time.sleep(1)  # 追加待機
 
-        # スクロールバーを非表示にする（必要に応じてコメントアウト）
+        # 2) スクロールバーを非表示にする（必要に応じてコメントアウト）
         driver.execute_script("""
             document.documentElement.style.overflow = 'hidden';
             document.body.style.overflow = 'hidden';
         """)
 
-        # ページ全体の幅・高さを取得
-        scroll_width = driver.execute_script("return document.body.scrollWidth")
-        scroll_height = driver.execute_script("return document.body.scrollHeight")
-
-        # ウィンドウサイズをページ全体の幅・高さに再設定
-        driver.set_window_size(scroll_width, scroll_height)
+        # 3) ページ全体のサイズを取得してウィンドウサイズを拡張
+        full_width, full_height = get_full_dimensions(driver)
+        driver.set_window_size(full_width, full_height)
+        time.sleep(2)  # レイアウト変化待ち
 
-        # レイアウト変化を待つ
-        time.sleep(2)
+        # 4) もう一度ページ全体のサイズを取得し、もし変化していれば再調整
+        new_full_width, new_full_height = get_full_dimensions(driver)
+        if (new_full_width != full_width) or (new_full_height != full_height):
+            driver.set_window_size(new_full_width, new_full_height)
+            time.sleep(2)  # もう一度待機
 
-        # ページ全体をスクリーンショット
+        # 5) ページ全体をスクリーンショット
         png = driver.get_screenshot_as_png()
 
     except Exception as e:
-        # 何らかのエラー時は1x1の黒画像を返す
+        # エラー時は1x1の黒画像を返す
         return Image.new('RGB', (1, 1), color=(0, 0, 0))
 
     finally: