make correlation and regression plots visible only after clicking button

folding_studio_demo/app.py

@@ -320,7 +320,7 @@ def create_correlation_tab():
         inputs=[prediction_dataframe],
         outputs=[prediction_dataframe],
     )
-    with gr.Row():
+    with gr.Row(visible=False) as correlation_row:
         with gr.Column(scale=0):
             with gr.Row():
                 correlation_type = gr.Radio(
@@ -338,7 +338,7 @@ def create_correlation_tab():
                 )
         with gr.Column():
             correlation_ranking_plot = gr.Plot(label="Correlation ranking")
-    with gr.Row():
+    with gr.Row(visible=False) as regression_row:
         with gr.Column(scale=0):
             
             # User can select the columns to display in the correlation plot
@@ -360,11 +360,21 @@ def create_correlation_tab():
             regression_plot = gr.Plot(label="Correlation with binding affinity")
 
     fake_predict_btn.click(
-        fn=lambda x: fake_predict_and_correlate(
-            spr_data_with_scores, SCORE_COLUMNS, ["Antibody Name", "KD (nM)"]
+        fn=lambda x: (
+            *fake_predict_and_correlate(
+                spr_data_with_scores, SCORE_COLUMNS, ["Antibody Name", "KD (nM)"]
+            ),
+            gr.Row(visible=True),
+            gr.Row(visible=True)
         ),
         inputs=[correlation_type],
-        outputs=[prediction_dataframe, correlation_ranking_plot, regression_plot],
+        outputs=[
+            prediction_dataframe,
+            correlation_ranking_plot,
+            regression_plot,
+            correlation_row,
+            regression_row,
+        ],
     )
 
     def update_plots_with_log(correlation_type, score, use_log):