Update app.py

app.py

@@ -187,15 +187,15 @@ driver_pool = WebDriverPool(max_drivers=int(os.environ.get("MAX_WEBDRIVERS", "3"
 class GeminiRequest(BaseModel):
     """Geminiへのリクエストデータモデル"""
     text: str
-    extension_percentage: float = 10.0  # デフォルト値10%
-    temperature: float = 0.5  # デフォルト値を0.5に設定
+    extension_percentage: float = 15.0  # デフォルト値15%
+    temperature: float = 1.0  # デフォルト値を1.0に設定
     trim_whitespace: bool = True  # 余白トリミングオプション（デフォルト有効）
     style: str = "standard"  # デフォルトはstandard
 
 class ScreenshotRequest(BaseModel):
     """スクリーンショットリクエストモデル"""
     html_code: str
-    extension_percentage: float = 10.0  # デフォルト値10%
+    extension_percentage: float = 15.0  # デフォルト値15%
     trim_whitespace: bool = True  # 余白トリミングオプション（デフォルト有効）
     style: str = "standard"  # デフォルトはstandard
 
@@ -295,7 +295,7 @@ def load_system_instruction(style="standard"):
     """
     try:
         # 有効なスタイル一覧
-        valid_styles = ["standard", "cute", "resort", "cool", "dental", "school"]
+        valid_styles = ["standard", "cute", "resort", "cool", "dental", "school","KOKUGO"]
 
         # スタイルの検証
         if style not in valid_styles:
@@ -861,7 +861,7 @@ def text_to_screenshot_parallel(text: str, extension_percentage: float, temperat
         return Image.new('RGB', (1, 1), color=(0, 0, 0)), None  # エラー時は黒画像とNone URL
 
 # 従来の非並列版も残す（互換性のため）
-def text_to_screenshot(text: str, extension_percentage: float, temperature: float = 0.3,
+def text_to_screenshot(text: str, extension_percentage: float, temperature: float = 1.0,
                     trim_whitespace: bool = True, style: str = "standard") -> tuple:
     """テキストをGemini APIでHTMLに変換し、スクリーンショットを生成する統合関数（レガシー版）"""
     # 並列処理版を呼び出す
@@ -1048,7 +1048,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
         with gr.Column(scale=1):
             # スタイル選択ドロップダウン
             style_dropdown = gr.Dropdown(
-                choices=["standard", "cute", "resort", "cool", "dental", "school"],
+                choices=["standard", "cute", "resort", "cool", "dental", "school","KOKUGO"],
                 value="standard",
                 label="デザインスタイル",
                 info="テキスト→HTML変換時のデザインテーマを選択します",
@@ -1060,7 +1060,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
                 minimum=0,
                 maximum=30,
                 step=1.0,
-                value=10,    # デフォルト値10%
+                value=15,    # デフォルト値15%
                 label="上下高さ拡張率（%）"
             )
 
@@ -1069,7 +1069,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
                 minimum=0.0,
                 maximum=1.0,
                 step=0.1,
-                value=0.5,    # デフォルト値を0.5に設定
+                value=1.0,    # デフォルト値を1.0に設定
                 label="生成時の温度（低い=一貫性高、高い=創造性高）",
                 visible=False  # 最初は非表示
             )
@@ -1124,7 +1124,7 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
     - 使用モデル: {gemini_model} (環境変数 GEMINI_MODEL で変更可能)
     - HuggingFaceリポジトリ: {hf_repo} (環境変数 HF_REPO_ID で変更可能)
     - WebDriverプール最大数: {driver_pool.max_drivers} (環境変数 MAX_WEBDRIVERS で変更可能)
-    - 対応スタイル: standard, cute, resort, cool, dental
+    - 対応スタイル: standard, cute, resort, cool, dental, KOKUGO
     """)
 
 # --- Mount Gradio App onto FastAPI ---