Update app.py

app.py

@@ -564,43 +564,91 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             scroll_height = 1000
             logger.warning(f"Falling back to dimensions: width={scroll_width}, height={scroll_height}")
         
-        # 9) レイアウト安定化の確認
+        # 9) レイアウト安定化の確認 (修正版)
         try:
             stability_script = """
-            return new Promise(resolve => {
-                let lastHeight = document.body.offsetHeight;
-                let lastWidth = document.body.offsetWidth;
+            return new Promise((resolve, reject) => {
+                let lastHeight = 0;
+                let lastWidth = 0;
                 let stableCount = 0;
-                
+                const maxChecks = 15; // 最大チェック回数（無限ループ防止）
+                let checkCount = 0;
+
                 const checkStability = () => {
+                    checkCount++;
+                    if (checkCount > maxChecks) {
+                        console.warn('Layout stability check reached max attempts.');
+                        resolve(false); // 安定しなかったと判断
+                        return;
+                    }
+
+                    // body要素が存在し、offsetHeight/offsetWidthが利用可能かチェック
+                    const bodyExists = document.body && typeof document.body.offsetHeight === 'number' && typeof document.body.offsetWidth === 'number';
+
+                    if (!bodyExists) {
+                        // bodyがまだ利用できない場合、少し待って再試行
+                        console.warn('Document body not fully available yet for stability check.');
+                        setTimeout(checkStability, 250); // 待機時間を少し長くする
+                        return;
+                    }
+
                     const currentHeight = document.body.offsetHeight;
                     const currentWidth = document.body.offsetWidth;
-                    
+
+                    // サイズが0の場合も不安定とみなす（初期化中など）
+                    if (currentHeight === 0 || currentWidth === 0) {
+                         stableCount = 0; // カウントリセット
+                         lastHeight = 0; // 前回値もリセット
+                         lastWidth = 0;
+                         console.warn('Body dimensions are zero, waiting...');
+                         setTimeout(checkStability, 250); // 再試行
+                         return;
+                    }
+
                     if (currentHeight === lastHeight && currentWidth === lastWidth) {
                         stableCount++;
                         if (stableCount >= 3) {  // 3回連続で同じなら安定と判断
-                            resolve(true);
+                            console.log('Layout deemed stable.');
+                            resolve(true); // 安定した
                             return;
                         }
                     } else {
-                        stableCount = 0;
+                        stableCount = 0; // サイズが変わったらリセット
                         lastHeight = currentHeight;
                         lastWidth = currentWidth;
+                        console.log(`Layout changed: ${lastWidth}x${lastHeight}. Resetting stability count.`);
                     }
-                    
-                    setTimeout(checkStability, 200);  // 200ms間隔でチェック
+
+                    // 次のチェックをスケジュール
+                    setTimeout(checkStability, 200);
                 };
-                
-                checkStability();
+
+                // 初回チェックを開始
+                // 少し遅延させて初回実行（DOMが完全に準備されるのを待つ）
+                setTimeout(checkStability, 100);
             });
             """
-            
+
             logger.info("Checking layout stability...")
-            driver.execute_async_script(f"const callback = arguments[arguments.length - 1]; {stability_script}.then(callback);")
-            logger.info("Layout is stable")
+            # 非同期スクリプトのタイムアウトを設定 (例: 20秒)
+            driver.set_script_timeout(20)
+            stable = driver.execute_async_script(f"const callback = arguments[arguments.length - 1]; {stability_script}.then(callback);")
+
+            if stable:
+                logger.info("Layout is stable")
+            else:
+                # 安定しなかった場合も警告を出す
+                logger.warning("Layout did not stabilize within the expected time. Proceeding anyway.")
+                time.sleep(2) # 念のため少し待つ
+
+        except TimeoutException:
+            # スクリプト実行がタイムアウトした場合
+            logger.warning("Layout stability check timed out. Proceeding anyway.")
+            time.sleep(2) # 念のため少し待つ
         except Exception as e:
-            logger.warning(f"Layout stability check failed: {e}")
-            time.sleep(2)  # 追加待機
+            # その他の予期せぬエラー（元のエラーが発生していた箇所）
+            logger.error(f"Error during layout stability check: {e}", exc_info=True) # スタックトレース付きでエラー出力
+            time.sleep(2)  # エラー時も待機して続行
 
         # 10) Calculate adjusted height with user-specified margin
         adjusted_height = int(scroll_height * (1 + extension_percentage / 100.0))