refactor: polish

.gitattributes

@@ -32,4 +32,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.xz filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -textres/screenshot_alfred-min.png filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.png filter=lfs diff=lfs merge=lfs -text

app.py

@@ -17,62 +17,75 @@ 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. \nThe analysis by Alfred may take up to 5 minutes. Please wait patiently.",
-            label="Information",
-            lines=2,
-        )
-    with gr.Row():
-        gr_df = gr.Dataframe(
-            value=ptbxl_df,
-            interactive=False,
-            max_height=200,
-            label="All PTB-XL v1.3.0 data (https://physionet.org/content/ptb-xl/1.0.3/ptbxl_database.csv). You can refer to the following URL(https://physionet.org/content/ptb-xl/1.0.3/scp_statements.csv) to understand what 'scp_codes' represent.",
-        )
+    with gr.Tab("Example"):
+        with gr.Row():
+            gr.Textbox(
+                "The test app is currently not functioning properly due to DNS issues. \nBelow is the output of the actual app running internally.",
+                label="Information",
+                lines=2,
+            )
+        with gr.Row():
+            gr.Image("./res/screenshot_alfred-min.png")
 
-    with gr.Row():
-        ecg_id_output = gr.Textbox(label="The selected ecg_id is", interactive=False)
+    with gr.Tab("App"):
+        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. \nThe analysis by Alfred may take up to 5 minutes. Please wait patiently.",
+                label="Information",
+                lines=3,
+            )
+        with gr.Row():
+            gr_df = gr.Dataframe(
+                value=ptbxl_df,
+                interactive=False,
+                max_height=200,
+                label="All PTB-XL v1.3.0 data (https://physionet.org/content/ptb-xl/1.0.3/ptbxl_database.csv). You can refer to the following URL(https://physionet.org/content/ptb-xl/1.0.3/scp_statements.csv) to understand what 'scp_codes' represent.",
+            )
 
-    with gr.Row():
-        ecg_viewer = gr.Image(interactive=False, label="The selected ecg is")
+        with gr.Row():
+            ecg_id_output = gr.Textbox(
+                label="The selected ecg_id is", interactive=False
+            )
 
-    with gr.Row():
-        alfred_result = gr.Markdown(
-            value="", label="Alfred said that", min_height=300, container=True
-        )
+        with gr.Row():
+            ecg_viewer = gr.Image(interactive=False, label="The selected ecg is")
 
-    def get_ecg_id_from_dataframe(df: pd.DataFrame, evt: gr.SelectData):
-        gr.Info(
-            "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)}
+        with gr.Row():
+            alfred_result = gr.Markdown(
+                value="", label="Alfred said that", min_height=300, container=True
+            )
 
-    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)
+        def get_ecg_id_from_dataframe(df: pd.DataFrame, evt: gr.SelectData):
+            gr.Info(
+                "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)}
 
-        image_response = requests.post(
-            f"{ALFRED_URL}/hf_demo/ptbxl_to_image",
-            params={"ecg_id": ecg_id},
-            timeout=600,
+        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,
+            )
+            image_response.raise_for_status()
+            ecg_image = imread(BytesIO(image_response.content))
+
+            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_analysis_and_image,
+            inputs=ecg_id_output,
+            outputs=[alfred_result, ecg_viewer, gr_df],
         )
-        image_response.raise_for_status()
-        ecg_image = imread(BytesIO(image_response.content))
-
-        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_analysis_and_image,
-        inputs=ecg_id_output,
-        outputs=[alfred_result, ecg_viewer, gr_df],
-    )
 
 demo.launch()