Update app.py

app.py

@@ -193,14 +193,21 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
                </a>""")
     
     with gr.Column(scale=2):
+        # 모델 선택 부분 추가
+        with gr.Accordion("Model Selection", open=True):
+            model_choice = gr.CheckboxGroup(
+                models,
+                label=f'Choose up to {int(num_models)} models',
+                value=default_models,
+                interactive=True
+            )
+
         with gr.Group():
             txt_input = gr.Textbox(
                 label='Your prompt:', 
                 value=preSetPrompt, 
                 lines=3, 
-                autofocus=1,
-                container=False,
-                elem_classes="modern-input"
+                autofocus=1
             )
             neg_input = gr.Textbox(
                 label='Negative prompt:', 
@@ -257,4 +264,20 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', fill_width=True, css=CSS) as demo:
             rows=2
         ) 
 
+    # 이벤트 핸들러 추가
+    model_choice.change(update_imgbox, model_choice, output)
+    model_choice.change(extend_choices, model_choice, current_models)
+    random_button.click(random_choices, None, model_choice)
+
+    for m, o in zip(current_models, output):
+        gen_event = gr.on(
+            triggers=[gen_button.click, txt_input.submit],
+            fn=gen_fn,
+            inputs=[m, txt_input, neg_input, height, width, steps, cfg, seed],
+            outputs=[o],
+            concurrency_limit=None,
+            queue=False
+        )
+        o.change(add_gallery, [o, m, gallery], [gallery])
+
 demo.launch(show_api=False, max_threads=400)