Refactor UI in app.py to implement a tabbed layout for the Trainer and Prompt Generator sections, enhancing organization and user experience. The update includes restructured input fields and improved accessibility for image uploads and prompt generation functionalities.

QIE_prompt_generator.py

@@ -0,0 +1,144 @@
+import base64
+import mimetypes
+from pathlib import Path
+import gradio as gr
+from typing import Optional, Tuple
+
+# ===== ユーティリティ =====
+
+def file_to_data_url(path: str) -> Optional[str]:
+    """画像ファイルを data URL に変換"""
+    if not path:
+        return None
+    p = Path(path)
+    if not p.exists():
+        return None
+    mime, _ = mimetypes.guess_type(str(p))
+    if not mime:
+        mime = "image/png"
+    with open(p, "rb") as f:
+        b64 = base64.b64encode(f.read()).decode("utf-8")
+    return f"data:{mime};base64,{b64}"
+
+
+# ===== Meta Prompt =====
+META_PROMPT = """You are an AI prompt generator for image-to-image transformation tasks in art and illustration pipelines.
+Given two images — A (input) and B (output) — and an optional description or notes, your goal is to write a precise, structured English prompt that fully explains how A transforms into B, including both conceptual and visual rules of the transformation.
+
+Instructions:
+1) Write 3–6 concise sentences describing the transformation.
+2) Begin with “The [type of A] transforms into …” or “Convert the [A] into …”.
+3) Clearly describe:
+   - What disappears or remains (e.g., hair, clothes, shadows, lines).
+   - What structural or stylistic simplifications occur (e.g., box, sphere, guide lines).
+   - How lines, colors, or lighting change (e.g., colored → lineart, flat → shaded).
+   - Any rules for anatomy, proportion, or rendering (e.g., keep pose, maintain base colors).
+   - Background or presentation constraints (e.g., white background, no shading).
+4) Integrate any user-provided notes naturally into the text.
+
+Output Format:
+- English Prompt: A full paragraph (3–6 sentences) describing the transformation from A to B.
+- Name Suggestions: Propose 3 short, descriptive task-name candidates (e.g., Image2Body, Body2Box, Sketch2Line, Flat2Shade, Guide2Color, Light2Render, etc.).
+- Optional Japanese Translation: Provide a brief Japanese version of the English prompt for understanding if requested.
+
+Make the wording professional, objective, and consistent with technical art pipeline prompts.
+"""
+
+
+# ===== OpenAI 呼び出し =====
+def call_openai_chat(api_key: str, a_data_url: Optional[str], b_data_url: Optional[str], notes: str, want_japanese: bool):
+    """OpenAI GPT-5 API 呼び出し"""
+    if not api_key:
+        return ("", "", "（エラー：API Key が未入力です）")
+
+    try:
+        from openai import OpenAI
+    except Exception:
+        return ("", "", "（エラー：openai パッケージが見つかりません。`pip install openai` を実行してください）")
+
+    client = OpenAI(api_key=api_key)
+    model = "gpt-5"
+
+    user_content = []
+    if a_data_url:
+        user_content.append({"type": "text", "text": "Image A (input):"})
+        user_content.append({"type": "image_url", "image_url": {"url": a_data_url}})
+    if b_data_url:
+        user_content.append({"type": "text", "text": "Image B (output):"})
+        user_content.append({"type": "image_url", "image_url": {"url": b_data_url}})
+    user_content.append({
+        "type": "text",
+        "text": f"Additional notes: {notes or '(none)'}\n{'Include Japanese translation.' if want_japanese else 'No Japanese translation.'}"
+    })
+
+    try:
+        resp = client.chat.completions.create(
+            model=model,
+            messages=[
+                {"role": "system", "content": META_PROMPT},
+                {"role": "user", "content": user_content},
+            ]
+        )
+        text = resp.choices[0].message.content.strip()
+    except Exception as e:
+        return ("", "", f"（APIエラー：{e}）")
+
+    # 簡易パース
+    english, names, japanese = "", "", ""
+    lower = text.lower()
+    if "name suggestions:" in lower:
+        split = text.split("English Prompt:")[1].split("Name Suggestions:")
+        english = split[0].strip()
+        rest = split[1]
+        if "Japanese Translation:" in rest:
+            name_part, jp_part = rest.split("Japanese Translation:", 1)
+            names = name_part.strip()
+            japanese = jp_part.strip()
+        else:
+            names = rest.strip()
+    else:
+        english = text
+
+    if not want_japanese:
+        japanese = ""
+
+    return english, names, japanese
+
+
+# ===== Gradio UI =====
+
+with gr.Blocks(title="A→B 変換プロンプト自動生成（GPT-5固定）") as demo:
+    gr.Markdown("""
+# 🎨 A→B 変換プロンプト自動生成
+画像A（入力）と画像B（出力）、補足説明を入力すると、  
+A→B の変換内容を**英語プロンプト**として自動生成し、  
+さらに**タスク名候補（3件）**を提案します。  
+モデルは `gpt-5` を使用します。
+""")
+
+    api_key = gr.Textbox(label="OpenAI API Key", type="password", placeholder="sk-...")
+    with gr.Row():
+        img_a = gr.Image(type="filepath", label="Image A (Input)", height=300)
+        img_b = gr.Image(type="filepath", label="Image B (Output)", height=300)
+
+    notes = gr.Textbox(label="補足説明（日本語可）", lines=4, placeholder="例）髪・服・背景は消す。目は四角。鎖骨はピンク線。など", value="この画像は例であって、汎用的なプロンプトにする")
+    want_japanese = gr.Checkbox(label="日本語訳を含める", value=True)
+    run_btn = gr.Button("生成する", variant="primary")
+
+    english_out = gr.Textbox(label="English Prompt", lines=8)
+    names_out = gr.Textbox(label="Name Suggestions", lines=4)
+    japanese_out = gr.Textbox(label="日本語訳（任意）", lines=8)
+
+    def on_click(api_key_in, a_path, b_path, notes_in, ja_flag):
+        a_url = file_to_data_url(a_path) if a_path else None
+        b_url = file_to_data_url(b_path) if b_path else None
+        return call_openai_chat(api_key_in, a_url, b_url, notes_in, ja_flag)
+
+    run_btn.click(
+        fn=on_click,
+        inputs=[api_key, img_a, img_b, notes, want_japanese],
+        outputs=[english_out, names_out, japanese_out],
+    )
+
+if __name__ == "__main__":
+    demo.launch()

app.py

@@ -11,6 +11,7 @@ import time
 import json
 
 import gradio as gr
+import importlib
 import spaces
 
 # Local modules
@@ -706,154 +707,190 @@ def build_ui() -> gr.Blocks:
     }
     """
     with gr.Blocks(title="Qwen-Image-Edit: Trainer", css=css) as demo:
-        gr.Markdown("""
-        # Qwen-Image-Edit Trainer
-        学習に使う画像をアップロードし、必要ならファイル名の前後にある共通の文字（prefix/suffix）を指定して、
-        自動でデータセットを作成し学習を開始します。難しい操作は不要です。
-        """)
-
-        with gr.Accordion("Settings", elem_classes=["pad-section"]):
-            with gr.Group():
-                with gr.Row():
-                    output_name = gr.Textbox(label="OUTPUT NAME", placeholder="my_lora_output", lines=1)
-                    caption = gr.Textbox(label="CAPTION", placeholder="A photo of ...", lines=2)
+        with gr.Tabs() as tabs:
+            with gr.TabItem("Training"):
+                gr.Markdown("""
+                # Qwen-Image-Edit Trainer
+                学習に使う画像をアップロードし、必要ならファイル名の前後にある共通の文字（prefix/suffix）を指定して、
+                自動でデータセットを作成し学習を開始します。難しい操作は不要です。
+                """)
+
+                with gr.Accordion("Settings", elem_classes=["pad-section"]):
+                    with gr.Group():
+                        with gr.Row():
+                            output_name = gr.Textbox(label="OUTPUT NAME", placeholder="my_lora_output", lines=1)
+                            caption = gr.Textbox(label="CAPTION", placeholder="A photo of ...", lines=2)
 
-                with gr.Row():
-                    lr_input = gr.Textbox(label="Learning rate", value="1e-3")
-                    dim_input = gr.Number(label="Network dim", value=4, precision=0)
-                    seed_input = gr.Number(label="Seed", value=42, precision=0)
-                    max_epochs = gr.Number(label="Max epochs", value=100, precision=0)
-                    save_every = gr.Number(label="Save every N epochs", value=10, precision=0)
-
-        with gr.Accordion("Target Image", elem_classes=["pad-section_0"]):
-            with gr.Group():
-                with gr.Row():
-                    images_input = gr.File(label="Upload target images", file_count="multiple", type="filepath", height=220, scale=3)
-                    main_gallery = gr.Gallery(label="Target preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
                         with gr.Row():
-                            main_prefix = gr.Textbox(label="Target prefix", placeholder="e.g., IMG_")
-                            main_suffix = gr.Textbox(label="Target suffix", placeholder="e.g., _v2")
-                        with gr.Accordion("prefix/sufixについて", open=False):
-                            gr.Markdown("""
-                            ファイルの同名判定のため、画像のファイル名から共通の先頭/末尾文字を取り除く指定（例: IMG_ や _v2）
-                            - まずターゲット画像のファイル名（拡張子なし）から、指定した Target prefix/suffix を取り除いたものを key とします。
-                            - 各コントロールは「付加」規則で、期待名 = control_prefix_i + key + control_suffix_i + ".png" を探して対応付けます。
-                            - アップロード時に画像は自動で .png に変換して保存します（元のファイル名のベースは維持）。
-                            - Control 0 は必須、Control 1〜7 は任意。コントロール画像が1枚だけのときは、すべてのターゲット画像に適用します。
-                            """)
-
-            # control_0 is required and shown outside the accordion
-        with gr.Accordion("Control 0", elem_classes=["pad-section_1"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl0_files = gr.File(label="Upload control_0 images (required)", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl0_gallery = gr.Gallery(label="control_0 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            lr_input = gr.Textbox(label="Learning rate", value="1e-3")
+                            dim_input = gr.Number(label="Network dim", value=4, precision=0)
+                            seed_input = gr.Number(label="Seed", value=42, precision=0)
+                            max_epochs = gr.Number(label="Max epochs", value=100, precision=0)
+                            save_every = gr.Number(label="Save every N epochs", value=10, precision=0)
+
+                with gr.Accordion("Target Image", elem_classes=["pad-section_0"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl0_prefix = gr.Textbox(label="control_0 prefix", placeholder="e.g., C0_")
-                            ctrl0_suffix = gr.Textbox(label="control_0 suffix", placeholder="e.g., _mask")
-
-            # Optional controls start from 1, accordion closed by default
-        with gr.Accordion("Control 1", open=False, elem_classes=["pad-section_0"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl1_files = gr.File(label="Upload control_1 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl1_gallery = gr.Gallery(label="control_1 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            images_input = gr.File(label="Upload target images", file_count="multiple", type="filepath", height=220, scale=3)
+                            main_gallery = gr.Gallery(label="Target preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    main_prefix = gr.Textbox(label="Target prefix", placeholder="e.g., IMG_")
+                                    main_suffix = gr.Textbox(label="Target suffix", placeholder="e.g., _v2")
+                                with gr.Accordion("prefix/sufixについて", open=False):
+                                    gr.Markdown("""
+                                    ファイルの同名判定のため、画像のファイル名から共通の先頭/末尾文字を取り除く指定（例: IMG_ や _v2）
+                                    - まずターゲット画像のファイル名（拡張子なし）から、指定した Target prefix/suffix を取り除いたものを key とします。
+                                    - 各コントロールは「付加」規則で、期待名 = control_prefix_i + key + control_suffix_i + ".png" を探して対応付けます。
+                                    - アップロード時に画像は自動で .png に変換して保存します（元のファイル名のベースは維持）。
+                                    - Control 0 は必須、Control 1〜7 は任意。コントロール画像が1枚だけのときは、すべてのターゲット画像に適用します。
+                                    """)
+
+                # control_0 is required and shown outside the accordion
+                with gr.Accordion("Control 0", elem_classes=["pad-section_1"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl1_prefix = gr.Textbox(label="control_1 prefix", placeholder="")
-                            ctrl1_suffix = gr.Textbox(label="control_1 suffix", placeholder="")
-        with gr.Accordion("Control 2", open=False, elem_classes=["pad-section_1"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl2_files = gr.File(label="Upload control_2 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl2_gallery = gr.Gallery(label="control_2 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            ctrl0_files = gr.File(label="Upload control_0 images (required)", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl0_gallery = gr.Gallery(label="control_0 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl0_prefix = gr.Textbox(label="control_0 prefix", placeholder="e.g., C0_")
+                                    ctrl0_suffix = gr.Textbox(label="control_0 suffix", placeholder="e.g., _mask")
+
+                # Optional controls start from 1, accordion closed by default
+                with gr.Accordion("Control 1", open=False, elem_classes=["pad-section_0"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl2_prefix = gr.Textbox(label="control_2 prefix", placeholder="")
-                            ctrl2_suffix = gr.Textbox(label="control_2 suffix", placeholder="")
-        with gr.Accordion("Control 3", open=False, elem_classes=["pad-section_0"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl3_files = gr.File(label="Upload control_3 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl3_gallery = gr.Gallery(label="control_3 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            ctrl1_files = gr.File(label="Upload control_1 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl1_gallery = gr.Gallery(label="control_1 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl1_prefix = gr.Textbox(label="control_1 prefix", placeholder="")
+                                    ctrl1_suffix = gr.Textbox(label="control_1 suffix", placeholder="")
+                with gr.Accordion("Control 2", open=False, elem_classes=["pad-section_1"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl3_prefix = gr.Textbox(label="control_3 prefix", placeholder="")
-                            ctrl3_suffix = gr.Textbox(label="control_3 suffix", placeholder="")
-        with gr.Accordion("Control 4", open=False, elem_classes=["pad-section_1"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl4_files = gr.File(label="Upload control_4 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl4_gallery = gr.Gallery(label="control_4 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            ctrl2_files = gr.File(label="Upload control_2 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl2_gallery = gr.Gallery(label="control_2 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl2_prefix = gr.Textbox(label="control_2 prefix", placeholder="")
+                                    ctrl2_suffix = gr.Textbox(label="control_2 suffix", placeholder="")
+                with gr.Accordion("Control 3", open=False, elem_classes=["pad-section_0"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl4_prefix = gr.Textbox(label="control_4 prefix", placeholder="")
-                            ctrl4_suffix = gr.Textbox(label="control_4 suffix", placeholder="")
-        with gr.Accordion("Control 5", open=False, elem_classes=["pad-section_0"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl5_files = gr.File(label="Upload control_5 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl5_gallery = gr.Gallery(label="control_5 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            ctrl3_files = gr.File(label="Upload control_3 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl3_gallery = gr.Gallery(label="control_3 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl3_prefix = gr.Textbox(label="control_3 prefix", placeholder="")
+                                    ctrl3_suffix = gr.Textbox(label="control_3 suffix", placeholder="")
+                with gr.Accordion("Control 4", open=False, elem_classes=["pad-section_1"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl5_prefix = gr.Textbox(label="control_5 prefix", placeholder="")
-                            ctrl5_suffix = gr.Textbox(label="control_5 suffix", placeholder="")
-        with gr.Accordion("Control 6", open=False, elem_classes=["pad-section_1"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl6_files = gr.File(label="Upload control_6 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl6_gallery = gr.Gallery(label="control_6 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            ctrl4_files = gr.File(label="Upload control_4 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl4_gallery = gr.Gallery(label="control_4 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl4_prefix = gr.Textbox(label="control_4 prefix", placeholder="")
+                                    ctrl4_suffix = gr.Textbox(label="control_4 suffix", placeholder="")
+                with gr.Accordion("Control 5", open=False, elem_classes=["pad-section_0"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl6_prefix = gr.Textbox(label="control_6 prefix", placeholder="")
-                            ctrl6_suffix = gr.Textbox(label="control_6 suffix", placeholder="")
-        with gr.Accordion("Control 7", open=False, elem_classes=["pad-section_0"]):
-            with gr.Group():
-                with gr.Row():
-                    ctrl7_files = gr.File(label="Upload control_7 images", file_count="multiple", type="filepath", height=220, scale=3)
-                    ctrl7_gallery = gr.Gallery(label="control_7 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
-                    with gr.Column(scale=1):
+                            ctrl5_files = gr.File(label="Upload control_5 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl5_gallery = gr.Gallery(label="control_5 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl5_prefix = gr.Textbox(label="control_5 prefix", placeholder="")
+                                    ctrl5_suffix = gr.Textbox(label="control_5 suffix", placeholder="")
+                with gr.Accordion("Control 6", open=False, elem_classes=["pad-section_1"]):
+                    with gr.Group():
                         with gr.Row():
-                            ctrl7_prefix = gr.Textbox(label="control_7 prefix", placeholder="")
-                            ctrl7_suffix = gr.Textbox(label="control_7 suffix", placeholder="")
-
-        # Models root / OUTPUT_DIR_BASE / DATASET_CONFIG are auto-resolved at runtime; no user input needed.
-
-        run_btn = gr.Button("Start Training", variant="primary")
-        logs = gr.Textbox(label="Logs", lines=20)
-        ckpt_files = gr.Files(label="Checkpoints (live)", interactive=False)
-
-        # moved max_epochs/save_every above next to OUTPUT NAME
-
-        # Wire previews
-        images_input.change(fn=_files_to_gallery, inputs=images_input, outputs=main_gallery)
-        ctrl0_files.change(fn=_files_to_gallery, inputs=ctrl0_files, outputs=ctrl0_gallery)
-        ctrl1_files.change(fn=_files_to_gallery, inputs=ctrl1_files, outputs=ctrl1_gallery)
-        ctrl2_files.change(fn=_files_to_gallery, inputs=ctrl2_files, outputs=ctrl2_gallery)
-        ctrl3_files.change(fn=_files_to_gallery, inputs=ctrl3_files, outputs=ctrl3_gallery)
-        ctrl4_files.change(fn=_files_to_gallery, inputs=ctrl4_files, outputs=ctrl4_gallery)
-        ctrl5_files.change(fn=_files_to_gallery, inputs=ctrl5_files, outputs=ctrl5_gallery)
-        ctrl6_files.change(fn=_files_to_gallery, inputs=ctrl6_files, outputs=ctrl6_gallery)
-        ctrl7_files.change(fn=_files_to_gallery, inputs=ctrl7_files, outputs=ctrl7_gallery)
-
-        run_btn.click(
-            fn=run_training,
-            inputs=[
-                output_name, caption, images_input, main_prefix, main_suffix,
-                ctrl0_files, ctrl0_prefix, ctrl0_suffix,
-                ctrl1_files, ctrl1_prefix, ctrl1_suffix,
-                ctrl2_files, ctrl2_prefix, ctrl2_suffix,
-                ctrl3_files, ctrl3_prefix, ctrl3_suffix,
-                ctrl4_files, ctrl4_prefix, ctrl4_suffix,
-                ctrl5_files, ctrl5_prefix, ctrl5_suffix,
-                ctrl6_files, ctrl6_prefix, ctrl6_suffix,
-                ctrl7_files, ctrl7_prefix, ctrl7_suffix,
-                lr_input, dim_input, seed_input, max_epochs, save_every,
-            ],
-            outputs=[logs, ckpt_files],
-        )
+                            ctrl6_files = gr.File(label="Upload control_6 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl6_gallery = gr.Gallery(label="control_6 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl6_prefix = gr.Textbox(label="control_6 prefix", placeholder="")
+                                    ctrl6_suffix = gr.Textbox(label="control_6 suffix", placeholder="")
+                with gr.Accordion("Control 7", open=False, elem_classes=["pad-section_0"]):
+                    with gr.Group():
+                        with gr.Row():
+                            ctrl7_files = gr.File(label="Upload control_7 images", file_count="multiple", type="filepath", height=220, scale=3)
+                            ctrl7_gallery = gr.Gallery(label="control_7 preview", columns=4, height=220, object_fit='contain', preview=True, scale=3)
+                            with gr.Column(scale=1):
+                                with gr.Row():
+                                    ctrl7_prefix = gr.Textbox(label="control_7 prefix", placeholder="")
+                                    ctrl7_suffix = gr.Textbox(label="control_7 suffix", placeholder="")
+
+                # Models root / OUTPUT_DIR_BASE / DATASET_CONFIG are auto-resolved at runtime; no user input needed.
+
+                run_btn = gr.Button("Start Training", variant="primary")
+                logs = gr.Textbox(label="Logs", lines=20)
+                ckpt_files = gr.Files(label="Checkpoints (live)", interactive=False)
+
+                # moved max_epochs/save_every above next to OUTPUT NAME
+
+                # Wire previews
+                images_input.change(fn=_files_to_gallery, inputs=images_input, outputs=main_gallery)
+                ctrl0_files.change(fn=_files_to_gallery, inputs=ctrl0_files, outputs=ctrl0_gallery)
+                ctrl1_files.change(fn=_files_to_gallery, inputs=ctrl1_files, outputs=ctrl1_gallery)
+                ctrl2_files.change(fn=_files_to_gallery, inputs=ctrl2_files, outputs=ctrl2_gallery)
+                ctrl3_files.change(fn=_files_to_gallery, inputs=ctrl3_files, outputs=ctrl3_gallery)
+                ctrl4_files.change(fn=_files_to_gallery, inputs=ctrl4_files, outputs=ctrl4_gallery)
+                ctrl5_files.change(fn=_files_to_gallery, inputs=ctrl5_files, outputs=ctrl5_gallery)
+                ctrl6_files.change(fn=_files_to_gallery, inputs=ctrl6_files, outputs=ctrl6_gallery)
+                ctrl7_files.change(fn=_files_to_gallery, inputs=ctrl7_files, outputs=ctrl7_gallery)
+
+                run_btn.click(
+                    fn=run_training,
+                    inputs=[
+                        output_name, caption, images_input, main_prefix, main_suffix,
+                        ctrl0_files, ctrl0_prefix, ctrl0_suffix,
+                        ctrl1_files, ctrl1_prefix, ctrl1_suffix,
+                        ctrl2_files, ctrl2_prefix, ctrl2_suffix,
+                        ctrl3_files, ctrl3_prefix, ctrl3_suffix,
+                        ctrl4_files, ctrl4_prefix, ctrl4_suffix,
+                        ctrl5_files, ctrl5_prefix, ctrl5_suffix,
+                        ctrl6_files, ctrl6_prefix, ctrl6_suffix,
+                        ctrl7_files, ctrl7_prefix, ctrl7_suffix,
+                        lr_input, dim_input, seed_input, max_epochs, save_every,
+                    ],
+                    outputs=[logs, ckpt_files],
+                )
+
+            with gr.TabItem("Prompt Generator"):
+                gr.Markdown("""
+                # 🎨 A→B 変換プロンプト自動生成
+                画像A（入力）と画像B（出力）、補足説明を入力すると、  
+                A→B の変換内容を英語プロンプトとして自動生成し、タスク名候補（3件）も提案します。  
+                モデルは `gpt-5` を使用します。
+                """)
+
+                api_key_pg = gr.Textbox(label="OpenAI API Key", type="password", placeholder="sk-...")
+                with gr.Row():
+                    img_a_pg = gr.Image(type="filepath", label="Image A (Input)", height=300)
+                    img_b_pg = gr.Image(type="filepath", label="Image B (Output)", height=300)
+
+                notes_pg = gr.Textbox(label="補足説明（日本語可）", lines=4, value="この画像は例であって、汎用的なプロンプトにする")
+                want_japanese_pg = gr.Checkbox(label="日本語訳を含める", value=True)
+                run_btn_pg = gr.Button("生成する", variant="primary")
+
+                english_out_pg = gr.Textbox(label="English Prompt", lines=8)
+                names_out_pg = gr.Textbox(label="Name Suggestions", lines=4)
+                japanese_out_pg = gr.Textbox(label="日本語訳（任意）", lines=8)
+
+                def _on_click_prompt(api_key_in, a_path, b_path, notes_in, ja_flag):
+                    # Lazy import to avoid constructing extra Blocks at startup
+                    qpg = importlib.import_module("QIE_prompt_generator")
+                    a_url = qpg.file_to_data_url(a_path) if a_path else None
+                    b_url = qpg.file_to_data_url(b_path) if b_path else None
+                    return qpg.call_openai_chat(api_key_in, a_url, b_url, notes_in, ja_flag)
+
+                run_btn_pg.click(
+                    fn=_on_click_prompt,
+                    inputs=[api_key_pg, img_a_pg, img_b_pg, notes_pg, want_japanese_pg],
+                    outputs=[english_out_pg, names_out_pg, japanese_out_pg],
+                )
 
     return demo