Update app.py

app.py

@@ -101,14 +101,13 @@ def infer(
     history.insert(0, (img, caption))
     progress(1.0, desc="Done!")
 
-    # Gallery用の画像リスト、State用のキャプションリスト
+    # Gallery用とState用リスト
     images = [item[0] for item in history]
     captions = [item[1] for item in history]
     return img, images, captions
 
 # 選択コールバック
 def on_select(evt: gr.SelectData, captions):
-    # evt.value は PIL.Image, evt.index は選択位置
     return evt.value, make_html_table(captions[evt.index])
 
 # CSS設定
@@ -132,22 +131,36 @@ with gr.Blocks(css=css) as demo:
         height = gr.Slider(256, MAX_SIZE, step=32, value=512, label="Height")
         cfg = gr.Slider(1.0, 30.0, step=0.1, value=7.5, label="CFG Scale")
         steps = gr.Slider(1, 50, step=1, value=12, label="Steps")
-        scheduler_type = gr.Radio(choices=["Euler Ancestral", "DPM++ 2M SDE"],
-                                  value="Euler Ancestral", label="Scheduler")
+        scheduler_type = gr.Radio(
+            choices=["Euler Ancestral", "DPM++ 2M SDE"],
+            value="Euler Ancestral",
+            label="Scheduler"
+        )
         run = gr.Button("Generate")
         img_out = gr.Image()
         table_html = gr.HTML()
         state = gr.State([])
-        history_gallery = gr.Gallery(label="生成履歴", columns=4, height=300,
-                                     show_label=True, interactive=True, type="pil", show_upload_button=False)
+        history_gallery = gr.Gallery(
+            label="生成履歴",
+            columns=4,
+            height=300,
+            show_label=True,
+            interactive=True,
+            type="pil",
+            show_upload_button=False
+        )
 
     # イベント設定
-    run.click(fn=infer,
-              inputs=[prompt, neg, seed_sl, rand, width, height, cfg, steps, scheduler_type],
-              outputs=[img_out, history_gallery, state])
-    history_gallery.select(fn=on_select,
-                           inputs=[state],
-                           outputs=[img_out, table_html])
+    run.click(
+        fn=infer,
+        inputs=[prompt, neg, seed_sl, rand, width, height, cfg, steps, scheduler_type],
+        outputs=[img_out, history_gallery, state]
+    )
+    history_gallery.select(
+        fn=on_select,
+        inputs=[state],
+        outputs=[img_out, table_html]
+    )
 
     demo.queue()
     demo.launch()