Refactor Gradio UI in QIE_prompt_generator.py to encapsulate the layout within a build_ui function, enhancing modularity and readability. The update includes improved organization of input fields and maintains the existing functionality for prompt generation.

QIE_prompt_generator.py

@@ -105,10 +105,11 @@ def call_openai_chat(api_key: str, a_data_url: Optional[str], b_data_url: Option
     return english, names, japanese
 
 
-# ===== Gradio UI =====
+# ===== Gradio UI (standalone) =====
 
-with gr.Blocks(title="A→B 変換プロンプト自動生成（GPT-5固定）") as demo:
-    gr.Markdown("""
+def build_ui() -> gr.Blocks:
+    with gr.Blocks(title="A→B 変換プロンプト自動生成（GPT-5固定）") as demo:
+        gr.Markdown("""
 # 🎨 A→B 変換プロンプト自動生成
 画像A（入力）と画像B（出力）、補足説明を入力すると、  
 A→B の変換内容を**英語プロンプト**として自動生成し、  
@@ -116,29 +117,32 @@ A→B の変換内容を**英語プロンプト**として自動生成し、
 モデルは `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)
+        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")
+        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)
+        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)
+        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],
+        )
+
+    return demo
 
-    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()
+    build_ui().launch()