Update app.py

app.py

@@ -28,10 +28,8 @@ def scroll_to_bottom(driver):
     一旦ページの最下部までスクロールし、
     Lazy Loadなどで追加要素が生成されるのを待つ
     """
-    # 現在の高さ
     last_height = driver.execute_script("return document.body.scrollHeight")
     while True:
-        # 最下部までスクロール
         driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
         time.sleep(1)  # スクロール後の描画待ち
         new_height = driver.execute_script("return document.body.scrollHeight")
@@ -65,36 +63,41 @@ def render_fullpage_screenshot(html_code):
         )
         time.sleep(1)
 
-        # 2) ページ全体サイズを取得してウィンドウを拡張
-        #    → Lazy Loadがあればスクロールして要素を読み込む
-        max_iterations = 3
-        for _ in range(max_iterations):
-            # ページ全体の幅・高さを取得
-            w1, h1 = get_full_dimensions(driver)
-            # ウィンドウをそのサイズに合わせる
-            driver.set_window_size(w1, h1)
+        # ---- (A) 一度ページ全体のサイズを取得してウィンドウを拡張 ----
+        w1, h1 = get_full_dimensions(driver)
+        driver.set_window_size(w1, h1)
+        time.sleep(1)
+
+        # ---- (B) ページ下端までスクロールしてLazy Load等を発火させる ----
+        scroll_to_bottom(driver)
+        time.sleep(1)
+
+        # 再度サイズを取得して変化があれば再設定
+        w2, h2 = get_full_dimensions(driver)
+        if (w2 != w1) or (h2 != h1):
+            driver.set_window_size(w2, h2)
             time.sleep(1)
-            # 一度最下部までスクロールし、Lazy Load等を促す
+            # 念のためもう一度最下部へ
             scroll_to_bottom(driver)
-            # 再度サイズを取得
-            w2, h2 = get_full_dimensions(driver)
-            # 変化がなければ打ち切り
-            if (w1 == w2) and (h1 == h2):
-                break
+            time.sleep(1)
 
-        # 3) 最後にスクロールバーを消したい場合はここでoverflow: hidden
+        # ---- (C) スクロールバーを非表示にする（不要ならコメントアウト）----
         driver.execute_script("""
             document.documentElement.style.overflow = 'hidden';
             document.body.style.overflow = 'hidden';
         """)
         time.sleep(1)
 
-        # overflow: hidden による再レイアウトを考慮してもう一度調整
+        # overflow: hidden による再レイアウトでサイズが変わる可能性がある
         w3, h3 = get_full_dimensions(driver)
         driver.set_window_size(w3, h3)
         time.sleep(1)
 
-        # 4) 最終的にスクリーンショットを撮る
+        # ---- (D) ★最終的にトップへ戻す★ ----
+        driver.execute_script("window.scrollTo(0, 0);")
+        time.sleep(1)
+
+        # ---- (E) スクリーンショット取得 ----
         png = driver.get_screenshot_as_png()
 
     except Exception as e: