HTMLviewer2_API

Paused

App Files Files Community

tomo2chin2 commited on Mar 28, 2025

Commit

2a69a40

verified ·

1 Parent(s): 38f7de3

Update app.py

Browse files

Files changed (1) hide show

app.py +254 -417

app.py CHANGED Viewed

@@ -1,3 +1,4 @@
 import gradio as gr
 from fastapi import FastAPI, HTTPException, Body
 from fastapi.responses import StreamingResponse
@@ -9,6 +10,8 @@ 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
@@ -49,40 +52,40 @@ def enhance_font_awesome_layout(html_code):
         margin-right: 8px !important;
         vertical-align: middle !important;
     }
     /* テキスト内のアイコン位置調整 */
-    h1 [class*="fa-"], h2 [class*="fa-"], h3 [class*="fa-"],
     h4 [class*="fa-"], h5 [class*="fa-"], h6 [class*="fa-"] {
         vertical-align: middle !important;
         margin-right: 10px !important;
     }
     /* 特定パターンの修正 */
     .fa + span, .fas + span, .far + span, .fab + span,
     span + .fa, span + .fas, span + .far, span + .fab {
         display: inline-block !important;
         margin-left: 5px !important;
     }
     /* カード内アイコン修正 */
     .card [class*="fa-"], .card-body [class*="fa-"] {
         float: none !important;
         clear: none !important;
         position: relative !important;
     }
     /* アイコンと文字が重なる場合の調整 */
     li [class*="fa-"], p [class*="fa-"] {
         margin-right: 10px !important;
     }
     /* インラインアイコンのスペーシング */
     .inline-icon {
         display: inline-flex !important;
         align-items: center !important;
         justify-content: flex-start !important;
     }
     /* アイコン後のテキスト */
     [class*="fa-"] + span {
         display: inline-block !important;
@@ -90,7 +93,7 @@ def enhance_font_awesome_layout(html_code):
     }
     </style>
     """
     # headタグがある場合はその中に追加
     if '<head>' in html_code:
         return html_code.replace('</head>', f'{fa_fix_css}</head>')
@@ -103,7 +106,7 @@ def enhance_font_awesome_layout(html_code):
             body_start = html_code.find('<body')
             if body_start > 0:
                 return html_code[:body_start] + f'<head>{fa_fix_css}</head>' + html_code[body_start:]
     # どちらもない場合は先頭に追加
     return f'<html><head>{fa_fix_css}</head>' + html_code + '</html>'
@@ -115,15 +118,15 @@ def generate_html_from_text(text, temperature=0.3):
         if not api_key:
             logger.error("GEMINI_API_KEY 環境変数が設定されていません")
             raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
         # モデル名の取得（環境変数から、なければデフォルト値）
         model_name = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
         logger.info(f"使用するGeminiモデル: {model_name}")
         # Gemini APIの設定
         genai.configure(api_key=api_key)
-        # システムプロンプト（リクエスト例と同じものを使用）
         system_instruction = """# グラフィックレコーディング風インフォグラフィック変換プロンプト V2
 ## 目的
 以下の内容を、超一流デザイナーが作成したような、日本語で完璧なグラフィックレコーディング風のHTMLインフォグラフィックに変換してください。情報設計とビジュアルデザインの両面で最高水準を目指します。
@@ -231,156 +234,102 @@ def generate_html_from_text(text, temperature=0.3):
 - フッターに出典情報と関連するFont Awesomeアイコン（fa-book, fa-citation等）を明記
 ## 変換する文章/記事
 ーーー＜ユーザーが入力（または添付）＞ーーー"""
         # モデルを初期化して処理
         logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}")
         # モデル初期化とフォールバック処理
         try:
             model = genai.GenerativeModel(model_name)
         except Exception as e:
-            # 指定されたモデルが使用できない場合はフォールバック
             fallback_model = "gemini-pro"
             logger.warning(f"{model_name}が利用できません: {e}, フォールバックモデル{fallback_model}を使用します")
             model = genai.GenerativeModel(fallback_model)
-        # 生成設定 - ばらつきを減らすために設定を調整
         generation_config = {
-            "temperature": temperature,  # より低い温度を設定
-            "top_p": 0.7,                # 0.95から0.7に下げて出力の多様性を制限
-            "top_k": 20,                 # 64から20に下げて候補を絞る
             "max_output_tokens": 8192,
-            "candidate_count": 1        # 候補は1つだけ生成
         }
-        # 安全設定 - デフォルトの安全設定を使用
         safety_settings = [
-            {
-                "category": "HARM_CATEGORY_HARASSMENT",
-                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-            },
-            {
-                "category": "HARM_CATEGORY_HATE_SPEECH",
-                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-            },
-            {
-                "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-            },
-            {
-                "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-                "threshold": "BLOCK_MEDIUM_AND_ABOVE"
-            }
         ]
         # プロンプト構築
         prompt = f"{system_instruction}\n\n{text}"
         # コンテンツ生成
         response = model.generate_content(
             prompt,
             generation_config=generation_config,
             safety_settings=safety_settings
         )
-        # レスポンスからHTMLを抽出
         raw_response = response.text
-        # HTMLタグ部分だけを抽出（```html と ``` の間）
         html_start = raw_response.find("```html")
         html_end = raw_response.rfind("```")
         if html_start != -1 and html_end != -1 and html_start < html_end:
-            html_start += 7  # "```html" の長さ分進める
             html_code = raw_response[html_start:html_end].strip()
             logger.info(f"HTMLの生成に成功: 長さ = {len(html_code)}")
-            # Font Awesomeのレイアウト改善
             html_code = enhance_font_awesome_layout(html_code)
             logger.info("Font Awesomeレイアウトの最適化を適用しました")
             return html_code
         else:
-            # HTMLタグが見つからない場合、レスポンス全体を返す
             logger.warning("レスポンスから```html```タグが見つかりませんでした。全テキストを返します。")
             return raw_response
     except Exception as e:
         logger.error(f"HTML生成中にエラーが発生: {e}", exc_info=True)
         raise Exception(f"Gemini APIでのHTML生成に失敗しました: {e}")
-# 画像から余分な空白領域をトリミングする関数
 def trim_image_whitespace(image, threshold=250, padding=10):
-    """
-    画像から余分な白い空白をトリミングする
-    Args:
-        image: PIL.Image - 入力画像
-        threshold: int - どの明るさ以上を空白と判断するか (0-255)
-        padding: int - トリミング後に残す余白のピクセル数
-    Returns:
-        トリミングされたPIL.Image
-    """
-    # グレースケールに変換
     gray = image.convert('L')
-    # ピクセルデータを配列として取得
     data = gray.getdata()
     width, height = gray.size
-    # 有効範囲を見つける
     min_x, min_y = width, height
     max_x = max_y = 0
-    # ピクセルデータを2次元配列に変換して処理
     pixels = list(data)
     pixels = [pixels[i * width:(i + 1) * width] for i in range(height)]
-    # 各行をスキャンして非空白ピクセルを見つける
     for y in range(height):
         for x in range(width):
-            if pixels[y][x] < threshold:  # 非空白ピクセル
                 min_x = min(min_x, x)
                 min_y = min(min_y, y)
                 max_x = max(max_x, x)
                 max_y = max(max_y, y)
-    # 境界外のトリミングの場合はエラー
     if min_x > max_x or min_y > max_y:
         logger.warning("トリミング領域が見つかりません。元の画像を返します。")
         return image
-    # パディングを追加
     min_x = max(0, min_x - padding)
     min_y = max(0, min_y - padding)
     max_x = min(width - 1, max_x + padding)
     max_y = min(height - 1, max_y + padding)
-    # 画像をトリミング
     trimmed = image.crop((min_x, min_y, max_x + 1, max_y + 1))
     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:
     """
     Renders HTML code to a full-page screenshot using Selenium.
-    Args:
-        html_code: The HTML source code string.
-        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. Returns a 1x1 black image on error.
     """
-    tmp_path = None # 初期化
-    driver = None # 初期化
-    # 1) Save HTML code to a temporary file
     try:
         with tempfile.NamedTemporaryFile(suffix=".html", delete=False, mode='w', encoding='utf-8') as tmp_file:
             tmp_path = tmp_file.name
@@ -388,15 +337,14 @@ 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)) # エラー時は黒画像
-    # 2) Headless Chrome(Chromium) options
     options = Options()
     options.add_argument("--headless")
     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,101 +353,117 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         driver = webdriver.Chrome(options=options)
         logger.info("WebDriver initialized.")
-        # 3) 初期ウィンドウサイズを設定（コンテンツの種類に関わらず同じサイズ）
-        initial_width = 1200
         initial_height = 1000
         driver.set_window_size(initial_width, initial_height)
         file_url = "file://" + tmp_path
         logger.info(f"Navigating to {file_url}")
         driver.get(file_url)
-        # 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 potential resource loading...")
-        # リソース読み込みの待機時間
-        time.sleep(3)  # 初期待機
-        # 5) Font Awesomeと外部リソースの読み込み完了を確認
         try:
-            # Font Awesomeと外部リソースの読み込み完了を確認するスクリプト
             resource_check_script = """
-            return new Promise((resolve) => {
-                // Font Awesomeの読み込み確認
-                const checkFontAwesome = () => {
-                    const 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(() => {
-                            console.log('All fonts loaded');
-                            setTimeout(resolve, 1000); // 安定化のための追加待機
-                        });
                     } else {
-                        // アイコンがない、またはすでに読み込み完了
-                        document.fonts.ready.then(() => setTimeout(resolve, 500));
                     }
-                };
-                // DOMContentLoadedまたはloadイベント後にチェック
-                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(f"const callback = arguments[arguments.length - 1]; {resource_check_script}.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)  # より長い待機時間を設定
-        # 6) スクロールを制御してコンテンツ全体が描画されるようにする
         try:
             scroll_script = """
-            return new Promise(resolve => {
-                const height = Math.max(
-                    document.documentElement.scrollHeight,
-                    document.body.scrollHeight
-                );
-                const viewportHeight = window.innerHeight;
-                // ページを少しずつスクロールして全体を描画させる
-                const scrollStep = Math.floor(viewportHeight * 0.8);
-                let currentPos = 0;
-                const 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(f"const callback = arguments[arguments.length - 1]; {scroll_script}.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';"
@@ -509,181 +473,164 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         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,
-                    document.documentElement.offsetWidth,
-                    document.documentElement.clientWidth,
-                    document.body ? document.body.scrollWidth : 0,
-                    document.body ? document.body.offsetWidth : 0,
-                    document.body ? document.body.clientWidth : 0
                 ),
                 height: Math.max(
-                    document.documentElement.scrollHeight,
-                    document.documentElement.offsetHeight,
-                    document.documentElement.clientHeight,
-                    document.body ? document.body.scrollHeight : 0,
-                    document.body ? document.body.offsetHeight : 0,
-                    document.body ? document.body.clientHeight : 0
                 )
             };
             """
             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, 4000)  # 最大高さ
         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}")
-        # 9) レイアウト安定化の確認 (修正版)
         try:
             stability_script = """
-            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回連続で同じなら安定と判断
-                            console.log('Layout deemed stable.');
-                            resolve(true); // 安定した
                             return;
                         }
-                    } else {
-                        stableCount = 0; // サイズが変わったらリセット
-                        lastHeight = currentHeight;
-                        lastWidth = currentWidth;
-                        console.log(`Layout changed: ${lastWidth}x${lastHeight}. Resetting stability count.`);
-                    }
-                    // 次のチェックをスケジュール
-                    setTimeout(checkStability, 200);
-                };
-                // 初回チェックを開始
-                // 少し遅延させて初回実行（DOMが完全に準備されるのを待つ）
-                setTimeout(checkStability, 100);
-            });
-            """
             logger.info("Checking layout stability...")
-            # 非同期スクリプトのタイムアウトを設定 (例: 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.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))
-        # 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) 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)
         logger.info("Waiting for layout stabilization after resize...")
-        # レイアウト安定化のための待機
-        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])')
             };
             """)
             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)
@@ -691,28 +638,20 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         except Exception as e:
              logger.warning(f"Could not scroll to top: {e}")
-        # 12) Take screenshot
         logger.info("Taking screenshot...")
         png = driver.get_screenshot_as_png()
         logger.info("Screenshot taken successfully.")
-        # 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"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:
@@ -728,23 +667,17 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             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に変換し、スクリーンショットを生成する統合関数"""
     try:
-        # 1. テキストからHTMLを生成（温度パラメータも渡す）
         html_code = generate_html_from_text(text, temperature)
-        # 2. HTMLからスクリーンショットを生成
         return render_fullpage_screenshot(html_code, extension_percentage, trim_whitespace)
     except Exception as e:
         logger.error(f"テキストからスクリーンショット生成中にエラーが発生: {e}", exc_info=True)
-        return Image.new('RGB', (1, 1), color=(0, 0, 0))  # エラー時は黒画像
-# --- FastAPI Setup ---
 app = FastAPI()
-# CORS設定を追加
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -752,203 +685,107 @@ app.add_middleware(
     allow_methods=["*"],
     allow_headers=["*"],
 )
-# 静的ファイルのサービング設定
-# Gradioのディレクトリを探索してアセットを見つける
 gradio_dir = os.path.dirname(gr.__file__)
 logger.info(f"Gradio version: {gr.__version__}")
 logger.info(f"Gradio directory: {gradio_dir}")
-# 基本的な静的ファイルディレクトリをマウント
 static_dir = os.path.join(gradio_dir, "templates", "frontend", "static")
 if os.path.exists(static_dir):
     logger.info(f"Mounting static directory: {static_dir}")
     app.mount("/static", StaticFiles(directory=static_dir), name="static")
-# _appディレクトリを探す（新しいSvelteKitベースのフロントエンド用）
 app_dir = os.path.join(gradio_dir, "templates", "frontend", "_app")
 if os.path.exists(app_dir):
     logger.info(f"Mounting _app directory: {app_dir}")
     app.mount("/_app", StaticFiles(directory=app_dir), name="_app")
-# assetsディレクトリを探す
 assets_dir = os.path.join(gradio_dir, "templates", "frontend", "assets")
 if os.path.exists(assets_dir):
     logger.info(f"Mounting assets directory: {assets_dir}")
     app.mount("/assets", StaticFiles(directory=assets_dir), name="assets")
-# cdnディレクトリがあれば追加
 cdn_dir = os.path.join(gradio_dir, "templates", "cdn")
 if os.path.exists(cdn_dir):
     logger.info(f"Mounting cdn directory: {cdn_dir}")
     app.mount("/cdn", StaticFiles(directory=cdn_dir), name="cdn")
-# API Endpoint for screenshot generation
-@app.post("/api/screenshot",
-          response_class=StreamingResponse,
-          tags=["Screenshot"],
-          summary="Render HTML to Full Page Screenshot",
-          description="Takes HTML code and an optional vertical extension percentage, renders it using a headless browser, and returns the full-page screenshot as a PNG image.")
 async def api_render_screenshot(request: ScreenshotRequest):
-    """
-    API endpoint to render HTML and return a screenshot.
-    """
     try:
         logger.info(f"API request received. Extension: {request.extension_percentage}%")
-        # Run the blocking Selenium code in a separate thread (FastAPI handles this)
         pil_image = render_fullpage_screenshot(
             request.html_code,
             request.extension_percentage,
             request.trim_whitespace
         )
         if pil_image.size == (1, 1):
              logger.error("Screenshot generation failed, returning 1x1 image.")
-             # Optionally return a proper error response instead of 1x1 image
-             # raise HTTPException(status_code=500, detail="Failed to generate screenshot")
-        # Convert PIL Image to PNG bytes
         img_byte_arr = BytesIO()
         pil_image.save(img_byte_arr, format='PNG')
-        img_byte_arr.seek(0)  # Go to the start of the BytesIO buffer
         logger.info("Returning screenshot as PNG stream.")
         return StreamingResponse(img_byte_arr, media_type="image/png")
     except Exception as e:
         logger.error(f"API Error: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Internal Server Error: {e}")
-# --- 新しいGemini API連携エンドポイント ---
-@app.post("/api/text-to-screenshot",
-          response_class=StreamingResponse,
-          tags=["Screenshot", "Gemini"],
-          summary="テキストからインフォグラフィックを生成",
-          description="テキストをGemini APIを使ってHTMLインフォグラフィックに変換し、スクリーンショットとして返します。")
 async def api_text_to_screenshot(request: GeminiRequest):
-    """
-    テキストからHTMLインフォグラフィックを生成してスクリーンショットを返すAPIエンドポイント
-    """
     try:
         logger.info(f"テキスト→スクリーンショットAPIリクエスト受信。テキスト長さ: {len(request.text)}, 拡張率: {request.extension_percentage}%, 温度: {request.temperature}")
-        # テキストからHTMLを生成してスクリーンショットを作成（温度パラメータも渡す）
         pil_image = text_to_screenshot(
             request.text,
             request.extension_percentage,
             request.temperature,
             request.trim_whitespace
         )
         if pil_image.size == (1, 1):
             logger.error("スクリーンショット生成に失敗しました。1x1画像を返します。")
-            # raise HTTPException(status_code=500, detail="スクリーンショット生成に失敗しました")
-        # PIL画像をPNGバイトに変換
         img_byte_arr = BytesIO()
         pil_image.save(img_byte_arr, format='PNG')
-        img_byte_arr.seek(0)  # BytesIOバッファの先頭に戻る
         logger.info("スクリーンショットをPNGストリームとして返します。")
         return StreamingResponse(img_byte_arr, media_type="image/png")
     except Exception as e:
         logger.error(f"API Error: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Internal Server Error: {e}")
-# --- Gradio Interface Definition ---
-# 入力モードの選択用Radioコンポーネント
 def process_input(input_mode, input_text, extension_percentage, temperature, trim_whitespace):
-    """入力モードに応じて適切な処理を行う"""
     if input_mode == "HTML入力":
-        # HTMLモードの場合は既存の処理
         return render_fullpage_screenshot(input_text, extension_percentage, trim_whitespace)
     else:
-        # テキスト入力モードの場合はGemini APIを使用
         return text_to_screenshot(input_text, extension_percentage, temperature, trim_whitespace)
-# Gradio UIの定義
 with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr.themes.Base()) as iface:
     gr.Markdown("# HTMLビューア & テキスト→インフォグラフィック変換")
     gr.Markdown("HTMLコードをレンダリングするか、テキストをGemini APIでインフォグラフィックに変換して画像として取得します。")
     with gr.Row():
-        input_mode = gr.Radio(
-            ["HTML入力", "テキスト入力"],
-            label="入力モード",
-            value="HTML入力"
-        )
-    # 共用のテキストボックス
-    input_text = gr.Textbox(
-        lines=15,
-        label="入力",
-        placeholder="HTMLコードまたはテキストを入力してください。入力モードに応じて処理されます。"
-    )
     with gr.Row():
-        extension_percentage = gr.Slider(
-            minimum=0,
-            maximum=30,
-            step=1.0,
-            value=6,    # デフォルト値6%
-            label="上下高さ拡張率（%）"
-        )
-        # 温度調整スライダー（テキストモード時のみ表示）
-        temperature = gr.Slider(
-            minimum=0.0,
-            maximum=1.0,
-            step=0.1,
-            value=0.3,    # デフォルト値を0.3に下げて創造性を抑制
-            label="生成時の温度（低い=一貫性高、高い=創造性高）",
-            visible=False  # 最初は非表示
-        )
-    # 余白トリミングオプション
-    trim_whitespace = gr.Checkbox(
-        label="余白を自動トリミング",
-        value=True,
-        info="生成される画像から余分な空白領域を自動的に削除します"
-    )
     submit_btn = gr.Button("生成")
     output_image = gr.Image(type="pil", label="ページ全体のスクリーンショット")
-    # 入力モード変更時のイベント処理（テキストモード時のみ温度スライダーを表示）
     def update_temperature_visibility(mode):
-        # Gradio 4.x用のアップデート方法
         return {"visible": mode == "テキスト入力", "__type__": "update"}
-    input_mode.change(
-        fn=update_temperature_visibility,
-        inputs=input_mode,
-        outputs=temperature
-    )
-    # 生成ボタンクリック時のイベント処理
-    submit_btn.click(
-        fn=process_input,
-        inputs=[input_mode, input_text, extension_percentage, temperature, trim_whitespace],
-        outputs=output_image
-    )
-    # 環境変数情報を表示
     gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
     gr.Markdown(f"""
     ## APIエンドポイント
-    - `/api/screenshot` - HTMLコードからスクリーンショットを生成
     - `/api/text-to-screenshot` - テキストからインフォグラフィックスクリーンショットを生成
     ## 設定情報
     - 使用モデル: {gemini_model} (環境変数 GEMINI_MODEL で変更可能)
     """)
-# --- Mount Gradio App onto FastAPI ---
 app = gr.mount_gradio_app(app, iface, path="/")
-# --- Run with Uvicorn (for local testing) ---
 if __name__ == "__main__":
     import uvicorn
     logger.info("Starting Uvicorn server for local development...")
-    uvicorn.run(app, host="0.0.0.0", port=7860)

+# -*- coding: utf-8 -*-
 import gradio as gr
 from fastapi import FastAPI, HTTPException, Body
 from fastapi.responses import StreamingResponse
 from selenium.webdriver.common.by import By
 from selenium.webdriver.support.ui import WebDriverWait
 from selenium.webdriver.support import expected_conditions as EC
+# TimeoutException と JavascriptException をインポート
+from selenium.common.exceptions import TimeoutException, JavascriptException
 from PIL import Image
 from io import BytesIO
 import tempfile
         margin-right: 8px !important;
         vertical-align: middle !important;
     }
     /* テキスト内のアイコン位置調整 */
+    h1 [class*="fa-"], h2 [class*="fa-"], h3 [class*="fa-"],
     h4 [class*="fa-"], h5 [class*="fa-"], h6 [class*="fa-"] {
         vertical-align: middle !important;
         margin-right: 10px !important;
     }
     /* 特定パターンの修正 */
     .fa + span, .fas + span, .far + span, .fab + span,
     span + .fa, span + .fas, span + .far, span + .fab {
         display: inline-block !important;
         margin-left: 5px !important;
     }
     /* カード内アイコン修正 */
     .card [class*="fa-"], .card-body [class*="fa-"] {
         float: none !important;
         clear: none !important;
         position: relative !important;
     }
     /* アイコンと文字が重なる場合の調整 */
     li [class*="fa-"], p [class*="fa-"] {
         margin-right: 10px !important;
     }
     /* インラインアイコンのスペーシング */
     .inline-icon {
         display: inline-flex !important;
         align-items: center !important;
         justify-content: flex-start !important;
     }
     /* アイコン後のテキスト */
     [class*="fa-"] + span {
         display: inline-block !important;
     }
     </style>
     """
     # headタグがある場合はその中に追加
     if '<head>' in html_code:
         return html_code.replace('</head>', f'{fa_fix_css}</head>')
             body_start = html_code.find('<body')
             if body_start > 0:
                 return html_code[:body_start] + f'<head>{fa_fix_css}</head>' + html_code[body_start:]
     # どちらもない場合は先頭に追加
     return f'<html><head>{fa_fix_css}</head>' + html_code + '</html>'
         if not api_key:
             logger.error("GEMINI_API_KEY 環境変数が設定されていません")
             raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
         # モデル名の取得（環境変数から、なければデフォルト値）
         model_name = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
         logger.info(f"使用するGeminiモデル: {model_name}")
         # Gemini APIの設定
         genai.configure(api_key=api_key)
+        # システムプロンプト（変更なし）
         system_instruction = """# グラフィックレコーディング風インフォグラフィック変換プロンプト V2
 ## 目的
 以下の内容を、超一流デザイナーが作成したような、日本語で完璧なグラフィックレコーディング風のHTMLインフォグラフィックに変換してください。情報設計とビジュアルデザインの両面で最高水準を目指します。
 - フッターに出典情報と関連するFont Awesomeアイコン（fa-book, fa-citation等）を明記
 ## 変換する文章/記事
 ーーー＜ユーザーが入力（または添付）＞ーーー"""
         # モデルを初期化して処理
         logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}")
         # モデル初期化とフォールバック処理
         try:
             model = genai.GenerativeModel(model_name)
         except Exception as e:
             fallback_model = "gemini-pro"
             logger.warning(f"{model_name}が利用できません: {e}, フォールバックモデル{fallback_model}を使用します")
             model = genai.GenerativeModel(fallback_model)
+        # 生成設定 (変更なし)
         generation_config = {
+            "temperature": temperature,
+            "top_p": 0.7,
+            "top_k": 20,
             "max_output_tokens": 8192,
+            "candidate_count": 1
         }
+        # 安全設定 (変更なし)
         safety_settings = [
+            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"}
         ]
         # プロンプト構築
         prompt = f"{system_instruction}\n\n{text}"
         # コンテンツ生成
         response = model.generate_content(
             prompt,
             generation_config=generation_config,
             safety_settings=safety_settings
         )
+        # レスポンスからHTMLを抽出 (変更なし)
         raw_response = response.text
         html_start = raw_response.find("```html")
         html_end = raw_response.rfind("```")
         if html_start != -1 and html_end != -1 and html_start < html_end:
+            html_start += 7
             html_code = raw_response[html_start:html_end].strip()
             logger.info(f"HTMLの生成に成功: 長さ = {len(html_code)}")
             html_code = enhance_font_awesome_layout(html_code)
             logger.info("Font Awesomeレイアウトの最適化を適用しました")
             return html_code
         else:
             logger.warning("レスポンスから```html```タグが見つかりませんでした。全テキストを返します。")
             return raw_response
     except Exception as e:
         logger.error(f"HTML生成中にエラーが発生: {e}", exc_info=True)
         raise Exception(f"Gemini APIでのHTML生成に失敗しました: {e}")
+# 画像から余分な空白領域をトリミングする関数 (変更なし)
 def trim_image_whitespace(image, threshold=250, padding=10):
     gray = image.convert('L')
     data = gray.getdata()
     width, height = gray.size
     min_x, min_y = width, height
     max_x = max_y = 0
     pixels = list(data)
     pixels = [pixels[i * width:(i + 1) * width] for i in range(height)]
     for y in range(height):
         for x in range(width):
+            if pixels[y][x] < threshold:
                 min_x = min(min_x, x)
                 min_y = min(min_y, y)
                 max_x = max(max_x, x)
                 max_y = max(max_y, y)
     if min_x > max_x or min_y > max_y:
         logger.warning("トリミング領域が見つかりません。元の画像を返します。")
         return image
     min_x = max(0, min_x - padding)
     min_y = max(0, min_y - padding)
     max_x = min(width - 1, max_x + padding)
     max_y = min(height - 1, max_y + padding)
     trimmed = image.crop((min_x, min_y, max_x + 1, max_y + 1))
     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:
     """
     Renders HTML code to a full-page screenshot using Selenium.
     """
+    tmp_path = None
+    driver = None
+    # 1) Save HTML code to a temporary file (変更なし)
     try:
         with tempfile.NamedTemporaryFile(suffix=".html", delete=False, mode='w', encoding='utf-8') as tmp_file:
             tmp_path = tmp_file.name
         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))
+    # 2) Headless Chrome options (変更なし)
     options = Options()
     options.add_argument("--headless")
     options.add_argument("--no-sandbox")
     options.add_argument("--disable-dev-shm-usage")
     options.add_argument("--force-device-scale-factor=1")
     options.add_argument("--disable-features=NetworkService")
     options.add_argument("--dns-prefetch-disable")
         driver = webdriver.Chrome(options=options)
         logger.info("WebDriver initialized.")
+        # 3) Initial window size and navigation (変更なし)
+        initial_width = 1200
         initial_height = 1000
         driver.set_window_size(initial_width, initial_height)
         file_url = "file://" + tmp_path
         logger.info(f"Navigating to {file_url}")
         driver.get(file_url)
+        # 4) Wait for body (変更なし)
         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 potential resource loading...")
+        time.sleep(3)
+        # 5) Font Awesome and resource check (修正: 即時実行関数 + コメント削除)
         try:
             resource_check_script = """
+            return (function() {
+                return new Promise((resolve) => {
+                    const checkFontAwesome = () => {
+                        const icons = document.querySelectorAll('.fa, .fas, .far, .fab, [class*="fa-"]');
+                        if (icons.length > 0) {
+                            document.fonts.ready.then(() => {
+                                setTimeout(resolve, 1000);
+                            });
+                        } else {
+                            document.fonts.ready.then(() => 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)
+            # execute_async_script に渡す前に文字列として結合
+            full_script = f"const callback = arguments[arguments.length - 1]; ({resource_check_script}).then(callback);"
+            driver.execute_async_script(full_script)
             logger.info("Resources loaded successfully")
+        except (TimeoutException, JavascriptException) as e: # JavascriptExceptionも捕捉
             logger.warning(f"Resource loading check failed: {e}. Using fallback wait.")
+            time.sleep(8)
+        except Exception as e: # その他の予期せぬエラー
+             logger.error(f"Unexpected error during resource check: {e}", exc_info=True)
+             time.sleep(8)
+        # 6) Ensure content rendering by scrolling (修正: 即時実行関数 + コメント削除 + ロジック微調整)
         try:
             scroll_script = """
+            return (function() {
+                return new Promise(resolve => {
+                    const pageHeight = Math.max(
+                        document.documentElement.scrollHeight || 0,
+                        document.body ? document.body.scrollHeight || 0 : 0
+                    );
+                    const viewportHeight = window.innerHeight || document.documentElement.clientHeight || 800;
+                    const scrollStep = Math.floor(viewportHeight * 0.8);
+                    let currentPos = 0;
+                    const scrollDown = () => {
+                        if (!document.body || pageHeight <= 0) {
+                             window.scrollTo(0, 0);
+                             setTimeout(resolve, 300);
+                             return;
+                        }
+                        if (currentPos < pageHeight) {
+                            window.scrollTo(0, currentPos);
+                            currentPos += scrollStep;
+                            if (currentPos >= pageHeight || scrollStep <= 0) {
+                                 window.scrollTo(0, pageHeight);
+                                 setTimeout(() => {
+                                     window.scrollTo(0, 0);
+                                     setTimeout(resolve, 300);
+                                 }, 150);
+                            } else {
+                                setTimeout(scrollDown, 100);
+                            }
+                        } else {
+                             window.scrollTo(0, 0);
+                             setTimeout(resolve, 300);
+                        }
+                    };
+                    setTimeout(scrollDown, 100);
+                });
+            })();
             """
             logger.info("Ensuring all content is rendered...")
+            driver.set_script_timeout(20) # スクロール用にタイムアウトを少し長く
+            # execute_async_script に渡す前に文字列として結合
+            full_script = f"const callback = arguments[arguments.length - 1]; ({scroll_script}).then(callback);"
+            driver.execute_async_script(full_script)
+            logger.info("Content rendering scroll finished.")
+        except (TimeoutException, JavascriptException) as e: # JavascriptExceptionも捕捉
+            logger.warning(f"Content rendering check failed: {e}. Scrolling to top and waiting.")
+            try:
+                driver.execute_script("window.scrollTo(0, 0);")
+            except Exception as scroll_err:
+                logger.error(f"Could not scroll to top after render check failure: {scroll_err}")
+            time.sleep(3) # フォールバック待機
+        except Exception as e: # その他の予期せぬエラー
+             logger.error(f"Unexpected error during content rendering scroll: {e}", exc_info=True)
+             time.sleep(3)
+        # 7) Hide scrollbars (変更なし)
         try:
             driver.execute_script(
                 "document.documentElement.style.overflow = 'hidden';"
         except Exception as e:
             logger.warning(f"Could not hide scrollbars via JS: {e}")
+        # 8) Get full page dimensions (変更なし)
         try:
             dimensions_script = """
             return {
                 width: Math.max(
+                    document.documentElement.scrollWidth || 0,
+                    document.documentElement.offsetWidth || 0,
+                    document.documentElement.clientWidth || 0,
+                    document.body ? document.body.scrollWidth || 0 : 0,
+                    document.body ? document.body.offsetWidth || 0 : 0,
+                    document.body ? document.body.clientWidth || 0 : 0
                 ),
                 height: Math.max(
+                    document.documentElement.scrollHeight || 0,
+                    document.documentElement.offsetHeight || 0,
+                    document.documentElement.clientHeight || 0,
+                    document.body ? document.body.scrollHeight || 0 : 0,
+                    document.body ? document.body.offsetHeight || 0 : 0,
+                    document.body ? document.body.clientHeight || 0 : 0
                 )
             };
             """
             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}")
+            # スクロール検証 (エラーハンドリング追加)
+            try:
+                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}")
+            except Exception as scroll_e:
+                 logger.warning(f"Error during dimension scroll check: {scroll_e}")
+            scroll_width = max(scroll_width, 100)
+            scroll_height = max(scroll_height, 100)
+            scroll_width = min(scroll_width, 2000)
+            scroll_height = min(scroll_height, 4000)
         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}")
+        # 9) Layout stability check (修正: 即時実行関数 + コメント削除 + JavascriptException捕捉)
         try:
             stability_script = """
+            return (function() {
+                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;
+                        }
+                        const bodyExists = document.body && typeof document.body.offsetHeight === 'number' && typeof document.body.offsetWidth === 'number';
+                        if (!bodyExists) {
+                            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;
+                        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) {
+                                console.log('Layout deemed stable.');
+                                resolve(true);
+                                return;
+                            }
+                        } else {
+                            stableCount = 0;
+                            lastHeight = currentHeight;
+                            lastWidth = currentWidth;
+                            console.log(`Layout changed: ${lastWidth}x${lastHeight}. Resetting stability count.`);
+                        }
+                        setTimeout(checkStability, 200);
+                    };
+                    setTimeout(checkStability, 100);
+                });
+            })();
+            """
             logger.info("Checking layout stability...")
             driver.set_script_timeout(20)
+            # execute_async_script に渡す前に文字列として結合
+            full_script = f"const callback = arguments[arguments.length - 1]; ({stability_script}).then(callback);"
+            stable = driver.execute_async_script(full_script)
             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 JavascriptException as e: # JavascriptExceptionを捕捉
+             logger.error(f"Javascript error during layout stability check: {e}", exc_info=True)
+             time.sleep(2)
         except Exception as e:
+            logger.error(f"Unexpected error during layout stability check: {e}", exc_info=True)
+            time.sleep(2)
+        # 10) Calculate adjusted height (変更なし)
         adjusted_height = int(scroll_height * (1 + extension_percentage / 100.0))
+        adjusted_height = max(adjusted_height, scroll_height, 100)
         logger.info(f"Adjusted height calculated: {adjusted_height} (extension: {extension_percentage}%)")
+        # 11) Set window size and wait (変更なし)
         adjusted_width = scroll_width
         logger.info(f"Resizing window to: width={adjusted_width}, height={adjusted_height}")
         driver.set_window_size(adjusted_width, adjusted_height)
         logger.info("Waiting for layout stabilization after resize...")
+        time.sleep(4)
+        # Resource state check (変更なし, エラーハンドリングは元々���り)
         try:
             resource_state = driver.execute_script("""
             return {
                 readyState: document.readyState,
+                resourcesComplete: !document.querySelector('img:not([complete])') &&
                                    !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...")
                 time.sleep(2)
         except Exception as e:
             logger.warning(f"Error checking resource state: {e}")
+        # Scroll to top (変更なし, エラーハンドリングは元々あり)
         try:
             driver.execute_script("window.scrollTo(0, 0)")
             time.sleep(1)
         except Exception as e:
              logger.warning(f"Could not scroll to top: {e}")
+        # 12) Take screenshot (変更なし)
         logger.info("Taking screenshot...")
         png = driver.get_screenshot_as_png()
         logger.info("Screenshot taken successfully.")
         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"An error occurred during screenshot generation: {e}", exc_info=True)
+        return Image.new('RGB', (1, 1), color=(0, 0, 0))
     finally:
         logger.info("Cleaning up...")
         if driver:
             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:
     try:
         html_code = generate_html_from_text(text, temperature)
         return render_fullpage_screenshot(html_code, extension_percentage, trim_whitespace)
     except Exception as e:
         logger.error(f"テキストからスクリーンショット生成中にエラーが発生: {e}", exc_info=True)
+        return Image.new('RGB', (1, 1), color=(0, 0, 0))
+# --- FastAPI Setup --- (変更なし)
 app = FastAPI()
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_methods=["*"],
     allow_headers=["*"],
 )
 gradio_dir = os.path.dirname(gr.__file__)
 logger.info(f"Gradio version: {gr.__version__}")
 logger.info(f"Gradio directory: {gradio_dir}")
 static_dir = os.path.join(gradio_dir, "templates", "frontend", "static")
 if os.path.exists(static_dir):
     logger.info(f"Mounting static directory: {static_dir}")
     app.mount("/static", StaticFiles(directory=static_dir), name="static")
 app_dir = os.path.join(gradio_dir, "templates", "frontend", "_app")
 if os.path.exists(app_dir):
     logger.info(f"Mounting _app directory: {app_dir}")
     app.mount("/_app", StaticFiles(directory=app_dir), name="_app")
 assets_dir = os.path.join(gradio_dir, "templates", "frontend", "assets")
 if os.path.exists(assets_dir):
     logger.info(f"Mounting assets directory: {assets_dir}")
     app.mount("/assets", StaticFiles(directory=assets_dir), name="assets")
 cdn_dir = os.path.join(gradio_dir, "templates", "cdn")
 if os.path.exists(cdn_dir):
     logger.info(f"Mounting cdn directory: {cdn_dir}")
     app.mount("/cdn", StaticFiles(directory=cdn_dir), name="cdn")
+# API Endpoint for screenshot generation (変更なし)
+@app.post("/api/screenshot", response_class=StreamingResponse, tags=["Screenshot"], summary="Render HTML to Full Page Screenshot")
 async def api_render_screenshot(request: ScreenshotRequest):
     try:
         logger.info(f"API request received. Extension: {request.extension_percentage}%")
         pil_image = render_fullpage_screenshot(
             request.html_code,
             request.extension_percentage,
             request.trim_whitespace
         )
         if pil_image.size == (1, 1):
              logger.error("Screenshot generation failed, returning 1x1 image.")
         img_byte_arr = BytesIO()
         pil_image.save(img_byte_arr, format='PNG')
+        img_byte_arr.seek(0)
         logger.info("Returning screenshot as PNG stream.")
         return StreamingResponse(img_byte_arr, media_type="image/png")
     except Exception as e:
         logger.error(f"API Error: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Internal Server Error: {e}")
+# Gemini API連携エンドポイント (変更なし)
+@app.post("/api/text-to-screenshot", response_class=StreamingResponse, tags=["Screenshot", "Gemini"], summary="テキストからインフォグラフィックを生成")
 async def api_text_to_screenshot(request: GeminiRequest):
     try:
         logger.info(f"テキスト→スクリーンショットAPIリクエスト受信。テキスト長さ: {len(request.text)}, 拡張率: {request.extension_percentage}%, 温度: {request.temperature}")
         pil_image = text_to_screenshot(
             request.text,
             request.extension_percentage,
             request.temperature,
             request.trim_whitespace
         )
         if pil_image.size == (1, 1):
             logger.error("スクリーンショット生成に失敗しました。1x1画像を返します。")
         img_byte_arr = BytesIO()
         pil_image.save(img_byte_arr, format='PNG')
+        img_byte_arr.seek(0)
         logger.info("スクリーンショットをPNGストリームとして返します。")
         return StreamingResponse(img_byte_arr, media_type="image/png")
     except Exception as e:
         logger.error(f"API Error: {e}", exc_info=True)
         raise HTTPException(status_code=500, detail=f"Internal Server Error: {e}")
+# --- Gradio Interface Definition --- (変更なし)
 def process_input(input_mode, input_text, extension_percentage, temperature, trim_whitespace):
     if input_mode == "HTML入力":
         return render_fullpage_screenshot(input_text, extension_percentage, trim_whitespace)
     else:
         return text_to_screenshot(input_text, extension_percentage, temperature, trim_whitespace)
 with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr.themes.Base()) as iface:
     gr.Markdown("# HTMLビューア & テキスト→インフォグラフィック変換")
     gr.Markdown("HTMLコードをレンダリングするか、テキストをGemini APIでインフォグラフィックに変換して画像として取得します。")
     with gr.Row():
+        input_mode = gr.Radio(["HTML入力", "テキスト入力"], label="入力モード", value="HTML入力")
+    input_text = gr.Textbox(lines=15, label="入力", placeholder="HTMLコードまたはテキストを入力してください。入力モードに応じて処理されます。")
     with gr.Row():
+        extension_percentage = gr.Slider(minimum=0, maximum=30, step=1.0, value=6, label="上下高さ拡張率（%）")
+        temperature = gr.Slider(minimum=0.0, maximum=1.0, step=0.1, value=0.3, label="生成時の温度（低い=一貫性高、高い=創造性高）", visible=False)
+    trim_whitespace = gr.Checkbox(label="余白を自動トリミング", value=True, info="生成される画像から余分な空白領域を自動的に削除します")
     submit_btn = gr.Button("生成")
     output_image = gr.Image(type="pil", label="ページ全体のスクリーンショット")
     def update_temperature_visibility(mode):
         return {"visible": mode == "テキスト入力", "__type__": "update"}
+    input_mode.change(fn=update_temperature_visibility, inputs=input_mode, outputs=temperature)
+    submit_btn.click(fn=process_input, inputs=[input_mode, input_text, extension_percentage, temperature, trim_whitespace], outputs=output_image)
     gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
     gr.Markdown(f"""
     ## APIエンドポイント
+    - `/api/screenshot` - HTMLコードからスクリーンショットを生成
     - `/api/text-to-screenshot` - テキストからインフォグラフィックスクリーンショットを生成
     ## 設定情報
     - 使用モデル: {gemini_model} (環境変数 GEMINI_MODEL で変更可能)
     """)
+# --- Mount Gradio App onto FastAPI --- (変更なし)
 app = gr.mount_gradio_app(app, iface, path="/")
+# --- Run with Uvicorn (for local testing) --- (変更なし)
 if __name__ == "__main__":
     import uvicorn
     logger.info("Starting Uvicorn server for local development...")
+    uvicorn.run(app, host="0.0.0.0", port=7860)