checkbox

app.py

@@ -401,11 +401,13 @@ def generate_heatmap(protein_fasta):
     return img
 
 
-def main_function_single(sequence):
+def main_function_single(sequence, show_explanation):
     # Process seq, and return outputs for both fam and don
     family_label, family_img, _, family_info = process_family_sequence(sequence)
     donor_label, donor_img, *_ = process_single_sequence(sequence)
-    figmap = generate_heatmap(sequence)
+    figmap = None
+    if show_explanation:
+        figmap = generate_heatmap(sequence)    
     return family_label, family_img, family_info, donor_label, donor_img, figmap
 
 def main_function_upload(protein_file): #, progress=gr.Progress()
@@ -415,7 +417,6 @@ prediction_imagefam = gr.outputs.Image(type='pil', label="Family prediction grap
 prediction_imagedonor = gr.outputs.Image(type='pil', label="Donor prediction graph")
 prediction_explain = gr.outputs.Image(type='pil', label="Donor prediction explanation")
 
-
 with gr.Blocks() as app:
     gr.Markdown("# Glydentify")
 
@@ -423,6 +424,7 @@ with gr.Blocks() as app:
         with gr.Row().style(equal_height=True):
             with gr.Column():
                 sequence = gr.inputs.Textbox(lines=16, placeholder='Enter Protein Sequence Here...', label="Protein Sequence")
+                explanation_checkbox = gr.inputs.Checkbox(label="Show Explanation", default=False)
             with gr.Column():
                 with gr.Accordion("Example:"):
                     gr.Markdown("""
@@ -445,9 +447,10 @@ with gr.Blocks() as app:
         with gr.Row().style(equal_height=True):
             with gr.Column():
                 predict_button = gr.Button("Predict")
-                predict_button.click(main_function_single, inputs=sequence,
-                                     outputs=[family_prediction, prediction_imagefam, info_markdown,
-                                              donor_prediction, prediction_imagedonor, prediction_explain])
+		predict_button.click(main_function_single, inputs=[sequence, explanation_checkbox],
+                     		     outputs=[family_prediction, prediction_imagefam, info_markdown,
+                                     donor_prediction, prediction_imagedonor, prediction_explain])
+
 
         # Family & Donor Section
         with gr.Row().style(equal_height=True):
@@ -458,8 +461,9 @@ with gr.Blocks() as app:
 
             # Explain Section
             with gr.Column():
-                with gr.Accordion("Donor explanation"):
-                    prediction_explain.render() # = gr.outputs.Image(type='pil', label="Donor prediction explaination")
+                if explanation_checkbox:  # Only render if the checkbox is checked
+                    with gr.Accordion("Donor explanation"):
+                        prediction_explain.render() # = gr.outputs.Image(type='pil', label="Donor prediction explaination")
     
     with gr.Tab("Multiple Sequence Prediction"):
         with gr.Row().style(equal_height=True):