Update app.py

app.py

@@ -8,11 +8,7 @@ from PIL import Image
 from io import BytesIO
 
 # --- 1. 設定: APIキーの読み込み ---
-# 環境変数 `TOGETHER_API_KEY` からAPIキーを読み込みます。
-# 事前に `export TOGETHER_API_KEY="あなたのAPIキー"` のように設定してください。
 api_key = os.environ.get("TOGETHER_API_KEY")
-
-# APIキーがあればTogetherAIクライアントを初期化
 if api_key:
     together_client = together.Together(api_key=api_key)
 else:
@@ -21,26 +17,13 @@ else:
 # 使用する画像生成モデル
 IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell-Free"
 
-
 # --- 2. 中核機能: 画像生成関数 ---
 def generate_image_from_prompt(prompt: str, history: list):
     """
     TogetherAI APIを使用して画像を生成し、履歴リストの先頭に追加します。
-
-    Args:
-        prompt (str): 画像生成のためのテキストプロンプト。
-        history (list): これまでの生成結果を格納したリスト。
-                        各要素は (PIL.Image, "プロンプト") のタプルです。
-
-    Returns:
-        tuple: 更新された履歴リスト (Gallery表示用) と、
-               更新された履歴リスト (State保持用) のタプル。
     """
-    # APIクライアントが設定されていない場合はエラーを返す
     if not together_client:
         raise gr.Error("TogetherAIのAPIクライアントが設定されていません。環境変数 TOGETHER_API_KEY を確認してください。")
-
-    # プロンプトが空の場合はエラーを返す
     if not prompt or not prompt.strip():
         raise gr.Error("画像を生成するには、プロンプトを入力してください。")
 
@@ -56,25 +39,36 @@ def generate_image_from_prompt(prompt: str, history: list):
             width=1024
         )
         
-        # レスポンスにはbase64形式で画像データが含まれている
-        image_b64 = response.data[0].b64_json
-        image_data = base64.b64decode(image_b64)
+        generated_images = []
+        # response.data にはリクエストした全ての画像データが含まれる
+        for image_data_item in response.data:
+            image_b64 = image_data_item.b64_json
+            
+            # ▼▼▼ 修正箇所: 安全チェックを追加 ▼▼▼
+            # image_b64がNoneでない（正常にデータが返された）場合のみ処理する
+            if image_b64:
+                image_data = base64.b64decode(image_b64)
+                image = Image.open(BytesIO(image_data))
+                # Gallery用に (image, caption) のタプルを作成
+                generated_images.append((image, prompt))
+
+        # 正常に生成された画像が1枚もなかった場合
+        if not generated_images:
+            raise gr.Error("画像の生成に失敗しました。プロンプトがコンテンツフィルターに抵触した可能性があります。別のプロンプトで試してください。")
+
+        # 新しい画像のリストを、既存の履歴の先頭に追加
+        new_history = generated_images + history
         
-        # バイトデータからPILイメージオブジェクトを作成
-        image = Image.open(BytesIO(image_data))
-
-        # 新しい画像とプロンプトを履歴リストの先頭に追加
-        # Galleryコンポーネントは (image, caption) のタプルのリストを期待します
-        new_history = [(image, prompt)] + history
-
-        print("画像の生成が完了しました。")
+        print(f"{len(generated_images)}枚の画像の生成が完了しました。")
         return new_history, new_history
 
     except Exception as e:
-        print(f"エラーが発生しました: {e}")
+        # 既にgr.Errorの場合はそのまま表示
+        if isinstance(e, gr.Error):
+            raise e
+        print(f"予期せぬエラーが発生しました: {e}")
         raise gr.Error(f"画像の生成に失敗しました。エラー詳細: {e}")
 
-
 # --- 3. Gradio UIの構築 ---
 with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px; margin: auto;}") as app:
     gr.Markdown(
@@ -85,39 +79,27 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px;
         """
     )
     gr.Markdown("---")
-
-    # 生成された画像の履歴を保持するためのState
-    # 履歴は [(PIL.Image, "プロンプト1"), (PIL.Image, "プロンプト2"), ...] のリスト
+    
     history_state = gr.State([])
 
     with gr.Column():
-        # 入力コンポーネント
         prompt_input = gr.Textbox(
             label="画像生成プロンプト",
             placeholder="例: 月面にいる、写真のようにリアルな猫の宇宙飛行士",
             lines=4,
             interactive=True,
         )
-        
-        # APIキーが設定されていない場合に警告を表示
         if not together_client:
              gr.Warning("環境変数 `TOGETHER_API_KEY` が見つかりません。画像生成機能を利用するにはAPIキーを設定してください。")
-
         generate_btn = gr.Button(
-            "画像生成 ✨",
+            "画像生成 ✨ (4枚)",
             variant="primary",
-            # APIクライアントがなければボタンを無効化
             interactive=(together_client is not None)
         )
 
     gr.Markdown("---")
-    
-    # ▼▼▼ 修正箇所 ▼▼▼
-    # gr.subheader を gr.Markdown("## ...") に変更
     gr.Markdown("## 生成された画像")
     
-    # 出力コンポーネント
-    # Galleryは、キャプション付きの画像リストを表示するのに最適です
     image_gallery = gr.Gallery(
         label="生成結果",
         show_label=False,
@@ -128,16 +110,13 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px;
         preview=True
     )
     
-    # ボタンのクリックイベントと生成関数を接続
     generate_btn.click(
         fn=generate_image_from_prompt,
         inputs=[prompt_input, history_state],
         outputs=[image_gallery, history_state],
-        # API呼び出し中にプログレスインジケーターを表示
         show_progress="full"
     )
     
-    # ユーザーのための入力例
     gr.Examples(
         examples=[
             "A high-resolution photo of a futuristic city with flying cars",
@@ -148,7 +127,5 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px;
         inputs=prompt_input
     )
 
-# `app.launch()` はGradioアプリを起動します
-# ローカルサーバーが立ち上がり、ブラウザでアクセスできるURLがターミナルに表示されます
 if __name__ == "__main__":
     app.launch(mcp_server=True)