Update app.py

app.py

@@ -38,6 +38,10 @@ def generate_html_from_text(text):
             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)
         
@@ -142,19 +146,14 @@ def generate_html_from_text(text):
         # モデルを初期化して処理
         logger.info(f"Gemini APIにリクエストを送信: テキスト長さ = {len(text)}")
         
-        # 利用可能なモデルの確認
+        # モデル初期化とフォールバック処理
         try:
-            models = genai.list_models()
-            logger.info(f"利用可能なモデル: {[m.name for m in models if 'gemini' in m.name]}")
+            model = genai.GenerativeModel(model_name)
         except Exception as e:
-            logger.warning(f"モデル一覧取得エラー: {e}")
-        
-        # モデル選択
-        try:
-            model = genai.GenerativeModel('gemini-2.0-flash-thinking-exp-01-21')
-        except Exception as e:
-            logger.warning(f"gemini-2.0-flash-thinking-exp-01-21モデル利用不可: {e}, gemini-proを試します")
-            model = genai.GenerativeModel('gemini-pro')
+            # 指定されたモデルが使用できない場合はフォールバック
+            fallback_model = "gemini-pro"
+            logger.warning(f"{model_name}が利用できません: {e}, フォールバックモデル{fallback_model}を使用します")
+            model = genai.GenerativeModel(fallback_model)
             
         # 生成設定
         generation_config = {
@@ -212,8 +211,8 @@ def render_fullpage_screenshot(html_code: str, extension_percentage: float, is_g
 
     # Gemini生成コンテンツの場合、拡張率を調整
     if is_gemini_content:
-        # 最低でも10%の拡張を確保
-        extension_percentage = max(extension_percentage, 10.0)
+        # 最低でも20%の拡張を確保
+        extension_percentage = max(extension_percentage, 20.0)
         logger.info(f"Gemini生成コンテンツ用に拡張率を調整: {extension_percentage}%")
 
     # 1) Save HTML code to a temporary file
@@ -570,7 +569,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
         minimum=0,
         maximum=30,  # より高い値を許可
         step=1.0,
-        value=8.0,    # デフォルト値を増加
+        value=15,    # デフォルト値を増加
         label="上下高さ拡張率（%）"
     )
     
@@ -583,10 +582,15 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
         outputs=output_image
     )
     
-    gr.Markdown("""
+    # 環境変数情報を表示
+    gemini_model = os.environ.get("GEMINI_MODEL", "gemini-1.5-pro")
+    gr.Markdown(f"""
     ## APIエンドポイント
     - `/api/screenshot` - HTMLコードからスクリーンショットを生成 
     - `/api/text-to-screenshot` - テキストからインフォグラフィックスクリーンショットを生成
+    
+    ## 設定情報
+    - 使用モデル: {gemini_model} (環境変数 GEMINI_MODEL で変更可能)
     """)
 
 # --- Mount Gradio App onto FastAPI ---