toggle inputs on tf ko

app.py

@@ -444,6 +444,12 @@ 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):
+    if tf_knockout:
+        return gr.update(visible=False)
+    else:
+        return gr.update(visible=True)
+
 
 with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
     with gr.Row():
@@ -452,11 +458,12 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
             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)
-            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)
+            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 = []
@@ -492,6 +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)
     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):