Update app.py

app.py

@@ -52,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-"], 
+    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 {
         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;
@@ -178,65 +178,106 @@ def load_system_instruction(style="standard"):
         raise ValueError(error_msg)
 
 def generate_html_from_text(text, temperature=0.3, style="standard"):
-    """テキストからHTMLを生成して返す"""
+    """テキストからHTMLを生成する"""
     try:
+        # APIキーの取得と設定
         api_key = os.environ.get("GEMINI_API_KEY")
         if not api_key:
+            logger.error("GEMINI_API_KEY 環境変数が設定されていません")
             raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
 
+        # モデル名の取得（環境変数から、なければデフォルト値）
+        # ユーザーの要望通り、デフォルト値は gemini-1.5-pro のままにし、環境変数で指定されたモデルを使用する
         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 = load_system_instruction(style)
 
-        # --- 共通の生成パラメータ ---
-        base_cfg = {
-            "temperature": temperature,
-            "top_p":      0.7,
-            "top_k":      20,
+        # モデル初期化
+        logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}, スタイル = {style}")
+
+        # モデル初期化
+        model = genai.GenerativeModel(model_name)
+
+        # 生成設定 - ばらつきを減らすために設定を調整
+        generation_config = {
+            "temperature": temperature,  # より低い温度を設定
+            "top_p": 0.7,                # 0.95から0.7に下げて出力の多様性を制限
+            "top_k": 20,                 # 64から20に下げて候補を絞る
             "max_output_tokens": 8192,
-            "candidate_count":   1,
+            "candidate_count": 1         # 候補は1つだけ生成
         }
 
-        # --- 2.5 Flash だけ thinking_budget=0 を追加 ---
+        # 安全設定 - デフォルトの安全設定を使用
+        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}"
+
+        # コンテンツ生成
+        # モデルが gemini-2.5-flash-preview-04-17 の場合のみ thinking_budget=0 を tool_config に含める
         if model_name == "gemini-2.5-flash-preview-04-17":
-            logger.info("thinking_budget=0 を設定します")
-            cfg = genai.types.GenerateContentConfig(
-                **base_cfg,
-                thinking_config=genai.types.ThinkingConfig(thinking_budget=0)
+            logger.info(f"使用モデルが {model_name} なので tool_config に thinking_budget=0 を追加します。")
+            response = model.generate_content(
+                prompt,
+                generation_config=generation_config,
+                safety_settings=safety_settings,
+                tool_config={"code_execution": {"enable": True, "thinking_budget": 0}}
             )
         else:
-            cfg = genai.types.GenerateContentConfig(**base_cfg)
-
-        model   = genai.GenerativeModel(model_name)
-        prompt  = f"{system_instruction}\n\n{text}"
-
-        response = model.generate_content(
-            prompt,
-            config=cfg,
-            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"},
-            ],
-        )
+            logger.info(f"使用モデルが {model_name} なので thinking_budget は追加しません。")
+            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レイアウトの最適化を適用しました")
 
-        # --- レスポンス処理は元コードと同じ ---
-        raw = response.text
-        start = raw.find("```html")
-        end   = raw.rfind("```")
-        if start != -1 and end != -1 and start < end:
-            html = raw[start + 7:end].strip()
-            html = enhance_font_awesome_layout(html)
-            return html
+            return html_code
         else:
-            logger.warning("```html``` タグが見つからず。全文を返します")
-            return raw
+            # HTMLタグが見つからない場合、レスポンス全体を返す
+            logger.warning("レスポンスから ```html ``` タグが見つかりませんでした。全テキストを返します。")
+            return raw_response
 
     except Exception as e:
-        logger.error(f"HTML生成中にエラー: {e}", exc_info=True)
+        logger.error(f"HTML生成中にエラーが発生: {e}", exc_info=True)
         raise Exception(f"Gemini APIでのHTML生成に失敗しました: {e}")
 
 # 画像から余分な空白領域をトリミングする関数
@@ -741,6 +782,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
     )
 
     # 環境変数情報を表示
+    # デフォルトは gemini-1.5-pro のままにし、環境変数で指定されたモデルが表示されるようにする
     gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
     gr.Markdown(f"""
     ## APIエンドポイント