refactor: polish visibility logic

app.py

@@ -56,28 +56,28 @@ with gr.Blocks() as demo:
             timeout=600,
         )
         response.raise_for_status()
-        return imread(BytesIO(response.content))
+        return {
+            ecg_viewer: imread(BytesIO(response.content)),
+            gr_df: gr.Dataframe(visible=True),
+        }
 
     def get_alfred_from_dataframe(ecg_id):
         response = requests.post(
             f"{ALFRED_URL}/hf_demo/alfred", params={"ecg_id": ecg_id}, timeout=600
         )
         response.raise_for_status()
-        return {
-            alfred_result: literal_eval(response.text),
-            gr_df: gr.Dataframe(visible=True),
-        }
+        return literal_eval(response.text)
 
     gr_df.select(
         fn=get_ecg_id_from_dataframe, inputs=gr_df, outputs=[ecg_id_output, gr_df]
     )
     ecg_id_output.change(
-        fn=get_ecg_image_from_dataframe, inputs=ecg_id_output, outputs=ecg_viewer
+        fn=get_ecg_image_from_dataframe,
+        inputs=ecg_id_output,
+        outputs=[ecg_viewer, gr_df],
     )
     ecg_id_output.change(
-        fn=get_alfred_from_dataframe,
-        inputs=ecg_id_output,
-        outputs=[alfred_result, gr_df],
+        fn=get_alfred_from_dataframe, inputs=ecg_id_output, outputs=alfred_result
     )
 
 demo.launch()