Upload app.py with huggingface_hub

app.py

@@ -243,31 +243,44 @@ def stream_predict(chunk, state):
 def add_label_cb(label):
     label = (label or "").strip()
     if not label:
-        return gr.update(), "ラベル名が空", dataset_stats_text(), dataset_table(), gr.update()
+        return gr.update(), dataset_table(), gr.update()
     if label not in LABELS:
         LABELS.append(label)
-    return gr.update(choices=LABELS, value=label), f"+ {label}", dataset_stats_text(), dataset_table(), gr.update(choices=LABELS)
+    return gr.update(choices=LABELS, value=label), dataset_table(), gr.update(choices=LABELS)
 
 def add_sample_cb(audio, label):
     label = (label or "").strip()
     if label not in LABELS:
-        return "ラベル未選択", dataset_stats_text(), dataset_table(), gr.update(value=None)
+        return dataset_table(), gr.update(value=None)
     if audio is None:
-        return "録音なし", dataset_stats_text(), dataset_table(), gr.update(value=None)
+        return dataset_table(), gr.update(value=None)
 
     audio_n = normalize_audio_tuple(audio)
     U = audio_to_sequence(audio_n)
     if U is None or len(U) < 5:
-        return "短すぎ", dataset_stats_text(), dataset_table(), gr.update(value=None)
+        return dataset_table(), gr.update(value=None)
 
     DATA.append({"audio": audio_n, "U": U, "label": label})
-    return f"追加 idx={len(DATA)-1}", dataset_stats_text(), dataset_table(), gr.update(value=None)
+    return dataset_table(), gr.update(value=None)
 
 def undo_last_cb():
     if len(DATA) == 0:
-        return "取り消しなし", dataset_stats_text(), dataset_table()
-    last = DATA.pop()
-    return f"undo ({last['label']})", dataset_stats_text(), dataset_table()
+        return dataset_table()
+    DATA.pop()
+    return dataset_table()
+
+def reset_all_cb():
+    global MODEL
+    DATA.clear()
+    LABELS.clear()
+    MODEL = None
+    return (
+        dataset_table(),
+        gr.update(choices=[], value=None),
+        gr.update(choices=[], value=None),
+        gr.update(value=None),
+        None,
+    )
 
 def clear_rec_cb():
     return gr.update(value=None)
@@ -294,20 +307,17 @@ def apply_relabel_cb(table, new_label):
 def relabel_selected_cb(selected_idx, new_label):
     new_label = (new_label or "").strip()
     if selected_idx is None or selected_idx < 0 or selected_idx >= len(DATA):
-        return "未選択", dataset_stats_text(), dataset_table()
+        return dataset_table()
     if new_label not in LABELS:
-        return "ラベル未登録", dataset_stats_text(), dataset_table()
-    old = DATA[selected_idx]["label"]
+        return dataset_table()
     DATA[selected_idx]["label"] = new_label
-    return f"relabel idx={selected_idx}: {old} -> {new_label}", dataset_stats_text(), dataset_table()
+    return dataset_table()
 
 def delete_selected_cb(selected_idx):
     if selected_idx is None or selected_idx < 0 or selected_idx >= len(DATA):
-        return "未選択", dataset_stats_text(), dataset_table(), None, gr.update(value=None)
-    lab = DATA[selected_idx]["label"]
+        return dataset_table(), None, gr.update(value=None)
     DATA.pop(selected_idx)
-    # selection reset
-    return f"delete idx={selected_idx} ({lab})", dataset_stats_text(), dataset_table(), None, gr.update(value=None)
+    return dataset_table(), None, gr.update(value=None)
 
 # Compact UI — Copy.py風フェミニン・パステルデザイン + 写真風ブルー→グリーングラデーション
 HEAD = """
@@ -821,16 +831,14 @@ button[class*="primary"] * {
 
 with gr.Blocks() as demo:
 
-    # ヒーローセクション
+    # ヒーローセクション + リセットボタン
     gr.HTML("""
     <div class="hero-section">
         <h1>Sound Classify</h1>
         <p>Streaming ESN &#183; Record &#183; Learn &#183; Predict</p>
     </div>
     """)
-
-    # ステータス
-    stats = gr.Markdown(dataset_stats_text(), elem_classes=["section-title"])
+    reset_btn = gr.Button("Reset All", size="sm")
 
     with gr.Tabs():
 
@@ -841,7 +849,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 label_box = gr.Textbox(label="新ラベル", placeholder="例: yes", scale=3)
                 add_btn = gr.Button("追加", size="lg", scale=1)
-            log1 = gr.Textbox(label="ログ", interactive=False, lines=1)
 
             # 録音 & サンプル追加
             gr.Markdown("### 録音")
@@ -851,7 +858,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 undo_btn = gr.Button("Undo", size="lg")
                 rerec_btn = gr.Button("Clear", size="lg")
-            log2 = gr.Textbox(label="ログ", interactive=False, lines=1)
 
             # データ一覧 & 編集
             gr.Markdown("### データ一覧")
@@ -870,7 +876,6 @@ with gr.Blocks() as demo:
             with gr.Row():
                 relabel_btn = gr.Button("ラベル更新", size="lg")
                 del_btn = gr.Button("削除", size="lg")
-            log3 = gr.Textbox(label="ログ", interactive=False, lines=1)
 
         # ── 学習タブ ──
         with gr.Tab("学習"):
@@ -887,25 +892,26 @@ with gr.Blocks() as demo:
             stream_audio.stream(stream_predict, inputs=[stream_audio, st], outputs=[pred_box, prob_box, st])
 
     # wiring
-    add_btn.click(add_label_cb, inputs=[label_box], outputs=[label_dd, log1, stats, table, relabel_dd])
-    add_sample_btn.click(add_sample_cb, inputs=[audio_rec, label_dd], outputs=[log2, stats, table, audio_rec])
-    undo_btn.click(undo_last_cb, inputs=[], outputs=[log2, stats, table])
+    add_btn.click(add_label_cb, inputs=[label_box], outputs=[label_dd, table, relabel_dd])
+    add_sample_btn.click(add_sample_cb, inputs=[audio_rec, label_dd], outputs=[table, audio_rec])
+    undo_btn.click(undo_last_cb, inputs=[], outputs=[table])
     rerec_btn.click(clear_rec_cb, inputs=[], outputs=[audio_rec])
+    reset_btn.click(reset_all_cb, inputs=[], outputs=[table, label_dd, relabel_dd, audio_rec, selected_idx_state])
 
     # select row -> update state + replay + relabel dropdown value
     def _select_and_store(evt: gr.SelectData):
         if evt is None or evt.index is None:
-            return None, None, gr.update(value=None), "未選択"
+            return None, None, gr.update(value=None)
         row = evt.index[0] if isinstance(evt.index, (tuple, list)) else int(evt.index)
         if row < 0 or row >= len(DATA):
-            return None, None, gr.update(value=None), "範囲外"
+            return None, None, gr.update(value=None)
         item = DATA[row]
-        return row, item["audio"], gr.update(value=item["label"]), f"selected idx={row}"
+        return row, item["audio"], gr.update(value=item["label"])
 
-    table.select(_select_and_store, inputs=None, outputs=[selected_idx_state, replay_audio, relabel_dd, log3])
+    table.select(_select_and_store, inputs=None, outputs=[selected_idx_state, replay_audio, relabel_dd])
 
-    relabel_btn.click(relabel_selected_cb, inputs=[selected_idx_state, relabel_dd], outputs=[log3, stats, table])
-    del_btn.click(delete_selected_cb, inputs=[selected_idx_state], outputs=[log3, stats, table, selected_idx_state, relabel_dd])
+    relabel_btn.click(relabel_selected_cb, inputs=[selected_idx_state, relabel_dd], outputs=[table])
+    del_btn.click(delete_selected_cb, inputs=[selected_idx_state], outputs=[table, selected_idx_state, relabel_dd])
 
     train_btn.click(train_random, inputs=[trials], outputs=[train_log])
 if __name__ == "__main__":