Upload app.py

app.py

@@ -244,6 +244,13 @@ def create_user(username, email, password, role="reviewer", language_ids=None):
     return uid
 
 
+def delete_user(user_id):
+    """Delete a user; cascades to their user_languages and ratings."""
+    with get_conn() as conn:
+        conn.execute("DELETE FROM users WHERE id = ?", (user_id,))
+    backup_db()
+
+
 def authenticate(username, password):
     with get_conn() as conn:
         row = conn.execute("SELECT * FROM users WHERE username = ?", ((username or "").strip(),)).fetchone()
@@ -377,6 +384,13 @@ def set_sample_reference(sample_id, is_ref):
     backup_db()
 
 
+def unmark_all_references():
+    with get_conn() as conn:
+        n = conn.execute("UPDATE samples SET is_reference = 0 WHERE is_reference = 1").rowcount
+    backup_db()
+    return n
+
+
 # --------------------------------------------------------------------------- #
 # Ratings
 # --------------------------------------------------------------------------- #
@@ -669,6 +683,8 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
                 tr_text = gr.Textbox(label="Reference text", lines=2, scale=3)
                 tr_btn = gr.Button("Save", scale=0, size="sm")
             tr_msg = gr.Markdown()
+            unmark_all_btn = gr.Button("⟲  Mark ALL samples as system (un-mark every anchor)", size="sm")
+            unmark_msg = gr.Markdown()
 
             gr.Markdown("### Users")
             refresh_users_btn = gr.Button("Refresh users", size="sm")
@@ -681,6 +697,12 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
                 active_choice = gr.Dropdown(label="Set active", choices=["yes", "no"], value="yes")
                 update_user_btn = gr.Button("Apply", scale=0, size="sm")
             user_admin_msg = gr.Markdown()
+            with gr.Row():
+                del_user_id = gr.Number(label="Delete user by id", precision=0)
+                del_user_btn = gr.Button("Delete user", variant="stop", scale=0, size="sm")
+            gr.Markdown("Deleting a user also removes all of their ratings. To keep their ratings, "
+                        "set **active = no** instead.")
+            del_user_msg = gr.Markdown()
 
             gr.Markdown("### Results")
             with gr.Row():
@@ -918,6 +940,14 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
         return (f"✅ Saved sample #{int(sid)} — now a {flag}.", _samples_table())
     tr_btn.click(save_sample_meta, [session, tr_sample, tr_text, tr_isref], [tr_msg, samples_tbl])
 
+    def do_unmark_all(sess):
+        if not sess or sess["role"] != "admin":
+            return "❌ Admin only.", _samples_table(), gr.update()
+        n = unmark_all_references()
+        return (f"✅ Un-marked {n} sample(s) — they now count toward the MOS. "
+                f"Go to Results and click **Compute MOS**.", _samples_table(), gr.update(value=False))
+    unmark_all_btn.click(do_unmark_all, [session], [unmark_msg, samples_tbl, tr_isref])
+
     # ---- Admin: users ----
     def _users_table():
         rows = []
@@ -947,6 +977,23 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
     update_user_btn.click(admin_update_user, [session, promote_id, role_choice, active_choice],
                           [user_admin_msg, users_tbl])
 
+    def admin_delete_user(sess, uid):
+        if not sess or sess["role"] != "admin":
+            return "❌ Admin only.", _users_table()
+        if not uid:
+            return "❌ Enter a user id.", _users_table()
+        uid = int(uid)
+        if uid == sess["id"]:
+            return "❌ You can't delete your own account while signed in.", _users_table()
+        with get_conn() as conn:
+            row = conn.execute("SELECT username FROM users WHERE id=?", (uid,)).fetchone()
+        if not row:
+            return f"❌ No user with id {uid}.", _users_table()
+        delete_user(uid)
+        return (f"✅ Deleted user #{uid} ({row['username']}) and all of their ratings.",
+                _users_table())
+    del_user_btn.click(admin_delete_user, [session, del_user_id], [del_user_msg, users_tbl])
+
     # ---- Admin: results ----
     def admin_results(sess, language_id):
         if not sess or sess["role"] != "admin":