refactor: set unvisible to df when alfred is running

app.py

@@ -16,31 +16,6 @@ ptbxl_df = pd.read_csv("./res/ptbxl_database.csv")
 ptbxl_df = ptbxl_df[selected_columns]
 
 
-def get_ecg_id_from_dataframe(df: pd.DataFrame, evt: gr.SelectData):
-    gr.Info(
-        "The analysis of the selected ECG may take up to 10 minutes. Please wait patiently.",
-        duration=15,
-    )
-    return evt.row_value[0]
-
-
-def get_ecg_image_from_dataframe(ecg_id):
-    response = requests.post(
-        f"{ALFRED_URL}/hf_demo/ptbxl_to_image", params={"ecg_id": ecg_id}, timeout=600
-    )
-    response.raise_for_status()
-    return imread(BytesIO(response.content))
-
-
-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 literal_eval(response.text)
-
-
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Textbox(
@@ -66,12 +41,43 @@ with gr.Blocks() as demo:
         alfred_result = gr.Markdown(
             value="", label="Alfred said that", min_height=300, container=True
         )
-    gr_df.select(fn=get_ecg_id_from_dataframe, inputs=gr_df, outputs=ecg_id_output)
+
+    def get_ecg_id_from_dataframe(df: pd.DataFrame, evt: gr.SelectData):
+        gr.Info(
+            "The analysis of the selected ECG may take up to 10 minutes. Please wait patiently.",
+            duration=15,
+        )
+        return {ecg_id_output: evt.row_value[0], gr_df: gr.Dataframe(visible=False)}
+
+    def get_ecg_image_from_dataframe(ecg_id):
+        response = requests.post(
+            f"{ALFRED_URL}/hf_demo/ptbxl_to_image",
+            params={"ecg_id": ecg_id},
+            timeout=600,
+        )
+        response.raise_for_status()
+        return imread(BytesIO(response.content))
+
+    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),
+        }
+
+    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
     )
     ecg_id_output.change(
-        fn=get_alfred_from_dataframe, inputs=ecg_id_output, outputs=alfred_result
+        fn=get_alfred_from_dataframe,
+        inputs=ecg_id_output,
+        outputs=[alfred_result, gr_df],
     )
 
 demo.launch()