tomo2chin2 commited on
Commit
4882a8a
·
verified ·
1 Parent(s): 8d9ed08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -1
app.py CHANGED
@@ -181,6 +181,11 @@ def generate_html_from_text(text: str, temperature=0.5, style="standard") -> str
181
  candidate_count=1
182
  )
183
 
 
 
 
 
 
184
  # Generate content
185
  response = client.models.generate_content(
186
  model=model_name,
@@ -341,8 +346,14 @@ with gr.Blocks(title="Full Page Screenshot (テキスト変換対応)", theme=gr
341
 
342
  btn.click(process_input, [mode, text, ext, temp, trim, style], out)
343
 
 
 
 
 
 
 
344
  gr.Markdown(f"**API** `/api/screenshot`, `/api/text-to-screenshot`    "
345
- f"使用モデル: `{os.getenv('GEMINI_MODEL','gemini-1.5-pro')}`")
346
 
347
  # ===============================================================
348
  # Gradio を /gradio にマウントし、明示リダイレクトを追加
 
181
  candidate_count=1
182
  )
183
 
184
+ # 新機能: Gemini 2.5 Flash Preview モデルの場合、思考モードをオフにする
185
+ if model_name == "gemini-2.5-flash-preview-04-17":
186
+ logger.info("gemini-2.5-flash-preview-04-17 モデル検出: 思考モードをオフに設定")
187
+ config.thinking_config = types.ThinkingConfig(thinking_budget=0)
188
+
189
  # Generate content
190
  response = client.models.generate_content(
191
  model=model_name,
 
346
 
347
  btn.click(process_input, [mode, text, ext, temp, trim, style], out)
348
 
349
+ # モデル名を表示し、Gemini 2.5の場合は思考モードのステータスも表示
350
+ model_name = os.getenv('GEMINI_MODEL', 'gemini-1.5-pro')
351
+ thinking_status = ""
352
+ if model_name == "gemini-2.5-flash-preview-04-17":
353
+ thinking_status = "(思考モード: オフ)"
354
+
355
  gr.Markdown(f"**API** `/api/screenshot`, `/api/text-to-screenshot`    "
356
+ f"使用モデル: `{model_name}` {thinking_status}")
357
 
358
  # ===============================================================
359
  # Gradio を /gradio にマウントし、明示リダイレクトを追加