HTMLviewer2_API

Paused

App Files Files Community

tomo2chin2 commited on Mar 28, 2025

Commit

6c102df

verified ·

1 Parent(s): 178eeaf

Update app.py

Browse files

Files changed (1) hide show

app.py +252 -551

app.py CHANGED Viewed

@@ -15,7 +15,6 @@ import tempfile
 import time
 import os
 import logging
-import re
 # 正しいGemini関連のインポート
 import google.generativeai as genai
@@ -31,7 +30,6 @@ class GeminiRequest(BaseModel):
     extension_percentage: float = 6.0  # デフォルト値6%
     temperature: float = 1.0  # デフォルト値1.0の温度パラメータ
     trim_whitespace: bool = True  # 余白トリミングオプション（デフォルト有効）
-    layout_type: str = "standard"  # レイアウトタイプ（standard, spaced, compact）
 class ScreenshotRequest(BaseModel):
     """スクリーンショットリクエストモデル"""
@@ -39,7 +37,77 @@ class ScreenshotRequest(BaseModel):
     extension_percentage: float = 6.0  # デフォルト値6%
     trim_whitespace: bool = True  # 余白トリミングオプション（デフォルト有効）
-def generate_html_from_text(text, temperature=1.0, layout_type="standard"):
     """テキストからHTMLを生成する"""
     try:
         # APIキーの取得と設定
@@ -55,88 +123,8 @@ def generate_html_from_text(text, temperature=1.0, layout_type="standard"):
         # Gemini APIの設定
         genai.configure(api_key=api_key)
-        # レイアウトタイプに応じた追加設定
-        layout_adjustments = ""
-        if layout_type == "spaced":
-            layout_adjustments = """
-## 特別なスタイル設定
-- 全てのテキスト要素には最低2rem以上の行間と要素間マージンを設定する
-- 各カードのパディングを1.5倍に増やす
-- 箇条書きリストのアイテム間は1.5emの間隔を確保する
-- すべての文字サイズを15%大きくする
-- 内容量に応じて自動的にグリッドやフレックスレイアウトを調整する
-- テキストが重ならないように要素間に十分なスペースを強制的に設ける
-"""
-        elif layout_type == "compact":
-            layout_adjustments = """
-## 特別なスタイル設定
-- レイアウトをよりコンパクトに配置
-- 小さめのアイコンサイズを使用（標準の75%）
-- フォントサイズを10%小さく設定
-- テキスト量が多い場合は自動的に複数列に分割表示
-"""
-        # レイアウト崩れ防止のための追加指示
-        layout_improvements = """
-## 重要なレイアウト指示（必ず守ってください）
-- カード要素は必ずdisplay: gridまたはflexboxで配置してください
-- カードの配置は絶対位置(position: absolute)を使わず、フロー配置を使ってください
-- 各カードには以下のスタイルを必ず適用してください:
-  ```css
-  .card {
-    position: relative;
-    clear: both;
-    display: flex;
-    flex-direction: column;
-    margin: 1.5rem 0;
-    padding: 1.2rem;
-    height: auto;
-    overflow: visible;
-    box-sizing: border-box;
-  }
-  ```
-- 兄弟要素間のマージンは必ず以下のように設定してください:
-  ```css
-  .card + .card,
-  .section + .section,
-  .info-box + .info-box,
-  .row + .row {
-    margin-top: 2rem;
-  }
-  ```
-- コンテナの高さは自動的に調整されるようにしてください:
-  ```css
-  .container, .row, .column {
-    height: auto;
-    overflow: hidden;
-    display: flex;
-    flex-direction: column;
-  }
-  ```
-- グリッドレイアウトを使用する場合は明示的な行の高さを設定してください:
-  ```css
-  .grid {
-    display: grid;
-    grid-template-columns: repeat(auto-fit, minmax(300px, 1fr));
-    grid-gap: 2rem;
-    grid-auto-rows: minmax(min-content, auto);
-  }
-  ```
-- すべてのテキスト要素に余白を設定してください:
-  ```css
-  p, h1, h2, h3, h4, h5, h6, li {
-    margin-bottom: 1rem;
-    line-height: 1.6;
-  }
-  ```
-"""
-        # システムプロンプト（強化版）
-        system_instruction = f"""# グラフィックレコーディング風インフォグラフィック変換プロンプト V3
 ## 目的
 以下の内容を、超一流デザイナーが作成したような、日本語で完璧なグラフィックレコーディング風のHTMLインフォグラフィックに変換してください。情報設計とビジュアルデザインの両面で最高水準を目指します。
 手書き風の図形やFont Awesomeアイコンを大きく活用して内容を視覚的かつ直感的に表現します。
@@ -183,8 +171,8 @@ def generate_html_from_text(text, temperature=1.0, layout_type="standard"):
 - フォント指定：
 ```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');
 </style>
 ```
 ### 5. レイアウト
@@ -196,81 +184,15 @@ def generate_html_from_text(text, temperature=1.0, layout_type="standard"):
 - 横幅は100%
 - 重要な要素は中央寄り、補足情報は周辺部に配置
-{layout_improvements}
-## レイアウト崩れ防止のための追加設定
-- テキストが重ならないようにカードやセクション間に十分なマージンを設定（最低25px）
-- テキスト量に応じて自動的に要素の高さが伸びるフレックスレイアウトを採用
-- テキストが多い場合は自動的にフォントサイズを調整する機能を実装
-- スマートスペーシング：内容量に応じて自動的に間隔を調整
-- 以下のCSSを必ず追加して、テキストの重なりを防止する：
-```css
-.card, .section, .info-box {{
-  overflow: hidden;
-  display: flex;
-  flex-direction: column;
-  margin-bottom: 1.5rem;
-  min-height: fit-content;
-}}
-.card-content {{
-  flex: 1;
-  display: flex;
-  flex-direction: column;
-}}
-.card-title {{
-  font-weight: bold;
-  margin-bottom: 0.75rem;
-  line-height: 1.4;
-}}
-.card-text {{
-  line-height: 1.6;
-  margin-bottom: 0.5rem;
-}}
-.info-item {{
-  margin-bottom: 0.75rem;
-  line-height: 1.5;
-}}
-li {{
-  margin-bottom: 0.5rem;
-  line-height: 1.5;
-}}
-p {{
-  margin-bottom: 0.75rem;
-  line-height: 1.5;
-}}
-```
-## 自動テキスト調整機能
-以下のJavaScriptを必ず追加して、テキスト量に応じたフォントサイズの自動調整を行う：
-```html
-<script>
-document.addEventListener('DOMContentLoaded', function() {{
-  // カード内のテキスト量に基づいてフォントサイズを調整
-  const cards = document.querySelectorAll('.card, .section, .info-box');
-  cards.forEach(card => {{
-    const cardContent = card.querySelector('.card-content') || card;
-    const textLength = cardContent.textContent.length;
-    // テキスト量に応じてフォントサイズを調整
-    if (textLength > 200) {{
-      card.style.fontSize = '0.9em';
-    }}
-    if (textLength > 300) {{
-      card.style.fontSize = '0.85em';
-    }}
-    // カード内の要素間の余白を確保
-    const elements = card.querySelectorAll('p, h3, h4, ul, ol, .info-item');
-    elements.forEach(el => {{
-      el.style.marginBottom = '0.75rem';
-      el.style.lineHeight = '1.5';
-    }});
-  }});
-}});
-</script>
-```
-{layout_adjustments}
 ## グラフィックレコーディング表現技法
 - テキストと視覚要素のバランスを重視（文字情報の50%以上をFont Awesomeアイコンで視覚的に補完）
@@ -311,7 +233,7 @@ document.addEventListener('DOMContentLoaded', function() {{
 ーーー＜ユーザーが入力（または添付）＞ーーー"""
         # モデルを初期化して処理
-        logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}, レイアウトタイプ = {layout_type}")
         # モデル初期化とフォールバック処理
         try:
@@ -351,8 +273,9 @@ document.addEventListener('DOMContentLoaded', function() {{
             html_code = raw_response[html_start:html_end].strip()
             logger.info(f"HTMLの生成に成功: 長さ = {len(html_code)}")
-            # HTML後処理：テキスト重なり防止のためのクラス・スタイル追加
-            html_code = enhance_html_for_layout(html_code)
             return html_code
         else:
@@ -364,237 +287,6 @@ document.addEventListener('DOMContentLoaded', function() {{
         logger.error(f"HTML生成中にエラーが発生: {e}", exc_info=True)
         raise Exception(f"Gemini APIでのHTML生成に失敗しました: {e}")
-# HTMLを後処理してレイアウト崩れを防止する関数
-def enhance_html_for_layout(html_code):
-    """HTMLを後処理してレイアウト崩れを防止するための調整を行う（強化版）"""
-    try:
-        # 基本的なスタイルとスクリプトが含まれているか確認
-        base_styles_needed = True
-        base_scripts_needed = True
-        if "<style>" in html_code:
-            style_pos = html_code.find("</style>")
-            if style_pos != -1:
-                # カード要素のスタイルが含まれているか確認
-                if ".card" in html_code[:style_pos] and "flex-direction: column" in html_code[:style_pos]:
-                    base_styles_needed = False
-                # レイアウト崩れ防止用の強化CSSを追加
-                enhanced_css = """
-/* --- レイアウト崩れ防止強化スタイル --- */
-.card, .section, .info-box {
-  position: relative !important;
-  display: flex !important;
-  flex-direction: column !important;
-  margin: 1.5rem 0 !important;
-  padding: 1.2rem !important;
-  height: auto !important;
-  min-height: fit-content !important;
-  overflow: visible !important;
-  box-sizing: border-box !important;
-  z-index: 1 !important;
-  clear: both !important;
-}
-/* フローティング要素のクリア */
-.clearfix::after {
-  content: "" !important;
-  display: table !important;
-  clear: both !important;
-}
-/* コンテナ要素の高さ自動調整 */
-.container, .row, .column {
-  height: auto !important;
-  overflow: hidden !important;
-  position: relative !important;
-}
-/* グリッドレイアウトの調整 */
-.grid {
-  display: grid !important;
-  grid-template-columns: repeat(auto-fit, minmax(300px, 1fr)) !important;
-  grid-gap: 2rem !important;
-  grid-auto-rows: minmax(min-content, auto) !important;
-}
-/* マージンとパディングの強制 */
-p, h1, h2, h3, h4, h5, h6, li {
-  margin-bottom: 1rem !important;
-  line-height: 1.6 !important;
-}
-/* 重なり解消のための特別なスタイル */
-body {
-  overflow-x: hidden !important;
-}
-.card + .card, .section + .section, .info-box + .info-box {
-  margin-top: 2rem !important;
-}
-/* モバイル表示の改善 */
-@media (max-width: 768px) {
-  .card, .section, .info-box {
-    width: 100% !important;
-    margin-left: 0 !important;
-    margin-right: 0 !important;
-  }
-}
-"""
-                # スタイルタグの閉じ直前に挿入
-                html_code = html_code[:style_pos] + enhanced_css + html_code[style_pos:]
-        # スタイルが存在しない場合は新たに追加
-        if base_styles_needed and "<style>" not in html_code:
-            head_end_pos = html_code.find("</head>")
-            if head_end_pos != -1:
-                base_styles = """
-<style>
-/* ベーススタイル - レイアウト崩れ防止 */
-.card, .section, .info-box {
-  position: relative !important;
-  display: flex !important;
-  flex-direction: column !important;
-  margin: 1.5rem 0 !important;
-  padding: 1.2rem !important;
-  height: auto !important;
-  min-height: fit-content !important;
-  overflow: visible !important;
-  box-sizing: border-box !important;
-  z-index: 1 !important;
-  clear: both !important;
-}
-.card-content {
-  flex: 1 !important;
-  display: flex !important;
-  flex-direction: column !important;
-}
-.container, .row, .column {
-  height: auto !important;
-  overflow: hidden !important;
-  position: relative !important;
-}
-p, h1, h2, h3, h4, h5, h6, li {
-  margin-bottom: 1rem !important;
-  line-height: 1.6 !important;
-}
-/* 重なり解消のための特別なスタイル */
-body {
-  overflow-x: hidden !important;
-}
-.card + .card, .section + .section, .info-box + .info-box {
-  margin-top: 2rem !important;
-}
-</style>
-"""
-                html_code = html_code[:head_end_pos] + base_styles + html_code[head_end_pos:]
-        # 自動レイアウト修正用のスクリプトを強化
-        if "</body>" in html_code:
-            has_stabilize_script = "stabilizeLayout" in html_code
-            if not has_stabilize_script:
-                body_end_pos = html_code.find("</body>")
-                if body_end_pos != -1:
-                    enhanced_script = """
-<script>
-// レイアウト安定化のための強化スクリプト
-function stabilizeLayout() {
-  // カードの重なりを解消
-  document.querySelectorAll('.card, .section, .info-box').forEach(card => {
-    // 位置を相対指定に強制
-    card.style.position = 'relative';
-    // z-indexをカード間で競合しないよう設定
-    card.style.zIndex = '1';
-    // マージンを強制的に設定
-    card.style.marginBottom = '2rem';
-    card.style.marginTop = '1rem';
-    // 浮動要素のクリア
-    let clearDiv = document.createElement('div');
-    clearDiv.style.clear = 'both';
-    clearDiv.style.height = '1px';
-    card.parentNode.insertBefore(clearDiv, card.nextSibling);
-    // クリアフィックスクラスを追加
-    card.classList.add('clearfix');
-  });
-  // 親要素の高さを適切に設定
-  document.querySelectorAll('.row, .container, .column').forEach(container => {
-    container.style.overflow = 'hidden';
-    container.style.height = 'auto';
-    container.classList.add('clearfix');
-    // 子要素の合計高さを取得して設定
-    let totalHeight = 0;
-    Array.from(container.children).forEach(child => {
-      if (child.offsetHeight) {
-        totalHeight += child.offsetHeight +
-                       parseInt(getComputedStyle(child).marginTop || 0) +
-                       parseInt(getComputedStyle(child).marginBottom || 0);
-      }
-    });
-    if (totalHeight > 0) {
-      container.style.minHeight = (totalHeight + 50) + 'px';
-    }
-  });
-  // テキスト量に基づいてフォントサイズを調整
-  document.querySelectorAll('.card, .section, .info-box').forEach(card => {
-    const cardContent = card.querySelector('.card-content') || card;
-    const textLength = cardContent.textContent.length;
-    // テキスト量に応じてフォントサイズを調整
-    if (textLength > 200) {
-      card.style.fontSize = '0.9em';
-    }
-    if (textLength > 300) {
-      card.style.fontSize = '0.85em';
-    }
-    // カード内の要素間の余白を確保
-    const elements = card.querySelectorAll('p, h3, h4, ul, ol, .info-item');
-    elements.forEach(el => {
-      el.style.marginBottom = '0.75rem';
-      el.style.lineHeight = '1.5';
-    });
-  });
-}
-// DOMContentLoadedイベントで実行
-document.addEventListener('DOMContentLoaded', stabilizeLayout);
-// 安全のためにウィンドウロード時にも実行
-window.addEventListener('load', stabilizeLayout);
-// 1秒後に再度実行して安定化を確実にする
-setTimeout(stabilizeLayout, 1000);
-</script>
-"""
-                    html_code = html_code[:body_end_pos] + enhanced_script + html_code[body_end_pos:]
-        # カードコンテナのクラス追加
-        html_code = re.sub(r'<div class="([^"]*)"([^>]*)>', lambda m: f'<div class="{m.group(1)} card clearfix"{m.group(2)}>', html_code)
-        # ボディタグにクラス追加
-        if "<body" in html_code and "clearfix" not in html_code:
-            html_code = html_code.replace("<body", '<body class="clearfix"')
-        return html_code
-    except Exception as e:
-        logger.warning(f"HTMLの後処理中にエラーが発���しました: {e}")
-        return html_code  # エラー時は元のHTMLを返す
 # 画像から余分な空白領域をトリミングする関数
 def trim_image_whitespace(image, threshold=250, padding=10):
     """
@@ -685,10 +377,6 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
     # Font Awesomeが読み込まれない場合があるため、読み込み待機時間を長く設定
     options.add_argument("--disable-features=NetworkService")
     options.add_argument("--dns-prefetch-disable")
-    # 新規追加：CORSエラー防止のためのフラグ
-    options.add_argument("--disable-web-security")
-    options.add_argument("--allow-file-access-from-files")
     try:
         logger.info("Initializing WebDriver...")
@@ -710,10 +398,86 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
         )
         logger.info("Body element found. Waiting for potential resource loading...")
-        # リソース読み込みの待機時間（増加：8→10秒）
-        time.sleep(10)  # 十分な待機時間
-        # 5) Hide scrollbars via CSS
         try:
             driver.execute_script(
                 "document.documentElement.style.overflow = 'hidden';"
@@ -723,7 +487,7 @@ 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}")
-        # 6) Get full page dimensions accurately with improved script
         try:
             # より正確なページ寸法を取得するためのJavaScriptコード
             dimensions_script = """
@@ -777,21 +541,59 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             scroll_width = 1200
             scroll_height = 1000
             logger.warning(f"Falling back to dimensions: width={scroll_width}, height={scroll_height}")
-        # 7) 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}%)")
-        # 8) 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...")
-        # レイアウト安定化のための待機（増加：6→8秒）
-        time.sleep(8)  # 統一した待機時間
         # 外部リソースの読み込み状態を確認
         try:
@@ -807,86 +609,10 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
             # ドキュメントの読み込みが完了していない場合、追加で待機
             if resource_state['readyState'] != 'complete':
                 logger.info("Document still loading, waiting additional time...")
-                time.sleep(3)  # 待機時間増加：2→3秒
         except Exception as e:
             logger.warning(f"Error checking resource state: {e}")
-        # レイアウト崩れ修正のためのJavaScriptを実行
-        try:
-            # レイアウト安定化のための強化JavaScriptを実行
-            stabilize_script = """
-            // カードの重なりを解消するための修正
-            document.querySelectorAll('.card, .section, .info-box').forEach(card => {
-              // 位置を相対指定に強制
-              card.style.position = 'relative';
-              // z-indexをカード間で競合しないよう設定
-              card.style.zIndex = '1';
-              // マージンを強制的に設定
-              card.style.marginBottom = '2rem';
-              card.style.marginTop = '1rem';
-              // 浮動要素のクリア
-              let clearDiv = document.createElement('div');
-              clearDiv.style.clear = 'both';
-              clearDiv.style.height = '1px';
-              card.parentNode.insertBefore(clearDiv, card.nextSibling);
-            });
-            // 親要素の高さを適切に設定
-            document.querySelectorAll('.row, .container, .column').forEach(container => {
-              container.style.overflow = 'hidden';
-              container.style.height = 'auto';
-              // 子要素の合計高さを取得して設定
-              let totalHeight = 0;
-              Array.from(container.children).forEach(child => {
-                if (child.offsetHeight) {
-                  totalHeight += child.offsetHeight +
-                                 parseInt(getComputedStyle(child).marginTop || 0) +
-                                 parseInt(getComputedStyle(child).marginBottom || 0);
-                }
-              });
-              if (totalHeight > 0) {
-                container.style.minHeight = (totalHeight + 50) + 'px';
-              }
-            });
-            // テキスト量に基づいてフォントサイズを調整
-            document.querySelectorAll('.card, .section, .info-box').forEach(card => {
-              const cardContent = card.querySelector('.card-content') || card;
-              const textLength = cardContent.textContent.length;
-              // テキスト量に応じてフォントサイズを調整
-              if (textLength > 200) {
-                card.style.fontSize = '0.9em';
-              }
-              if (textLength > 300) {
-                card.style.fontSize = '0.85em';
-              }
-              // カード内の要素間の余白を確保
-              const elements = card.querySelectorAll('p, h3, h4, ul, ol, .info-item');
-              elements.forEach(el => {
-                el.style.marginBottom = '0.75rem';
-                el.style.lineHeight = '1.5';
-              });
-            });
-            // 完了通知
-            return "Layout stabilization completed";
-            """
-            stabilize_result = driver.execute_script(stabilize_script)
-            logger.info(f"Layout stabilization: {stabilize_result}")
-            # 安定化後の追加待機
-            time.sleep(2)
-        except Exception as e:
-            logger.warning(f"Failed to execute layout fix script: {e}")
         # Scroll to top just in case
         try:
             driver.execute_script("window.scrollTo(0, 0)")
@@ -895,7 +621,7 @@ 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}")
-        # 9) Take screenshot
         logger.info("Taking screenshot...")
         png = driver.get_screenshot_as_png()
         logger.info("Screenshot taken successfully.")
@@ -933,12 +659,11 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float = 6.0
                 logger.error(f"Error removing temporary file {tmp_path}: {e}")
 # --- Geminiを使った新しい関数 ---
-def text_to_screenshot(text: str, extension_percentage: float, temperature: float = 1.0,
-                       trim_whitespace: bool = True, layout_type: str = "standard") -> Image.Image:
     """テキストをGemini APIでHTMLに変換し、スクリーンショットを生成する統合関数"""
     try:
-        # 1. テキストからHTMLを生成（温度パラメータとレイアウトタイプも渡す）
-        html_code = generate_html_from_text(text, temperature, layout_type)
         # 2. HTMLからスクリーンショットを生成
         return render_fullpage_screenshot(html_code, extension_percentage, trim_whitespace)
@@ -1035,15 +760,14 @@ async def api_text_to_screenshot(request: GeminiRequest):
     テキストからHTMLインフォグラフィックを生成してスクリーンショットを返すAPIエンドポイント
     """
     try:
-        logger.info(f"テキスト→スクリーンショットAPIリクエスト受信。テキスト長さ: {len(request.text)}, 拡張率: {request.extension_percentage}%, 温度: {request.temperature}, レイアウト: {request.layout_type}")
-        # テキストからHTMLを生成してスクリーンショットを作成（温度パラメータとレイアウトタイプも渡す）
         pil_image = text_to_screenshot(
             request.text,
             request.extension_percentage,
             request.temperature,
-            request.trim_whitespace,
-            request.layout_type
         )
         if pil_image.size == (1, 1):
@@ -1064,14 +788,14 @@ async def api_text_to_screenshot(request: GeminiRequest):
 # --- Gradio Interface Definition ---
 # 入力モードの選択用Radioコンポーネント
-def process_input(input_mode, input_text, extension_percentage, temperature, trim_whitespace, layout_type):
     """入力モードに応じて適切な処理を行う"""
     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, layout_type)
 # Gradio UIの定義
 with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr.themes.Base()) as iface:
@@ -1093,84 +817,61 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
     )
     with gr.Row():
-        with gr.Column(scale=1):
-            extension_percentage = gr.Slider(
-                minimum=0,
-                maximum=30,
-                step=1.0,
-                value=6,    # デフォルト値6%
-                label="上下高さ拡張率（%）"
-            )
-            # 温度調整スライダー（テキストモード時のみ表示）
-            temperature = gr.Slider(
-                minimum=0.0,
-                maximum=1.4,
-                step=0.1,
-                value=1.0,    # デフォルト値1.0
-                label="生成時の温度（創造性）",
-                visible=False  # 最初は非表示
-            )
-            # 余白トリミングオプション
-            trim_whitespace = gr.Checkbox(
-                label="余白を自動トリミング",
-                value=True,
-                info="生成される画像から余分な空白領域を自動的に削除します"
-            )
-        with gr.Column(scale=1):
-            # 新機能：レイアウトタイプの選択（テキストモード時のみ表示）
-            layout_type = gr.Radio(
-                ["standard", "spaced", "compact"],
-                label="レイアウトタイプ",
-                value="standard",
-                info="standard: 標準レイアウト, spaced: 広々レイアウト, compact: コンパクトレイアウト",
-                visible=False
-            )
-    submit_btn = gr.Button("生成", variant="primary")
     output_image = gr.Image(type="pil", label="ページ全体のスクリーンショット")
-    # 入力モード変更時のイベント処理（テキストモード時のみ温度スライダーとレイアウトタイプを表示）
-    def update_ui_visibility(mode):
-        return [
-            {"visible": mode == "テキスト入力", "__type__": "update"},  # temperature
-            {"visible": mode == "テキスト入力", "__type__": "update"}   # layout_type
-        ]
     input_mode.change(
-        fn=update_ui_visibility,
         inputs=input_mode,
-        outputs=[temperature, layout_type]
     )
     # 生成ボタンクリック時のイベント処理
     submit_btn.click(
         fn=process_input,
-        inputs=[input_mode, input_text, extension_percentage, temperature, trim_whitespace, layout_type],
         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 で変更可能)
-    - レイアウトタイプ:
-      - standard: 標準レイアウト
-      - spaced: 広々レイアウト（テキストが重なりやすい場合に推奨）
-      - compact: コンパクトレイアウト（テキストが少ない場合に推奨）
     """)
 # --- Mount Gradio App onto FastAPI ---

 import time
 import os
 import logging
 # 正しいGemini関連のインポート
 import google.generativeai as genai
     extension_percentage: float = 6.0  # デフォルト値6%
     temperature: float = 1.0  # デフォルト値1.0の温度パラメータ
     trim_whitespace: bool = True  # 余白トリミングオプション（デフォルト有効）
 class ScreenshotRequest(BaseModel):
     """スクリーンショットリクエストモデル"""
     extension_percentage: float = 6.0  # デフォルト値6%
     trim_whitespace: bool = True  # 余白トリミングオプション（デフォルト有効）
+# HTMLのFont Awesomeレイアウトを改善する関数
+def enhance_font_awesome_layout(html_code):
+    """Font Awesomeレイアウトを改善するCSSを追加"""
+    # CSSを追加
+    fa_fix_css = """
+    <style>
+    /* Font Awesomeアイコンのレイアウト修正 */
+    [class*="fa-"] {
+        display: inline-block !important;
+        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;
+        vertical-align: middle !important;
+    }
+    </style>
+    """
+    # headタグがある場合はその中に追加
+    if '<head>' in html_code:
+        return html_code.replace('</head>', f'{fa_fix_css}</head>')
+    # HTMLタグがある場合はその後に追加
+    elif '<html' in html_code:
+        head_end = html_code.find('</head>')
+        if head_end > 0:
+            return html_code[:head_end] + fa_fix_css + html_code[head_end:]
+        else:
+            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>'
+def generate_html_from_text(text, temperature=1.0):
     """テキストからHTMLを生成する"""
     try:
         # APIキーの取得と設定
         # Gemini APIの設定
         genai.configure(api_key=api_key)
+        # システムプロンプト（リクエスト例と同じものを使用）
+        system_instruction = """# グラフィックレコーディング風インフォグラフィック変換プロンプト V2
 ## 目的
 以下の内容を、超一流デザイナーが作成したような、日本語で完璧なグラフィックレコーディング風のHTMLインフォグラフィックに変換してください。情報設計とビジュアルデザインの両面で最高水準を目指します。
 手書き風の図形やFont Awesomeアイコンを大きく活用して内容を視覚的かつ直感的に表現します。
 - フォント指定：
 ```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');
 </style>
 ```
 ### 5. レイアウト
 - 横幅は100%
 - 重要な要素は中央寄り、補足情報は周辺部に配置
+### 重要なレイアウト注意事項
+- Font Awesomeアイコンと文字が重ならないよう適切なマージンを設定する
+- アイコンには必ず`margin-right: 10px;`や`margin-left: 10px;`などの余白を設定する
+- アイコンと文字列は必ず`display: inline-block;`または`display: inline-flex;`で配置し、`align-items: center;`を使用する
+- 複数のアイコンを配置する場合は十分な余白を確保する
+- レイアウト崩れを防ぐために、親要素には`overflow: hidden;`を適用する
+- アイコンの配置はコンテンツフローを妨げないよう注意する
+- カード内でのアイコン配置は、絶対位置指定ではなく相対位置で設定する
+- モバイル表示も考慮し、レスポンシブ対応を徹底する
 ## グラフィックレコーディング表現技法
 - テキストと視覚要素のバランスを重視（文字情報の50%以上をFont Awesomeアイコンで視覚的に補完）
 ーーー＜ユーザーが入力（または添付）＞ーーー"""
         # モデルを初期化して処理
+        logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}")
         # モデル初期化とフォールバック処理
         try:
             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:
         logger.error(f"HTML生成中にエラーが発生: {e}", exc_info=True)
         raise Exception(f"Gemini APIでのHTML生成に失敗しました: {e}")
 # 画像から余分な空白領域をトリミングする関数
 def trim_image_whitespace(image, threshold=250, padding=10):
     """
     # Font Awesomeが読み込まれない場合があるため、読み込み待機時間を長く設定
     options.add_argument("--disable-features=NetworkService")
     options.add_argument("--dns-prefetch-disable")
     try:
         logger.info("Initializing WebDriver...")
         )
         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';"
         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 = """
             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 => {
+                let lastHeight = document.body.offsetHeight;
+                let lastWidth = document.body.offsetWidth;
+                let stableCount = 0;
+                const checkStability = () => {
+                    const currentHeight = document.body.offsetHeight;
+                    const currentWidth = document.body.offsetWidth;
+                    if (currentHeight === lastHeight && currentWidth === lastWidth) {
+                        stableCount++;
+                        if (stableCount >= 3) {  // 3回連続で同じなら安定と判断
+                            resolve(true);
+                            return;
+                        }
+                    } else {
+                        stableCount = 0;
+                        lastHeight = currentHeight;
+                        lastWidth = currentWidth;
+                    }
+                    setTimeout(checkStability, 200);  // 200ms間隔でチェック
+                };
+                checkStability();
+            });
+            """
+            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")
+        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))
         # 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:
             # ドキュメントの読み込みが完了していない場合、追加で待機
             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)")
         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.")
                 logger.error(f"Error removing temporary file {tmp_path}: {e}")
 # --- Geminiを使った新しい関数 ---
+def text_to_screenshot(text: str, extension_percentage: float, temperature: float = 1.0, 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)
     テキストから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):
 # --- 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:
     )
     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.4,
+            step=0.1,
+            value=1.0,    # デフォルト値1.0
+            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 ---