Update app.py

app.py

@@ -1,13 +1,12 @@
 # app.py
 
 import gradio as gr
-#from together import Together
 import together
-
 import os
 import base64
 from PIL import Image
 from io import BytesIO
+import requests  # ▼▼▼ 修正箇所: requestsライブラリをインポート ▼▼▼
 
 # --- 1. 設定: APIキーの読み込み ---
 api_key = os.environ.get("TOGETHER_API_KEY")
@@ -21,9 +20,6 @@ IMAGE_MODEL = "black-forest-labs/FLUX.1-schnell-Free"
 
 # --- 2. 中核機能: 画像生成関数 ---
 def generate_image_from_prompt(prompt: str, history: list):
-    """
-    TogetherAI APIを使用して画像を生成し、履歴リストの先頭に追加します。
-    """
     if not together_client:
         raise gr.Error("TogetherAIのAPIクライアントが設定されていません。環境変数 TOGETHER_API_KEY を確認してください。")
     if not prompt or not prompt.strip():
@@ -31,7 +27,6 @@ def generate_image_from_prompt(prompt: str, history: list):
 
     print(f"プロンプトを元に画像を生成中: '{prompt}'")
     try:
-        # TogetherAI APIを呼び出し
         response = together_client.images.generate(
             prompt=prompt,
             model=IMAGE_MODEL,
@@ -40,43 +35,50 @@ def generate_image_from_prompt(prompt: str, history: list):
             height=1024,
             width=1024
         )
-
-         # ▼▼▼ デバッグのためにこの行を追加 ▼▼▼
-        print("---------- API Response ----------")
-        print(response)
-        print("--------------------------------")
         
         generated_images = []
-        # response.data にはリクエストした全ての画像データが含まれる
+        
+        # ▼▼▼ 修正箇所: URLから画像をダウンロードするロジックに変更 ▼▼▼
         for image_data_item in response.data:
-            image_b64 = image_data_item.b64_json
+            # APIの応答に 'url' が含まれているかチェック
+            if hasattr(image_data_item, 'url') and image_data_item.url:
+                try:
+                    image_url = image_data_item.url
+                    print(f"URLから画像をダウンロード中: {image_url}")
+                    
+                    # URLにアクセスして画像データを取得
+                    image_response = requests.get(image_url)
+                    image_response.raise_for_status()  # エラーがあれば例外を発生させる
+                    
+                    # ダウンロードしたバイトデータからPILイメージを作成
+                    image = Image.open(BytesIO(image_response.content))
+                    generated_images.append((image, prompt))
+                    
+                except requests.exceptions.RequestException as e:
+                    print(f"画像のダウンロードに失敗しました: {e}")
+                    continue # この画像の処理をスキップして次に進む
             
-            # ▼▼▼ 修正箇所: 安全チェックを追加 ▼▼▼
-            # image_b64がNoneでない（正常にデータが返された）場合のみ処理する
-            if image_b64:
+            # (参考) b64_jsonが返ってくるモデルのためのフォールバック処理
+            elif image_data_item.b64_json:
+                image_b64 = image_data_item.b64_json
                 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("画像の生成に失敗しました。プロンプトがコンテンツフィルターに抵触した可能性があります。別のプロンプトで試してください。")
+            raise gr.Error("画像の生成に失敗しました。APIから有効な画像データやURLが返されませんでした。")
 
-        # 新しい画像のリストを、既存の履歴の先頭に追加
         new_history = generated_images + history
-        
         print(f"{len(generated_images)}枚の画像の生成が完了しました。")
         return new_history, new_history
 
     except Exception as e:
-        # 既にgr.Errorの場合はそのまま表示
         if isinstance(e, gr.Error):
             raise e
         print(f"予期せぬエラーが発生しました: {e}")
         raise gr.Error(f"画像の生成に失敗しました。エラー詳細: {e}")
 
-# --- 3. Gradio UIの構築 ---
+# --- 3. Gradio UIの構築 (変更なし) ---
 with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px; margin: auto;}") as app:
     gr.Markdown(
         """
@@ -86,9 +88,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px;
         """
     )
     gr.Markdown("---")
-    
     history_state = gr.State([])
-
     with gr.Column():
         prompt_input = gr.Textbox(
             label="画像生成プロンプト",
@@ -98,32 +98,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=".gradio-container {max-width: 800px;
         )
         if not together_client:
              gr.Warning("環境変数 `TOGETHER_API_KEY` が見つかりません。画像生成機能を利用するにはAPIキーを設定してください。")
-        generate_btn = gr.Button(
-            "画像生成 ✨",
-            variant="primary",
-            interactive=(together_client is not None)
-        )
-
+        generate_btn = gr.Button("画像生成 ✨ (4枚)", variant="primary", interactive=(together_client is not None))
     gr.Markdown("---")
     gr.Markdown("## 生成された画像")
-    
-    image_gallery = gr.Gallery(
-        label="生成結果",
-        show_label=False,
-        columns=2,
-        height="auto",
-        object_fit="contain",
-        value=None,
-        preview=True
-    )
-    
-    generate_btn.click(
-        fn=generate_image_from_prompt,
-        inputs=[prompt_input, history_state],
-        outputs=[image_gallery, history_state],
-        show_progress="full"
-    )
-    
+    image_gallery = gr.Gallery(label="生成結果", show_label=False, columns=2, height="auto", object_fit="contain", value=None, preview=True)
+    generate_btn.click(fn=generate_image_from_prompt, inputs=[prompt_input, history_state], outputs=[image_gallery, history_state], show_progress="full")
     gr.Examples(
         examples=[
             "A high-resolution photo of a futuristic city with flying cars",