大幅改善。

app.py

@@ -110,6 +110,13 @@ def make_html_table(caption):
 def create_dummy_image(width=512, height=512, alpha=0):
     return Image.new("RGBA", (width, height), (0, 0, 0, alpha))
 
+def update_history_tables_on_select(evt: gr.SelectData):
+    if evt.index is not None and 0 <= evt.index < len(history):
+        # 選択された画像のキャプションを取得
+        selected_caption = history[evt.index][1]
+        return make_html_table(selected_caption)
+    return "" # 何も選択されていない場合は空文字列を返す
+
 def update_history():
     tables_html = "".join(
         f'<div style="margin-bottom:12px">{make_html_table(item[1])}</div>'
@@ -159,7 +166,11 @@ def infer(prompt, neg, seed, rand, w, h, cfg, steps, scheduler_type,
     progress(1.0, desc="Done!")
 
     gallery_items = [(item[0], item[1]) for item in history]
-#    tables_html = "".join(f'<div style="margin-bottom:12px">{make_html_table(item[1])}</div>' for item in history)
+
+    processed_img, processed_gallery_items = process_image(img, gallery_items)
+    
+    latest_caption_table = make_html_table(caption_text)
+    return processed_img, processed_gallery_items, latest_caption_table
 
 import gc
 import torch
@@ -307,17 +318,17 @@ with gr.Blocks(css=css, theme=gr.themes.Default(font=[gr.themes.GoogleFont("Play
         gr.HTML('<section class="nobackground"><a href="https://civitai.com/models/1553716?modelVersionId=1855218" target="_blank">Link: Civitai</a></section>')
 
         with gr.Row():
-            prompt = gr.Textbox(lines=1, placeholder="Prompt…", value="1girl, cocoart, masterpiece, anime,", label="Prompt")
-            neg    = gr.Textbox(lines=1, placeholder="Negative prompt", value="low quality, worst quality, bad shadow, lowres, error, miss stroke, sketch art, smoke, ugly, extra digits, creepy, imprecise, glowing blur,", label="Negative prompt")
+            prompt = gr.Textbox(lines=1, placeholder="Prompt…", value="1girl, cocoart, masterpiece, anime, high quality lighting,", label="Prompt")
+            neg    = gr.Textbox(lines=1, placeholder="Negative prompt", value="low quality, worst quality, bad, lowres, error, miss stroke, smoke, ugly, extra digits, creepy, imprecise, blurry,", label="Negative prompt")
         with gr.Row():
             seed_sl = gr.Slider(0, MAX_SEED, step=1, value=0, label="Seed")
             rand    = gr.Checkbox(True, label="Randomize seed")
         with gr.Row():
-            width  = gr.Slider(256, 512, step=32, value=512, label="Width")
-            height = gr.Slider(256, 768, step=32, value=512, label="Height")
+            width  = gr.Slider(256, 768, step=32, value=512, label="Width")
+            height = gr.Slider(256, 1024, step=32, value=512, label="Height")
         with gr.Row():
-            cfg    = gr.Slider(1.0, 30.0, step=0.5, value=7.5, label="CFG Scale")
-            steps  = gr.Slider(1, 20, step=1, value=12,   label="Steps")
+            cfg    = gr.Slider(1.0, 30.0, step=0.5, value=6, label="CFG Scale")
+            steps  = gr.Slider(1, 12, step=1, value=12,   label="Steps")
         with gr.Row():
             scheduler_type = gr.Radio(choices=["Euler Ancestral", "DPM++ 2M SDE"], value="Euler Ancestral", label="Scheduler")
             run = gr.Button("Generate")
@@ -390,22 +401,31 @@ window.addEventListener('load', () => {
     run.click(
         fn=infer,
         inputs=[prompt, neg, seed_sl, rand, width, height, cfg, steps, scheduler_type],
-        outputs=[img_out, history_gallery]
-    ).then(
-        fn=update_history,
-        inputs=[],
+        outputs=[img_out, history_gallery, history_tables]
+    )
+    history_gallery.select(
+        fn=update_history_tables_on_select,
+        inputs=None, # select イベントは自動的にイベントデータ (gr.SelectData) を渡す
         outputs=[history_tables]
     )
 
 # ページロード時に history から初期表示
+#    demo.load(
+#        fn=lambda: (
+#        [(img_or_url, caption) for img_or_url, caption in history],
+#        "".join(f'<div style="margin-bottom:12px">{make_html_table(caption)}</div>' for _, caption in history)
+#    ),
+#        inputs=[],
+#        outputs=[history_gallery, history_tables]
+#    ) 
     demo.load(
         fn=lambda: (
-        [(img_or_url, caption) for img_or_url, caption in history],
-        "".join(f'<div style="margin-bottom:12px">{make_html_table(caption)}</div>' for _, caption in history)
-    ),
+            [(img_or_url, caption) for img_or_url, caption in history],
+            make_html_table(history[0][1]) if history else ""
+        ),
         inputs=[],
         outputs=[history_gallery, history_tables]
-    ) 
+    )
     
 demo.queue()
 demo.launch()