Update app.py

app.py

@@ -1,40 +1,62 @@
 import gradio as gr
 from selenium import webdriver
-from selenium.common.exceptions import WebDriverException
+from selenium.webdriver.chrome.options import Options
+from selenium.webdriver.common.by import By
+from selenium.webdriver.support.ui import WebDriverWait
+from selenium.webdriver.support import expected_conditions as EC
 from PIL import Image
 from io import BytesIO
-import tempfile, os
+import tempfile
+import time
+import os
 
 def render_html_to_image(html_code):
-    # 入力HTMLを一時ファイルに保存
+    # HTMLコードを一時ファイルに保存
     tmp_file = tempfile.NamedTemporaryFile(suffix=".html", delete=False)
     tmp_path = tmp_file.name
     tmp_file.write(html_code.encode('utf-8'))
     tmp_file.close()
-    
-    # Chrome（Chromium）をヘッドレスモードで起動
-    options = webdriver.ChromeOptions()
-    options.add_argument('--headless')               # ヘッドレス（画面非表示）モード
-    options.add_argument('--no-sandbox')             # コンテナ環境でのサンドボックス無効化
-    options.add_argument('--disable-dev-shm-usage')  # /dev/shm の共有メモリ領域を不使用
+
+    # ヘッドレスChrome(Chromium)起動オプションを設定
+    options = Options()
+    options.add_argument("--headless")               # ヘッドレスモード
+    options.add_argument("--no-sandbox")             # コンテナ環境でのサンドボックス無効化
+    options.add_argument("--disable-dev-shm-usage")  # /dev/shmを使わない
+    options.add_argument("--force-device-scale-factor=1")  # スケールファクター固定
+
     try:
+        # ブラウザ起動
         driver = webdriver.Chrome(options=options)
-        driver.set_window_size(1200, 800)  # ウィンドウサイズを適宜指定
-        # ローカルHTMLファイルを開く（file:// スキームを使用）
+        # レンダリングする画面サイズを固定
+        driver.set_window_size(1200, 800)
+
+        # ローカルHTMLを開く
         driver.get("file://" + tmp_path)
-        driver.implicitly_wait(3)         # レンダリング待ち (最大3秒)
+
+        # ---- 待機処理 (1) 明示的待機で「body」タグの読み込み完了を待つ ----
+        WebDriverWait(driver, 10).until(
+            EC.presence_of_element_located((By.TAG_NAME, "body"))
+        )
+
+        # ---- 待機処理 (2) 追加のスリープでフォント等の読み込みをさらに確実化 ----
+        time.sleep(2)
+
+        # スクリーンショットをPNGバイナリとして取得
         png = driver.get_screenshot_as_png()
-    except WebDriverException as e:
-        # 失敗時は1x1の透明画像を返すなどの処理
-        return Image.new('RGB', (1, 1), color=(0,0,0))
+
+    except Exception as e:
+        # 何らかのエラーが発生した場合は1x1ピクセルの黒画像を返すなどのfallback
+        return Image.new('RGB', (1, 1), color=(0, 0, 0))
+
     finally:
         driver.quit()
         if os.path.exists(tmp_path):
             os.remove(tmp_path)
-    # PillowでPNGバイナリからImageオブジェクトを生成
+
+    # PNGバイナリをPIL.Imageに変換して返す
     return Image.open(BytesIO(png))
 
-# Gradioインタフェースの定義
+# Gradioインターフェースの定義
 iface = gr.Interface(
     fn=render_html_to_image,
     inputs=gr.Textbox(lines=15, label="HTMLコード入力"),