Update app.py

app.py

@@ -18,7 +18,8 @@ import logging
 from huggingface_hub import hf_hub_download  # 追加: HuggingFace Hubからファイルを直接ダウンロード
 
 # 正しいGemini関連のインポート
-import google.generativeai as genai
+from google import genai
+from google.genai import types
 
 # ロギング設定
 logging.basicConfig(level=logging.INFO)
@@ -190,83 +191,63 @@ def generate_html_from_text(text, temperature=0.3, style="standard"):
         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に下げて候補を絞る
-            "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"
-            }
-        ]
-
+        
+        # Gemini APIクライアントの初期化 (新しいSDK方式)
+        client = genai.Client(api_key=api_key)
+        
         # プロンプト構築
         prompt = f"{system_instruction}\n\n{text}"
-
-        # コンテンツ生成
-        response = model.generate_content(
-            prompt,
-            generation_config=generation_config,
-            safety_settings=safety_settings
+        
+        # 生成設定
+        generation_config = types.GenerateContentConfig(
+            temperature=temperature,
+            top_p=0.7,
+            top_k=20,
+            max_output_tokens=8192,
+            candidate_count=1
         )
-
+        
+        # モデル名がgemini-2.5-flash-preview-04-17の場合、thinking_budgetを0に設定
+        if model_name == "gemini-2.5-flash-preview-04-17":
+            logger.info("Gemini 2.5 Flash検出: thinking_budgetを0に設定します")
+            generation_config.thinking_config = types.ThinkingConfig(thinking_budget=0)
+        
+        # コンテンツ生成 - 新しいSDKの構文に合わせて更新
+        response = client.models.generate_content(
+            model=model_name,
+            contents=prompt,
+            config=generation_config
+        )
+        
         # レスポンスからHTMLを抽出
         raw_response = response.text
-
-        # HTMLタグ部分だけを抽出（```html と ``` の間）
-        html_start = raw_response.find("```html")
+        
+        # HTMLタグ部分だけを抽出（`````` の間）
+        html_start = raw_response.find("```
         html_end = raw_response.rfind("```")
-
         if html_start != -1 and html_end != -1 and html_start < html_end:
-            html_start += 7  # "```html" の長さ分進める
+            html_start += 7  # "```
             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 ``` タグが見つかりませんでした。全テキストを返します。")
+            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):
     """