Upload app_space.py

app_space.py

@@ -9,6 +9,7 @@ import gradio as gr
 import matplotlib.pyplot as plt
 import numpy as np
 import matplotlib.image as mpimg
+import pandas as pd
 
 # Ensure local package is importable when running in Hugging Face Spaces
 ROOT = os.path.dirname(os.path.abspath(__file__))
@@ -75,18 +76,21 @@ def plot_gating(signal_text: str):
     buf.seek(0)
     np_img = mpimg.imread(buf)
     windows = meta.get("windows", [])
-    table = [
-        {
-            "start": w.get("start"),
-            "end": w.get("end"),
-            "significance": round(float(w.get("significance", 0.0)), 3),
-            "prob": round(float(w.get("probability", 0.0)), 3),
-            "selected": bool(w.get("selected")),
-        }
-        for w in windows
-    ]
+    table_rows = []
+    for w in windows:
+        table_rows.append(
+            [
+                w.get("start"),
+                w.get("end"),
+                round(float(w.get("significance", 0.0)), 3),
+                round(float(w.get("probability", 0.0)), 3),
+                bool(w.get("selected")),
+                bool(w.get("forced", False)),
+            ]
+        )
+    df = pd.DataFrame(table_rows, columns=["start", "end", "significance", "prob", "selected", "forced"])
     summary = f"Windows kept: {meta.get('selected_windows',0)}/{meta.get('total_windows',0)} • ratio={meta.get('ratio',1):.2f}"
-    return np_img, summary, table
+    return np_img, summary, df
 
 
 # Demo signals with more structure so gating can skip/keep meaningfully
@@ -117,7 +121,11 @@ with gr.Blocks(title="Sundew ECG Demo") as demo:
             )
             img = gr.Image(type="numpy", label="Raw vs Gated")
             summary_box = gr.Textbox(label="Gating summary")
-            table = gr.Dataframe(headers=["start", "end", "significance", "prob", "selected"], datatype=["number", "number", "number", "number", "bool"], wrap=True)
+            table = gr.Dataframe(
+                headers=["start", "end", "significance", "prob", "selected", "forced"],
+                datatype=["number", "number", "number", "number", "bool", "bool"],
+                wrap=True,
+            )
             btn2 = gr.Button("Show gating")
             btn2.click(plot_gating, inputs=inp2, outputs=[img, summary_box, table])
         with gr.Tab("Demos"):