Update app.py

app.py

@@ -380,6 +380,94 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
     tmp_path = None # 初期化
     driver = None # 初期化
 
+    # JavaScript関数を定義 - 古い構文を使用して互換性を確保
+    resource_check_script = """
+    return (function() {
+        var callback = arguments[arguments.length - 1];
+        
+        // Font Awesomeの読み込み確認
+        function checkFontAwesome() {
+            var icons = document.querySelectorAll('.fa, .fas, .far, .fab, [class*="fa-"]');
+            if (icons.length > 0) {
+                console.log('Font Awesome icons found:', icons.length);
+                // タイムアウトベースの簡易アプローチ
+                setTimeout(function() { callback(true); }, 1500);
+            } else {
+                // アイコンがない場合
+                setTimeout(function() { callback(true); }, 800);
+            }
+        }
+        
+        // DOMContentLoadedまたはloadイベント後にチェック
+        if (document.readyState === 'complete') {
+            checkFontAwesome();
+        } else {
+            window.addEventListener('load', checkFontAwesome);
+        }
+    })();
+    """
+    
+    scroll_script = """
+    return (function() {
+        var callback = arguments[arguments.length - 1];
+        
+        var height = Math.max(
+            document.documentElement.scrollHeight,
+            document.body.scrollHeight
+        );
+        var viewportHeight = window.innerHeight;
+        
+        // ページを少しずつスクロールして全体を描画させる
+        var scrollStep = Math.floor(viewportHeight * 0.8);
+        var currentPos = 0;
+        
+        function scrollDown() {
+            if (currentPos < height) {
+                window.scrollTo(0, currentPos);
+                currentPos += scrollStep;
+                setTimeout(scrollDown, 100);
+            } else {
+                // 最後にトップに戻す
+                window.scrollTo(0, 0);
+                setTimeout(function() { callback(true); }, 300);
+            }
+        }
+        
+        scrollDown();
+    })();
+    """
+    
+    stability_script = """
+    return (function() {
+        var callback = arguments[arguments.length - 1];
+        
+        var lastHeight = document.body.offsetHeight;
+        var lastWidth = document.body.offsetWidth;
+        var stableCount = 0;
+        
+        function checkStability() {
+            var currentHeight = document.body.offsetHeight;
+            var currentWidth = document.body.offsetWidth;
+            
+            if (currentHeight === lastHeight && currentWidth === lastWidth) {
+                stableCount++;
+                if (stableCount >= 3) {  // 3回連続で同じなら安定と判断
+                    callback(true);
+                    return;
+                }
+            } else {
+                stableCount = 0;
+                lastHeight = currentHeight;
+                lastWidth = currentWidth;
+            }
+            
+            setTimeout(checkStability, 200);  // 200ms間隔でチェック
+        }
+        
+        checkStability();
+    })();
+    """
+
     # 1) Save HTML code to a temporary file
     try:
         with tempfile.NamedTemporaryFile(suffix=".html", delete=False, mode='w', encoding='utf-8') as tmp_file:
@@ -420,29 +508,42 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         )
         logger.info("Body element found. Waiting for potential resource loading...")
         
-        # リソース読み込みの待機時間
-        time.sleep(3)  # 初期待機
+        # リソース読み込みの基本的な待機時間
+        time.sleep(2)
         
-        # 5) Font Awesomeと外部リソースの読み込み完了を確認
+        # 5) Font Awesomeと外部リソースの読み込み完了を確認 - エラーハンドリング改善
+        logger.info("Waiting for Font Awesome and other resources to load...")
         try:
-            logger.info("Waiting for Font Awesome and other resources to load...")
-            driver.set_script_timeout(15)  # 15秒のタイムアウト
+            driver.set_script_timeout(10)  # タイムアウト時間を短縮
             driver.execute_async_script(resource_check_script)
-            logger.info("Resources loaded successfully")
+            logger.info("Resources loading check completed")
         except Exception as e:
-            logger.warning(f"Resource loading check failed: {e}. Using fallback wait.")
-            time.sleep(8)  # より長い待機時間を設定
+            if "not defined" in str(e):
+                # 定義エラーは発生しないはず - ログ出力のみ
+                logger.warning(f"スクリプト変数エラー: {e}")
+                time.sleep(3)
+            else:
+                # 他のエラーでも処理を続行（短いタイムアウト）
+                logger.info(f"リソースチェックをスキップ: {str(e).split('Stacktrace')[0]}")
+                time.sleep(4)
         
-        # 6) スクロールを制御してコンテンツ全体が描画されるようにする
+        # 6) コンテンツ全体を描画するためのスクロール処理
+        logger.info("Rendering full page content...")
         try:
-            logger.info("Ensuring all content is rendered...")
             driver.execute_async_script(scroll_script)
         except Exception as e:
-            logger.warning(f"Content rendering check failed: {e}")
-            # スクロールを元の位置に戻す
-            driver.execute_script("window.scrollTo(0, 0);")
-            time.sleep(1)
-            
+            # エラーログを簡素化
+            logger.info(f"スクロール処理をスキップ: {str(e).split('Stacktrace')[0]}")
+            # フォールバック: 単純なスクロール処理
+            try:
+                driver.execute_script("""
+                    window.scrollTo(0, document.body.scrollHeight);
+                    setTimeout(function() { window.scrollTo(0, 0); }, 500);
+                """)
+                time.sleep(1.5)
+            except:
+                pass
+
         # 7) Hide scrollbars via CSS
         try:
             driver.execute_script(
@@ -451,7 +552,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             )
             logger.info("Scrollbars hidden via JS.")
         except Exception as e:
-            logger.warning(f"Could not hide scrollbars via JS: {e}")
+            logger.info(f"Could not hide scrollbars: {str(e).split('Stacktrace')[0]}")
 
         # 8) Get full page dimensions accurately with improved script
         try:
@@ -499,7 +600,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             scroll_height = max(scroll_height, 100)  # 最小高さ
             
             scroll_width = min(scroll_width, 2000)  # 最大幅
-            scroll_height = min(scroll_height, 6000)  # 最大高さ
+            scroll_height = min(scroll_height, 4000)  # 最大高さ
 
         except Exception as e:
             logger.error(f"Error getting page dimensions: {e}")
@@ -508,15 +609,18 @@ 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) レイアウト安定化の確認 - 簡素化
+        logger.info("Checking layout stability...")
         try:
-            logger.info("Checking layout stability...")
+            driver.set_script_timeout(5)  # 短いタイムアウト
             driver.execute_async_script(stability_script)
-            logger.info("Layout is stable")
+            logger.info("Layout stability verified")
         except Exception as e:
-            logger.warning(f"Layout stability check failed: {e}")
-            time.sleep(2)  # 追加待機
-    
+            # エラーログを簡素化
+            logger.info(f"安定性チェックをスキップ: {str(e).split('Stacktrace')[0]}")
+            # 代わりに固定待機
+            time.sleep(2)
+
         # 10) Calculate adjusted height with user-specified margin
         adjusted_height = int(scroll_height * (1 + extension_percentage / 100.0))
         # Ensure adjusted height is not excessively large or small
@@ -530,7 +634,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         logger.info("Waiting for layout stabilization after resize...")
         
         # レイアウト安定化のための待機
-        time.sleep(4)  # 統一した待機時間
+        time.sleep(3)  # 統一した待機時間
 
         # 外部リソースの読み込み状態を確認
         try:
@@ -548,7 +652,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
                 logger.info("Document still loading, waiting additional time...")
                 time.sleep(2)
         except Exception as e:
-            logger.warning(f"Error checking resource state: {e}")
+            logger.info(f"Resource state check skipped: {str(e).split('Stacktrace')[0]}")
 
         # Scroll to top just in case
         try:
@@ -556,7 +660,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             time.sleep(1)
             logger.info("Scrolled to top.")
         except Exception as e:
-             logger.warning(f"Could not scroll to top: {e}")
+             logger.info(f"Could not scroll to top: {str(e).split('Stacktrace')[0]}")
 
         # 12) Take screenshot
         logger.info("Taking screenshot...")