Update app.py

app.py

@@ -171,8 +171,8 @@ def generate_html_from_text(text, temperature=0.3):
 - フォント指定：
 ```html
 <style>
-@ import url('https ://fonts.googleapis.com/css2?family=Kaisei+Decol&family=Yomogi&family=Zen+Kurenaido&display=swap');
-@ import url('https ://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css');
+@ import url('https://fonts.googleapis.com/css2?family=Kaisei+Decol&family=Yomogi&family=Zen+Kurenaido&display=swap');
+@ import url('https://cdnjs.cloudflare.com/ajax/libs/font-awesome/6.4.0/css/all.min.css');
 </style>
 ```
 ### 5. レイアウト
@@ -184,16 +184,6 @@ def generate_html_from_text(text, temperature=0.3):
 - 横幅は100%
 - 重要な要素は中央寄り、補足情報は周辺部に配置
 
-### 重要なレイアウト注意事項
-- Font Awesomeアイコンと文字が重ならないよう適切なマージンを設定する
-- アイコンには必ず`margin-right: 10px;`や`margin-left: 10px;`などの余白を設定する
-- アイコンと文字列は必ず`display: inline-block;`または`display: inline-flex;`で配置し、`align-items: center;`を使用する
-- 複数のアイコンを配置する場合は十分な余白を確保する
-- レイアウト崩れを防ぐために、親要素には`overflow: hidden;`を適用する
-- アイコンの配置はコンテンツフローを妨げないよう注意する
-- カード内でのアイコン配置は、絶対位置指定ではなく相対位置で設定する
-- モバイル表示も考慮し、レスポンシブ対応を徹底する
-
 ## グラフィックレコーディング表現技法
 - テキストと視覚要素のバランスを重視（文字情報の50%以上をFont Awesomeアイコンで視覚的に補完）
 - キーワードを囲み線や色で強調し、関連するFont Awesomeアイコンを必ず添える
@@ -363,8 +353,7 @@ def trim_image_whitespace(image, threshold=250, padding=10):
     logger.info(f"画像をトリミングしました: 元サイズ {width}x{height} → トリミング後 {trimmed.width}x{trimmed.height}")
     return trimmed
 
-# 非同期スクリプトを使わず、同期的なスクリプトのみ使用する改善版
-
+# --- Core Screenshot Logic ---
 def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0, 
                               trim_whitespace: bool = True) -> Image.Image:
     """
@@ -372,14 +361,14 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
 
     Args:
         html_code: The HTML source code string.
-        extension_percentage: Percentage of extra space to add vertically.
+        extension_percentage: Percentage of extra space to add vertically (e.g., 4 means 4% total).
         trim_whitespace: Whether to trim excess whitespace from the image.
 
     Returns:
-        A PIL Image object of the screenshot.
+        A PIL Image object of the screenshot. Returns a 1x1 black image on error.
     """
-    tmp_path = None
-    driver = None
+    tmp_path = None # 初期化
+    driver = None # 初期化
 
     # 1) Save HTML code to a temporary file
     try:
@@ -389,7 +378,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         logger.info(f"HTML saved to temporary file: {tmp_path}")
     except Exception as e:
         logger.error(f"Error writing temporary HTML file: {e}")
-        return Image.new('RGB', (1, 1), color=(0, 0, 0))
+        return Image.new('RGB', (1, 1), color=(0, 0, 0)) # エラー時は黒画像
 
     # 2) Headless Chrome(Chromium) options
     options = Options()
@@ -397,6 +386,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
     options.add_argument("--force-device-scale-factor=1")
+    # Font Awesomeが読み込まれない場合があるため、読み込み待機時間を長く設定
     options.add_argument("--disable-features=NetworkService")
     options.add_argument("--dns-prefetch-disable")
 
@@ -405,7 +395,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         driver = webdriver.Chrome(options=options)
         logger.info("WebDriver initialized.")
 
-        # 3) 初期ウィンドウサイズを設定
+        # 3) 初期ウィンドウサイズを設定（コンテンツの種類に関わらず同じサイズ）
         initial_width = 1200  
         initial_height = 1000
         driver.set_window_size(initial_width, initial_height)
@@ -413,58 +403,97 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         logger.info(f"Navigating to {file_url}")
         driver.get(file_url)
 
-        # 4) ページ読み込み待機
+        # 4) Wait for page load with extended timeout
         logger.info("Waiting for body element...")
         WebDriverWait(driver, 15).until(
             EC.presence_of_element_located((By.TAG_NAME, "body"))
         )
-        logger.info("Body element found. Waiting for resource loading...")
-        
-        # 5) 基本的なリソース読み込み待機 - タイムアウト回避
-        time.sleep(3)
-        
-        # Font Awesome読み込み確認 - 非同期を使わない
-        logger.info("Checking for Font Awesome resources...")
-        fa_count = driver.execute_script("""
-            var icons = document.querySelectorAll('.fa, .fas, .far, .fab, [class*="fa-"]');
-            return icons.length;
-        """)
-        logger.info(f"Found {fa_count} Font Awesome elements")
-        
-        # リソース読み込み状態を確認
-        doc_ready = driver.execute_script("return document.readyState;")
-        logger.info(f"Document ready state: {doc_ready}")
-        
-        # Font Awesomeが多い場合は追加待機
-        if fa_count > 50:
-            logger.info("Many Font Awesome icons detected, waiting additional time")
-            time.sleep(2)
+        logger.info("Body element found. Waiting for potential resource loading...")
         
-        # 6) コンテンツレンダリングのためのスクロール処理 - 同期的に実行
-        logger.info("Performing content rendering scroll...")
-        total_height = driver.execute_script("return Math.max(document.body.scrollHeight, document.documentElement.scrollHeight);")
-        viewport_height = driver.execute_script("return window.innerHeight;")
-        scrolls_needed = max(1, total_height // viewport_height)
+        # リソース読み込みの待機時間
+        time.sleep(3)  # 初期待機
         
-        for i in range(scrolls_needed + 1):
-            scroll_pos = i * (viewport_height - 200)  # オーバーラップさせる
-            driver.execute_script(f"window.scrollTo(0, {scroll_pos});")
-            time.sleep(0.2)  # 短い待機
+        # 5) Font Awesomeと外部リソースの読み込み完了を確認
+        try:
+            # 修正後のresource_check_script: arrow functionを廃止し、function式に置換
+            resource_check_script = """
+            return new Promise(function(resolve) {
+                function checkFontAwesome() {
+                    var icons = document.querySelectorAll('.fa, .fas, .far, .fab, [class*=\\"fa-\\" ]');
+                    if (icons.length > 0) {
+                        console.log('Font Awesome icons found:', icons.length);
+                        document.fonts.ready.then(function() {
+                            console.log('All fonts loaded');
+                            setTimeout(resolve, 1000);
+                        });
+                    } else {
+                        document.fonts.ready.then(function() { setTimeout(resolve, 500); });
+                    }
+                }
+                if (document.readyState === 'complete') {
+                    checkFontAwesome();
+                } else {
+                    window.addEventListener('load', checkFontAwesome);
+                }
+            });
+            """
+            
+            logger.info("Waiting for Font Awesome and other resources to load...")
+            driver.set_script_timeout(15)  # 15秒のタイムアウト
+            driver.execute_async_script("const callback = arguments[arguments.length - 1]; " + resource_check_script.strip() + ".then(callback);")
+            logger.info("Resources loaded successfully")
+        except Exception as e:
+            logger.warning(f"Resource loading check failed: {e}. Using fallback wait.")
+            time.sleep(8)  # より長い待機時間を設定
         
-        # トップに戻る
-        driver.execute_script("window.scrollTo(0, 0);")
-        time.sleep(0.5)
-        logger.info("Scroll rendering completed")
-
-        # 7) スクロールバーを非表示に
-        driver.execute_script("""
-            document.documentElement.style.overflow = 'hidden';
-            document.body.style.overflow = 'hidden';
-        """)
-        logger.info("Scrollbars hidden")
-
-        # 8) ページの寸法を取得
-        dimensions = driver.execute_script("""
+        # 6) スクロールを制御してコンテンツ全体が描画されるようにする
+        try:
+            // 修正後のscroll_script: function式に置換
+            scroll_script = """
+            return new Promise(function(resolve) {
+                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(resolve, 300);
+                    }
+                }
+                scrollDown();
+            });
+            """
+            
+            logger.info("Ensuring all content is rendered...")
+            driver.execute_async_script("const callback = arguments[arguments.length - 1]; " + scroll_script.strip() + ".then(callback);")
+        except Exception as e:
+            logger.warning(f"Content rendering check failed: {e}")
+            # スクロールを元の位置に戻す
+            driver.execute_script("window.scrollTo(0, 0);")
+            time.sleep(1)
+
+        # 7) Hide scrollbars via CSS
+        try:
+            driver.execute_script(
+                "document.documentElement.style.overflow = 'hidden';"
+                "document.body.style.overflow = 'hidden';"
+            )
+            logger.info("Scrollbars hidden via JS.")
+        except Exception as e:
+            logger.warning(f"Could not hide scrollbars via JS: {e}")
+
+        # 8) Get full page dimensions accurately with improved script
+        try:
+            // より正確なページ寸法を取得するためのJavaScriptコード
+            dimensions_script = """
             return {
                 width: Math.max(
                     document.documentElement.scrollWidth,
@@ -483,80 +512,137 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
                     document.body ? document.body.clientHeight : 0
                 )
             };
-        """)
-        scroll_width = dimensions['width']
-        scroll_height = dimensions['height']
-        logger.info(f"Detected dimensions: width={scroll_width}, height={scroll_height}")
-        
-        # 再検証 - 短いスクロールで再確認
-        driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
-        time.sleep(0.5)
-        driver.execute_script("window.scrollTo(0, 0);")
-        time.sleep(0.5)
-        
-        dimensions_after = driver.execute_script("return {height: Math.max(document.documentElement.scrollHeight, document.body.scrollHeight)};")
-        scroll_height = max(scroll_height, dimensions_after['height'])
-        logger.info(f"After scroll check, height={scroll_height}")
-        
-        # 最小/最大値の設定
-        scroll_width = max(scroll_width, 100)
-        scroll_height = max(scroll_height, 100)
-        scroll_width = min(scroll_width, 2000)
-        scroll_height = min(scroll_height, 4000)
-        
-        # 9) レイアウト安定化のための単純な待機 - タイムアウト回避
-        logger.info("Waiting for layout stabilization...")
-        time.sleep(2)
+            """
+            dimensions = driver.execute_script(dimensions_script)
+            scroll_width = dimensions['width']
+            scroll_height = dimensions['height']
+            
+            logger.info(f"Detected dimensions: width={scroll_width}, height={scroll_height}")
+            
+            # スクロールして確認する追加の検証
+            driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
+            time.sleep(1)  # スクロール完了を待つ
+            driver.execute_script("window.scrollTo(0, 0);")
+            time.sleep(1)  # 元の位置に戻す
+            
+            # 再検証
+            dimensions_after_scroll = driver.execute_script(dimensions_script)
+            scroll_height = max(scroll_height, dimensions_after_scroll['height'])
+            
+            logger.info(f"After scroll check, height={scroll_height}")
+            
+            # 最小値と最大値の設定
+            scroll_width = max(scroll_width, 100)  # 最小幅
+            scroll_height = max(scroll_height, 100)  # 最小高さ
+            
+            scroll_width = min(scroll_width, 2000)  # 最大幅
+            scroll_height = min(scroll_height, 6000)  # 最大高さ
+
+        except Exception as e:
+            logger.error(f"Error getting page dimensions: {e}")
+            # フォールバックとしてデフォルト値を設定
+            scroll_width = 1200
+            scroll_height = 1000
+            logger.warning(f"Falling back to dimensions: width={scroll_width}, height={scroll_height}")
         
-        # 10) 高さに余白を追加
+        # 9) レイアウト安定化の確認
+        try:
+            // 修正後のstability_script: arrow functionをfunction式に置換
+            stability_script = """
+            return new Promise(function(resolve) {
+                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) {
+                            resolve(true);
+                            return;
+                        }
+                    } else {
+                        stableCount = 0;
+                        lastHeight = currentHeight;
+                        lastWidth = currentWidth;
+                    }
+                    setTimeout(checkStability, 200);
+                }
+                checkStability();
+            });
+            """
+            
+            logger.info("Checking layout stability...")
+            driver.execute_async_script("const callback = arguments[arguments.length - 1]; " + stability_script.strip() + ".then(callback);")
+            logger.info("Layout is stable")
+        except Exception as e:
+            logger.warning(f"Layout stability check failed: {e}")
+            time.sleep(2)  # 追加待機
+
+        # 10) Calculate adjusted height with user-specified margin
         adjusted_height = int(scroll_height * (1 + extension_percentage / 100.0))
-        adjusted_height = max(adjusted_height, scroll_height, 100)
+        # Ensure adjusted height is not excessively large or small
+        adjusted_height = max(adjusted_height, scroll_height, 100)  # 最小高さを確保
         logger.info(f"Adjusted height calculated: {adjusted_height} (extension: {extension_percentage}%)")
 
-        # 11) ウィンドウサイズを調整
+        # 11) Set window size to full page dimensions
         adjusted_width = scroll_width
         logger.info(f"Resizing window to: width={adjusted_width}, height={adjusted_height}")
         driver.set_window_size(adjusted_width, adjusted_height)
-        time.sleep(1)
+        logger.info("Waiting for layout stabilization after resize...")
         
-        # リソース状態を確認 - 同期的スクリプト
-        resource_state = driver.execute_script("""
+        # レイアウト安定化のための待機
+        time.sleep(4)  # 統一した待機時間
+
+        # 外部リソースの読み込み状態を確認
+        try:
+            resource_state = driver.execute_script("""
             return {
                 readyState: document.readyState,
                 resourcesComplete: !document.querySelector('img:not([complete])') && 
-                                  !document.querySelector('link[rel="stylesheet"]:not([loaded])')
+                                   !document.querySelector('link[rel="stylesheet"]:not([loaded])')
             };
-        """)
-        logger.info(f"Resource state: {resource_state}")
-        
-        if resource_state['readyState'] != 'complete':
-            logger.info("Document still loading, waiting additional time...")
+            """)
+            logger.info(f"Resource state: {resource_state}")
+            
+            # ドキュメントの読み込みが完了していない場合、追加で待機
+            if resource_state['readyState'] != 'complete':
+                logger.info("Document still loading, waiting additional time...")
+                time.sleep(2)
+        except Exception as e:
+            logger.warning(f"Error checking resource state: {e}")
+
+        # Scroll to top just in case
+        try:
+            driver.execute_script("window.scrollTo(0, 0)")
             time.sleep(1)
-        
-        # トップにスクロール
-        driver.execute_script("window.scrollTo(0, 0);")
-        time.sleep(0.5)
-        logger.info("Scrolled to top.")
+            logger.info("Scrolled to top.")
+        except Exception as e:
+             logger.warning(f"Could not scroll to top: {e}")
 
-        # 12) スクリーンショット取得
+        # 12) Take screenshot
         logger.info("Taking screenshot...")
         png = driver.get_screenshot_as_png()
         logger.info("Screenshot taken successfully.")
 
-        # PIL画像に変換
+        # Convert to PIL Image
         img = Image.open(BytesIO(png))
+        
+        # 画像サイズの確認とログ
         logger.info(f"Screenshot dimensions: {img.width}x{img.height}")
         
-        # 余白トリミング
+        # 余白トリミングが有効な場合
         if trim_whitespace:
+            # 余分な空白をトリミング
             img = trim_image_whitespace(img, threshold=248, padding=20)
             logger.info(f"Trimmed dimensions: {img.width}x{img.height}")
         
         return img
 
     except Exception as e:
-        logger.error(f"Error during screenshot generation: {e}")
-        return Image.new('RGB', (1, 1), color=(0, 0, 0))
+        logger.error(f"An error occurred during screenshot generation: {e}", exc_info=True)
+        return Image.new('RGB', (1, 1), color=(0, 0, 0)) # Return black 1x1 image on error
     finally:
         logger.info("Cleaning up...")
         if driver:
@@ -571,7 +657,7 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
                 logger.info(f"Temporary file {tmp_path} removed.")
             except Exception as e:
                 logger.error(f"Error removing temporary file {tmp_path}: {e}")
-                
+
 # --- Geminiを使った新しい関数 ---
 def text_to_screenshot(text: str, extension_percentage: float, temperature: float = 0.3, trim_whitespace: bool = True) -> Image.Image:
     """テキストをGemini APIでHTMLに変換し、スクリーンショットを生成する統合関数"""
@@ -795,4 +881,4 @@ app = gr.mount_gradio_app(app, iface, path="/")
 if __name__ == "__main__":
     import uvicorn
     logger.info("Starting Uvicorn server for local development...")
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    uvicorn.run(app, host="0.0.0.0", port=7860)