Update app.py

app.py

@@ -16,9 +16,8 @@ import time
 import os
 import logging
 
-# 新しいGemini関連のインポート
+# 新しいGemini関連のインポート - 最新バージョン修正版
 import google.generativeai as genai
-from google.generativeai import types
 
 # ロギング設定
 logging.basicConfig(level=logging.INFO)
@@ -30,20 +29,16 @@ class GeminiRequest(BaseModel):
     text: str
     extension_percentage: float = 8.0  # Default value same as Gradio slider
 
-def init_gemini_client():
-    """Gemini APIクライアントを初期化する"""
-    api_key = os.environ.get("GEMINI_API_KEY")
-    if not api_key:
-        logger.error("GEMINI_API_KEY 環境変数が設定されていません")
-        raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
-    
-    return genai.Client(api_key=api_key)
-
 def generate_html_from_text(text):
     """テキストからHTMLを生成する"""
     try:
-        client = init_gemini_client()
-        model = "gemini-2.5-pro-exp-03-25"  # リクエスト例と同じモデルを使用
+        # APIキーの設定
+        api_key = os.environ.get("GEMINI_API_KEY")
+        if not api_key:
+            logger.error("GEMINI_API_KEY 環境変数が設定されていません")
+            raise ValueError("GEMINI_API_KEY 環境変数が設定されていません")
+        
+        genai.configure(api_key=api_key)
         
         # システムプロンプト（リクエスト例と同じものを使用）
         system_instruction = """# グラフィックレコーディング風インフォグラフィック変換プロンプト V2
@@ -143,26 +138,32 @@ def generate_html_from_text(text):
 ## 変換する文章/記事
 ーーー＜ユーザーが入力（または添付）＞ーーー"""
 
-        # 実際のユーザー入力を指定
-        contents = [
-            types.Content(
-                role="user", 
-                parts=[types.Part.from_text(text=text)]
-            )
-        ]
-        
         # 生成設定
-        generate_content_config = types.GenerateContentConfig(
-            response_mime_type="text/plain",
-            system_instruction=[types.Part.from_text(text=system_instruction)],
+        generation_config = {
+            "temperature": 0.7,
+            "top_p": 0.95,
+            "top_k": 64,
+            "max_output_tokens": 8192,
+        }
+        
+        # 利用可能なモデルを使用（gemini-1.5-proが最新の安定バージョン）
+        model = genai.GenerativeModel(
+            model_name="gemini-1.5-pro",
+            generation_config=generation_config
         )
         
         # HTMLを生成
         logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}")
-        response = client.models.generate_content(
-            model=model,
-            contents=contents,
-            config=generate_content_config,
+        
+        # システムインストラクションとユーザープロンプトを設定
+        response = model.generate_content(
+            contents=[
+                {
+                    "role": "user",
+                    "parts": [text]
+                }
+            ],
+            system_instruction=system_instruction
         )
         
         # レスポンスからHTMLを抽出