Update app.py

app.py

@@ -180,64 +180,96 @@ def load_system_instruction(style="standard"):
 def generate_html_from_text(text, temperature=0.3, style="standard"):
     """テキストからHTMLを生成する"""
     try:
-        # --- API キー & モデル判定 ---
+        # 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)
-
-        # --- GenerationConfig (thinking_budget は含めない) ---
+        
+        # モデル初期化
+        logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}, スタイル = {style}")
+        
+        # モデル初期化
+        model = genai.GenerativeModel(model_name)
+        
+        # 生成設定 - ばらつきを減らすために設定を調整
         generation_config = {
-            "temperature": temperature,
-            "top_p": 0.7,
-            "top_k": 20,
+            "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つだけ生成
         }
-
-        # --- 追加: Flash‑preview 用の追加 kwargs ---
-        extra_kwargs = {}
-        if model_name == "gemini-2.5-flash-preview-04-17":
-            extra_kwargs["thinking_budget"] = 0
-            logger.info("thinking_budget=0 をトップレベルで付与しました")
-
-        # --- モデル初期化 & 生成 ---
-        model = genai.GenerativeModel(model_name)
-
+        
+        # 安全設定 - デフォルトの安全設定を使用
+        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=[
-                {"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"},
-            ],
-            **extra_kwargs  # ← ここで thinking_budget を渡す
+            safety_settings=safety_settings
         )
-
-        # --- レスポンス処理は従来どおり ---
+        
+        # レスポンスからHTMLを抽出
         raw_response = response.text
-        html_start = raw_response.find("```html")
-        html_end   = raw_response.rfind("```")
-
+        
+        # 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_code = raw_response[html_start + 7:html_end].strip()
+            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(f"HTMLの生成に成功 (len={len(html_code)})")
+            logger.info("Font Awesomeレイアウトの最適化を適用しました")
+            
             return html_code
         else:
-            logger.warning("```html``` タグが見つかりませんでした。全文を返します。")
+            # 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}")