update UI

app.py

@@ -1105,9 +1105,9 @@ def run_fn(
 
 
 
-def make_input_images_section(rows=1, height="auto"):
+def make_input_images_section(rows=1, cols=3, height="auto"):
     gr.Markdown('### Input Images')
-    input_gallery = gr.Gallery(value=None, label="Select images", show_label=True, elem_id="images", columns=[3], rows=[rows], object_fit="contain", height=height, type="pil", show_share_button=False)
+    input_gallery = gr.Gallery(value=None, label="Select images", show_label=True, elem_id="images", columns=[cols], rows=[rows], object_fit="contain", height=height, type="pil", show_share_button=False)
     submit_button = gr.Button("🔴 RUN", elem_id="submit_button", variant='primary')
     clear_images_button = gr.Button("🗑️Clear", elem_id='clear_button', variant='stop')
     return input_gallery, submit_button, clear_images_button
@@ -1570,6 +1570,12 @@ with demo:
         gr.Markdown('---')
 
         with gr.Row():
+            with gr.Column(scale=5, min_width=200):
+                input_gallery, submit_button, clear_images_button = make_input_images_section()
+                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
+                num_images_slider.value = 100
+                clear_images_button.visible = False
+                logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
             with gr.Column(scale=5, min_width=200):
                 gr.Markdown('### Output (Recursion #1)')
                 l1_gallery = gr.Gallery(format='png', value=[], label="Recursion #1", show_label=True, elem_id="ncut_l1", columns=[3], rows=[5], object_fit="contain", height="auto", show_fullscreen_button=True, interactive=False)
@@ -1583,12 +1589,6 @@ with demo:
                 l3_gallery = gr.Gallery(format='png', value=[], label="Recursion #3", show_label=True, elem_id="ncut_l3", columns=[3], rows=[5], object_fit="contain", height="auto", show_fullscreen_button=True, interactive=False)
                 add_output_images_buttons(l3_gallery)
         with gr.Row():
-            with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button = make_input_images_section()
-                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
-                num_images_slider.value = 100
-                clear_images_button.visible = False
-                logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
                 
             with gr.Column(scale=5, min_width=200):
                 with gr.Accordion("➡️ Recursion config", open=True):
@@ -1599,6 +1599,7 @@ with demo:
                     l1_affinity_focal_gamma_slider = gr.Slider(0.01, 1, step=0.01, label="Recursion #1: Affinity focal gamma", value=0.5, elem_id="recursion_l1_gamma")
                     l2_affinity_focal_gamma_slider = gr.Slider(0.01, 1, step=0.01, label="Recursion #2: Affinity focal gamma", value=0.5, elem_id="recursion_l2_gamma")
                     l3_affinity_focal_gamma_slider = gr.Slider(0.01, 1, step=0.01, label="Recursion #3: Affinity focal gamma", value=0.5, elem_id="recursion_l3_gamma")
+            with gr.Column(scale=5, min_width=200):
                 [
                     model_dropdown, layer_slider, node_type_dropdown, num_eig_slider, 
                     affinity_focal_gamma_slider, num_sample_ncut_slider, ncut_knn_slider, ncut_indirect_connection, ncut_make_orthogonal, 
@@ -1637,6 +1638,12 @@ with demo:
     with gr.Tab('Recursive Cut (Advanced)', visible=False) as tab_recursivecut_advanced:
 
         with gr.Row():
+            with gr.Column(scale=5, min_width=200):
+                input_gallery, submit_button, clear_images_button = make_input_images_section(rows=5, cols=3, height='auto')
+                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
+                num_images_slider.value = 100
+                clear_images_button.visible = False
+                logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
             with gr.Column(scale=5, min_width=200):
                 gr.Markdown('### Output (Recursion #1)')
                 l1_gallery = gr.Gallery(format='png', value=[], label="Recursion #1", show_label=True, elem_id="ncut_l1", columns=[3], rows=[5], object_fit="contain", height="auto", show_fullscreen_button=True, interactive=False)
@@ -1655,13 +1662,8 @@ with demo:
                 add_output_images_buttons(l3_gallery)
                 l3_norm_gallery = gr.Gallery(value=[], label="Recursion #3 Eigenvector Magnitude", show_label=True, elem_id="eig_norm", columns=[3], rows=[1], object_fit="contain", height="auto", show_share_button=True, preview=False, interactive=False)
                 l3_cluster_gallery = gr.Gallery(value=[], label="Recursion #3 Clusters", show_label=True, elem_id="clusters", columns=[2], rows=[4], object_fit="contain", height=600, show_share_button=True, preview=True, interactive=False)
-        with gr.Row():
-            with gr.Column(scale=5, min_width=200):
-                input_gallery, submit_button, clear_images_button = make_input_images_section(rows=5, height=1000)
-                dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
-                num_images_slider.value = 100
-                clear_images_button.visible = False
-                
+        
+        with gr.Row():        
             with gr.Column(scale=5, min_width=200):
                 with gr.Accordion("➡️ Recursion config", open=True):
                     l1_num_eig_slider = gr.Slider(1, 1000, step=1, label="Recursion #1: N eigenvectors", value=100, elem_id="l1_num_eig")
@@ -1671,6 +1673,7 @@ with demo:
                     l1_affinity_focal_gamma_slider = gr.Slider(0.01, 1, step=0.01, label="Recursion #1: Affinity focal gamma", value=0.5, elem_id="recursion_l1_gamma")
                     l2_affinity_focal_gamma_slider = gr.Slider(0.01, 1, step=0.01, label="Recursion #2: Affinity focal gamma", value=0.5, elem_id="recursion_l2_gamma")
                     l3_affinity_focal_gamma_slider = gr.Slider(0.01, 1, step=0.01, label="Recursion #3: Affinity focal gamma", value=0.5, elem_id="recursion_l3_gamma")
+            with gr.Column(scale=5, min_width=200):
                 [
                     model_dropdown, layer_slider, node_type_dropdown, num_eig_slider, 
                     affinity_focal_gamma_slider, num_sample_ncut_slider, ncut_knn_slider, ncut_indirect_connection, ncut_make_orthogonal, 
@@ -1680,7 +1683,6 @@ with demo:
                 ] = make_parameters_section()
                 num_eig_slider.visible = False
                 affinity_focal_gamma_slider.visible = False
-                logging_text = gr.Textbox("Logging information", label="Logging", elem_id="logging", type="text", placeholder="Logging information")
         true_placeholder = gr.Checkbox(label="True placeholder", value=True, elem_id="true_placeholder")
         true_placeholder.visible = False
         false_placeholder = gr.Checkbox(label="False placeholder", value=False, elem_id="false_placeholder")
@@ -2002,7 +2004,7 @@ with demo:
                 dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
 
                 
-            for i in range(2):
+            for i in range(3):
                 add_one_model()
                 
         # Create rows and buttons in a loop
@@ -2014,7 +2016,7 @@ with demo:
             rows.append(row)
             
             with row:
-                for j in range(3):
+                for j in range(4):
                     with gr.Column(scale=5, min_width=200):
                         add_one_model()
 
@@ -2077,7 +2079,7 @@ with demo:
                 dataset_dropdown, num_images_slider, random_seed_slider, load_images_button = make_dataset_images_section(advanced=True)
 
                 
-            for i in range(2):
+            for i in range(3):
                 add_one_model()
                 
         # Create rows and buttons in a loop
@@ -2089,7 +2091,7 @@ with demo:
             rows.append(row)
             
             with row:
-                for j in range(3):
+                for j in range(4):
                     with gr.Column(scale=5, min_width=200):
                         add_one_model()