refactor: remove textbox of thresholds

app.py

@@ -55,7 +55,7 @@ def remove_short_duration(seg, th):
     return np.concatenate(seg_groups)
 
 
-def gen_each_image(input, seg, image_path, ths, pp=False):
+def gen_each_image(input, seg, image_path, ths=[5, 25, 25, 25, 15, 25], pp=False):
     fig = plt.figure(figsize=(15, 18))
     plt.subplots_adjust(left=0.02, right=0.98, top=0.98, bottom=0.02, hspace=0.2)
     for idx, (in_by_lead, seg_by_lead) in enumerate(zip(input, seg)):
@@ -98,11 +98,11 @@ def gen_each_image(input, seg, image_path, ths, pp=False):
     plt.close()
 
 
-def gen_image(subject_id, image_path, pp_image_path, ths):
+def gen_image(subject_id, image_path, pp_image_path):
     try:
         input, seg = gen_seg(subject_id)
-        gen_each_image(input, seg, image_path, ths)
-        gen_each_image(input, seg, pp_image_path, ths, True)
+        gen_each_image(input, seg, image_path)
+        gen_each_image(input, seg, pp_image_path, pp=True)
         return True
     except Exception:
         print(tb.format_exc())
@@ -113,7 +113,7 @@ with gr.Blocks() as demo:
     with gr.Tab("App"):
         with gr.Row():
             gr.Textbox(
-                "Welcome to visit ECG-Delineation space",
+                "Welcome to visit ECG-Delineation space.",
                 label="Information",
                 lines=1,
             )
@@ -148,45 +148,11 @@ with gr.Blocks() as demo:
                 )
             )
 
-        with gr.Row():
-            gr_ths = [
-                gr.Textbox(
-                    label="Interval Threshold of P (ms)",
-                    lines=1,
-                    value="10",
-                ),
-                gr.Textbox(
-                    label="Duration Threshold of P (ms)",
-                    lines=1,
-                    value="50",
-                ),
-                gr.Textbox(
-                    label="Interval Threshold of QRS (ms)",
-                    lines=1,
-                    value="50",
-                ),
-                gr.Textbox(
-                    label="Duration Threshold of QRS (ms)",
-                    lines=1,
-                    value="50",
-                ),
-                gr.Textbox(
-                    label="Interval Threshold of T (ms)",
-                    lines=1,
-                    value="30",
-                ),
-                gr.Textbox(
-                    label="Duration Threshold of T (ms)",
-                    lines=1,
-                    value="50",
-                ),
-            ]
-
         with gr.Row():
             gr_image = gr.Image(type="filepath", label="Output")
             gr_pp_image = gr.Image(type="filepath", label="PostProcessed Output")
 
-        def show_image(df: pd.DataFrame, evt: gr.SelectData, *ths):
+        def show_image(df: pd.DataFrame, evt: gr.SelectData):
             subject_id = evt.row_value[0]
 
             image_path = f"{tempfile.gettempdir()}/ludb_{subject_id}.png"
@@ -196,17 +162,11 @@ with gr.Blocks() as demo:
             ):
                 return [image_path, pp_image_path]
 
-            gen_image(
-                subject_id, image_path, pp_image_path, [int(th) / 2 for th in ths]
-            )
+            gen_image(subject_id, image_path, pp_image_path)
 
             return [image_path, pp_image_path]
 
         for gr_df in gr_dfs:
-            gr_df.select(
-                fn=show_image,
-                inputs=[gr_df, *gr_ths],
-                outputs=[gr_image, gr_pp_image],
-            )
+            gr_df.select(fn=show_image, inputs=[gr_df], outputs=[gr_image, gr_pp_image])
 
 demo.launch()