refactor: call apis sequencially

app.py

@@ -19,7 +19,7 @@ ptbxl_df = ptbxl_df[selected_columns]
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Textbox(
-            "It takes about 10 seconds to load the PTB-XL table. Please wait for a moment. \nWhen you select the ECG to analyze from the PTB-XL table below, Alfred will begin the analysis.",
+            "It takes about 10 seconds to load the PTB-XL table. Please wait for a moment. \nWhen you select the ECG to analyze from the PTB-XL table below, Alfred will begin the analysis. \nThe analysis by Alfred may take up to 5 minutes. Please wait patiently.",
             label="Information",
             lines=2,
         )
@@ -44,40 +44,35 @@ with gr.Blocks() as demo:
 
     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.",
+            "The analysis of the selected ECG may take up to 5 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(
+    def get_analysis_and_image(ecg_id):
+        alfred_response = requests.post(
+            f"{ALFRED_URL}/hf_demo/alfred", params={"ecg_id": ecg_id}, timeout=600
+        )
+        alfred_response.raise_for_status()
+        alfred_result = literal_eval(alfred_response.text)
+
+        image_response = requests.post(
             f"{ALFRED_URL}/hf_demo/ptbxl_to_image",
             params={"ecg_id": ecg_id},
             timeout=600,
         )
-        response.raise_for_status()
-        return {
-            ecg_viewer: imread(BytesIO(response.content)),
-            gr_df: gr.Dataframe(visible=True),
-        }
+        image_response.raise_for_status()
+        ecg_image = imread(BytesIO(image_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)
+        return [alfred_result, ecg_image, 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,
+        fn=get_analysis_and_image,
         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
+        outputs=[alfred_result, ecg_viewer, gr_df],
     )
 
 demo.launch()