Update app.py

app.py

@@ -46,8 +46,14 @@ def gen_fn(model_str, prompt):
 def make_me():
     with gr.Row():
         with gr.Column(scale=1):
-            txt_input = gr.Textbox(label='Your prompt:', lines=3, container=False, elem_id="custom_textbox", placeholder="Prompt")
-            with gr.Row():
+            txt_input = gr.Textbox(
+                label='Your prompt:',
+                lines=3,
+                container=False,
+                elem_id="custom_textbox",
+                placeholder="Prompt"
+            )
+            with gr.Row(): 
                 gen_button = gr.Button('Generate images', elem_id="custom_gen_button")
                 stop_button = gr.Button('Stop', variant='secondary', interactive=False, elem_id="custom_stop_button")
                 
@@ -60,18 +66,53 @@ def make_me():
                 gen_button.click(on_generate_click, inputs=None, outputs=[gen_button, stop_button])
                 stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button])
     
+    # Делим вывод изображений на два ряда
     with gr.Row():
-        output = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models]
-        current_models = [gr.Textbox(m, visible=False) for m in default_models]
-        for m, o in zip(current_models, output):
+        half = len(default_models) // 2
+        output_row1 = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models[:half]]
+        output_row2 = [gr.Image(label=m, min_width=250, height=250, elem_id="custom_image") for m in default_models[half:]]
+        current_models_row1 = [gr.Textbox(m, visible=False) for m in default_models[:half]]
+        current_models_row2 = [gr.Textbox(m, visible=False) for m in default_models[half:]]
+        
+        # Первый ряд изображений
+        for m, o in zip(current_models_row1, output_row1):
+            gen_event = gen_button.click(gen_fn, [m, txt_input], o)
+            stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])
+        
+        # Второй ряд изображений
+        for m, o in zip(current_models_row2, output_row2):
             gen_event = gen_button.click(gen_fn, [m, txt_input], o)
             stop_button.click(on_stop_click, inputs=None, outputs=[gen_button, stop_button], cancels=[gen_event])
     
+    # Разделяем чекбоксы на два ряда
     with gr.Accordion('Model selection', elem_id="custom_accordion"):
-        model_choice = gr.CheckboxGroup(models, label=f'{num_models} different models selected', value=default_models, interactive=True, elem_id="custom_checkbox_group")
-        model_choice.change(update_imgbox, model_choice, output)
-        model_choice.change(extend_choices, model_choice, current_models)
-    
+        half = len(models) // 2
+        model_choice_row1 = models[:half]
+        model_choice_row2 = models[half:]
+        
+        model_choice1 = gr.CheckboxGroup(
+            model_choice_row1,
+            label=f'{len(model_choice_row1)} models selected',
+            value=default_models[:half],
+            interactive=True,
+            elem_id="custom_checkbox_group"
+        )
+        model_choice2 = gr.CheckboxGroup(
+            model_choice_row2,
+            label=f'{len(model_choice_row2)} models selected',
+            value=default_models[half:],
+            interactive=True,
+            elem_id="custom_checkbox_group"
+        )
+        
+        # Обновляем первый ряд
+        model_choice1.change(update_imgbox, model_choice1, output_row1)
+        model_choice1.change(extend_choices, model_choice1, current_models_row1)
+        
+        # Обновляем второй ряд
+        model_choice2.change(update_imgbox, model_choice2, output_row2)
+        model_choice2.change(extend_choices, model_choice2, current_models_row2)
+
     with gr.Row():
         gr.HTML("")