Update app.py

app.py

@@ -178,93 +178,65 @@ 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 環境変数が設定されていません")
 
-        # モデル名の取得（環境変数から、なければデフォルト値）
         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)
 
-        # モデル初期化
-        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に下げて候補を絞る
+        # --- 共通の生成パラメータ ---
+        base_cfg = {
+            "temperature": temperature,
+            "top_p":      0.7,
+            "top_k":      20,
             "max_output_tokens": 8192,
-            "candidate_count": 1         # 候補は1つだけ生成
+            "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"
-            }
-        ]
+        # --- 2.5 Flash だけ thinking_budget=0 を追加 ---
+        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)
+            )
+        else:
+            cfg = genai.types.GenerateContentConfig(**base_cfg)
 
-        # プロンプト構築
-        prompt = f"{system_instruction}\n\n{text}"
+        model   = genai.GenerativeModel(model_name)
+        prompt  = f"{system_instruction}\n\n{text}"
 
-        # コンテンツ生成
         response = model.generate_content(
             prompt,
-            generation_config=generation_config,
-            safety_settings=safety_settings
+            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"},
+            ],
         )
 
-        # レスポンスから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
+        # --- レスポンス処理は元コードと同じ ---
+        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
         else:
-            # HTMLタグが見つからない場合、レスポンス全体を返す
-            logger.warning("レスポンスから ```html ``` タグが見つかりませんでした。全テキストを返します。")
-            return raw_response
+            logger.warning("```html``` タグが見つからず。全文を返します")
+            return raw
 
     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}")
 
 # 画像から余分な空白領域をトリミングする関数