make ui better

app.py

@@ -444,31 +444,31 @@ def perturb(gene, locus1, locus2, use_max_gene, tf_knockout):
     return image_from_plot, violin_img, violin_img_new, violin_img_fc
 
 
-def update_on_tf_ko(tf_knockout):
+def update_on_tf_ko(tf_knockout, use_max_gene):
     if tf_knockout:
-        return gr.update(visible=False)
+        return gr.update(visible=False), True
     else:
-        return gr.update(visible=True)
+        return gr.update(visible=True), use_max_gene
 
 
 with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
     with gr.Row():
         with gr.Column():
-            in_locus = gr.Textbox(label="Target Gene", elem_id='in-locus', scale=1)
+            in_locus = gr.Textbox(label="Gene/TF", elem_id='in-locus', scale=1)
             with gr.Row():
                 max_gene_checkbox = gr.Checkbox(label="Max Gene", elem_id='max-gene-checkbox', scale=1, checked=False)
                 tf_ko = gr.Checkbox(label="TF KO", elem_id='tf-ko', scale=1, checked=False)
-            with gr.Column() as heatmap_col:
-                anchor1 = gr.Textbox(label="Locus 1 (gene or genomic coords)", elem_id='anchor1', scale=1)
-                anchor2 = gr.Textbox(label="Locus 2 (gene or genomic coords)", elem_id='anchor2', scale=1)
-                heatmap_x = gr.Textbox(label="Heatmap X", elem_id='heatmap-x', scale=1, visible=False)
-                heatmap_y = gr.Textbox(label="Heatmap Y", elem_id='heatmap-y', scale=1, visible=False)
-                heatmap_size = gr.Slider(label="Heatmap Range", info="kb range aroung input gene locus to expand", minimum=50, maximum=5000, value=250)
-    heatmap_button = gr.Button(value="Generate Heatmaps", elem_id='heatmap-button', scale=1)
-    with gr.Row():
-        out_heatmaps = []
-        for celltype in cooler_celltypes:
-            out_heatmaps.append(gr.Image(label=celltype, elem_id=f'out-heatmap-{celltype}', scale=1))
+    with gr.Column() as heatmap_col:
+        anchor1 = gr.Textbox(label="Locus 1 (gene or genomic coords)", elem_id='anchor1', scale=1)
+        anchor2 = gr.Textbox(label="Locus 2 (gene or genomic coords)", elem_id='anchor2', scale=1)
+        heatmap_x = gr.Textbox(label="Heatmap X", elem_id='heatmap-x', scale=1, visible=False)
+        heatmap_y = gr.Textbox(label="Heatmap Y", elem_id='heatmap-y', scale=1, visible=False)
+        heatmap_size = gr.Slider(label="Heatmap Range", info="kb range aroung input gene locus to expand", minimum=50, maximum=5000, value=250)
+        heatmap_button = gr.Button(value="Generate Heatmaps", elem_id='heatmap-button', scale=1)
+        with gr.Row():
+            out_heatmaps = []
+            for celltype in cooler_celltypes:
+                out_heatmaps.append(gr.Image(label=celltype, elem_id=f'out-heatmap-{celltype}', scale=1))
     with gr.Row():
         run_button = gr.Button(value="Run", elem_id='run-button', scale=1)
     with gr.Row():
@@ -499,7 +499,7 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
                 fn=perturb, cache_examples=os.getenv('SYSTEM') == 'spaces')
     run_button.click(perturb, inputs, outputs=outputs)
     heatmap_button.click(get_heatmaps, [in_locus, anchor1, anchor2, heatmap_size], outputs=out_heatmaps)
-    tf_ko.change(update_on_tf_ko, tf_ko, outputs=heatmap_col)
+    tf_ko.change(update_on_tf_ko, [tf_ko, max_gene_checkbox], outputs=[heatmap_col, max_gene_checkbox])
     anchor1.change(get_heatmaps, [in_locus, anchor1, anchor2, heatmap_size], outputs=out_heatmaps)
     anchor2.change(get_heatmaps, [in_locus, anchor1, anchor2, heatmap_size], outputs=out_heatmaps)
     def set_loop(img, gene, locus1, locus2, heatmap_range, evt: gr.SelectData):