Update app.py

app.py

@@ -15,9 +15,9 @@ import tempfile
 import time
 import os
 import logging
-from huggingface_hub import hf_hub_download  # 追加: HuggingFace Hubからファイルを直接ダウンロード
+from huggingface_hub import hf_hub_download
 
-# 正しいGemini関連のインポート
+# 新しいGemini関連のインポート（最新のGoogle Gen AI SDK）
 from google import genai
 from google.genai import types
 
@@ -194,60 +194,75 @@ def generate_html_from_text(text, temperature=0.3, style="standard"):
         # 指定されたスタイルのシステムインストラクションを読み込む
         system_instruction = load_system_instruction(style)
 
-        # モデル初期化と情報ログ
-        logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}, スタイル = {style}")
-        
-        # Gemini APIクライアントの初期化 (新しいSDK方式)
+        # Gemini APIクライアントの初期化（最新SDKに対応）
         client = genai.Client(api_key=api_key)
         
+        logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}, 温度 = {temperature}, スタイル = {style}")
+
         # プロンプト構築
         prompt = f"{system_instruction}\n\n{text}"
         
-        # 生成設定
-        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に設定
+        # 共通の生成設定
+        generation_config = {
+            "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を設定
         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
-        )
-        
+            logger.info("gemini-2.5-flash-preview-04-17モデル検出: thinking_budget=0 を設定")
+            # 新しいSDKでの正しい実装方法
+            response = client.models.generate_content(
+                model=model_name,
+                contents=prompt,
+                config=types.GenerateContentConfig(
+                    temperature=temperature,
+                    top_p=0.7,
+                    top_k=20,
+                    max_output_tokens=8192,
+                    candidate_count=1,
+                    thinking_config=types.ThinkingConfig(
+                        thinking_budget=0  # 思考機能を無効化
+                    )
+                )
+            )
+        else:
+            # 他のモデルの場合は従来通りの呼び出し
+            response = client.models.generate_content(
+                model=model_name,
+                contents=prompt,
+                generation_config=generation_config
+            )
+
         # レスポンスからHTMLを抽出
         raw_response = response.text
-        
-        # HTMLタグ部分だけを抽出（`````` の間）
-        html_start = raw_response.find("```
+
+        # 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_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
         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):
     """