Update app.py

app.py

@@ -9,40 +9,25 @@ import os
 from google import genai
 from google.genai import types
 
-MAX_SEED = np.iinfo(np.int32).max
 
 def infer(
     google_api_key,
     prompt,
-    seed,
-    randomize_seed,
     progress=gr.Progress(track_tqdm=True),
 ):
     # Google APIキーが入力されているかチェック
     if not google_api_key:
         raise gr.Error("Google API Key is required. Please enter your key.")
-    
-    # APIキーを設定
-    # try:
-    #     genai.configure(api_key=google_api_key)
-    # except Exception as e:
-    #     raise gr.Error(f"Failed to configure Google API: {e}")
-
-    if randomize_seed:
-        seed = random.randint(0, MAX_SEED)
-        
-    # --- ここから画像生成コードを組み込み ---
-    try:
 
+    try:
+        # Clientの初期化時にAPIキーを渡す
         client = genai.Client(api_key=google_api_key)
-        # シード値をリクエストに含める（APIが対応している場合）
-        # 注: generate_contentが直接seedをサポートしていない場合、プロンプトに含めるなどの工夫が必要になることがあります。
-        # ここでは、APIがseedを直接受け取らないため、プロンプトに情報を追加する例を示します。
-        # 実際の生成結果のランダム性はAPIの実装に依存します。
-        full_prompt = f"{prompt} (Seed: {seed})"
         
+        # プロンプトにシード情報を含めることで、再現性のヒントとする
+        full_prompt = f"{prompt}"
+
         response = client.models.generate_content(
-            model="gemini-1.5-flash", # モデル名を更新
+            model="gemini-2.5-flash-image-preview",
             contents=[full_prompt],
         )
 
@@ -52,12 +37,10 @@ def infer(
             if part.inline_data is not None:
                 image_part = part
                 break
-        
+
         if image_part:
             image = Image.open(BytesIO(image_part.inline_data.data))
-            # 生成された画像を保存する場合は以下の行のコメントを外します
-            # image.save(f"generated_image_seed_{seed}.png")
-            return image, seed
+            return image
         else:
             # テキスト部分があればエラーとして表示
             text_response = ""
@@ -68,7 +51,6 @@ def infer(
 
     except Exception as e:
         raise gr.Error(f"An error occurred during image generation: {e}")
-    # --- ここまで ---
 
 examples = [
     "Create a picture of a nano banana dish in a fancy restaurant with a Gemini theme",
@@ -86,14 +68,14 @@ css = """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
         gr.Markdown(" # Text-to-Image with Google Gemini")
-        
+
         google_api_key = gr.Textbox(
             label="Google API Key",
             placeholder="Enter your Google API Key here...",
             type="password",
             show_label=True,
         )
-        
+
         with gr.Row():
             prompt = gr.Text(
                 label="Prompt",
@@ -103,20 +85,11 @@ with gr.Blocks(css=css) as demo:
                 container=False,
             )
             run_button = gr.Button("Run", scale=0, variant="primary")
-            
+
         result = gr.Image(label="Result", show_label=False)
-        
-        # Gemini APIに不要な設定項目を削除し、シード値のみ残す
-        with gr.Accordion("Advanced Settings", open=False):
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=0,
-            )
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-                
+
+        # --- Advanced Settingsを削除 ---
+
         gr.Examples(examples=examples, inputs=[prompt])
 
     gr.on(
@@ -125,10 +98,8 @@ with gr.Blocks(css=css) as demo:
         inputs=[
             google_api_key,
             prompt,
-            seed,
-            randomize_seed,
         ],
-        outputs=[result, seed],
+        outputs=[result],
     )
 
 if __name__ == "__main__":