Upload app.py

app.py

@@ -281,6 +281,18 @@ def delete_sample(sample_id):
         conn.execute("DELETE FROM samples WHERE id = ?", (sample_id,))
 
 
+def set_sample_transcript(sample_id, transcript):
+    with get_conn() as conn:
+        conn.execute("UPDATE samples SET transcript = ? WHERE id = ?",
+                     ((transcript or "").strip(), sample_id))
+
+
+def get_sample_transcript(sample_id):
+    with get_conn() as conn:
+        row = conn.execute("SELECT transcript FROM samples WHERE id = ?", (sample_id,)).fetchone()
+    return (row["transcript"] if row else "") or ""
+
+
 # --------------------------------------------------------------------------- #
 # Ratings
 # --------------------------------------------------------------------------- #
@@ -490,6 +502,9 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
 
                 rate_audio = gr.Audio(label="▶  Listen to the full sample",
                                       type="filepath", interactive=False)
+                rate_transcript = gr.Textbox(
+                    label="📄  Reference text (what the speaker should be saying)",
+                    interactive=False, lines=2, elem_id="ref-text")
                 gr.Markdown(
                     "Use headphones in a quiet room.    "
                     "**1** Very Poor · **2** Poor · **3** Fair · **4** Good · **5** Excellent",
@@ -554,6 +569,16 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
                                          allow_custom_value=True, scale=3)
                 del_btn = gr.Button("Delete selected", variant="stop", scale=0, size="sm")
 
+            gr.Markdown("#### Reference text per sample")
+            gr.Markdown("Pick a sample to view its current reference text, edit it, and save. "
+                        "This text is shown to reviewers beside the audio.")
+            with gr.Row():
+                tr_sample = gr.Dropdown(label="Sample", choices=[],
+                                        allow_custom_value=True, scale=2)
+                tr_text = gr.Textbox(label="Reference text", lines=2, scale=3)
+                tr_btn = gr.Button("Save text", scale=0, size="sm")
+            tr_msg = gr.Markdown()
+
             gr.Markdown("### Users")
             refresh_users_btn = gr.Button("Refresh users", size="sm")
             users_tbl = gr.Dataframe(
@@ -646,18 +671,19 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
 
     def load_sample(sess, sample_id):
         if not sample_id:
-            return (None, None, *[None] * len(CRITERIA_KEYS), "")
+            return (None, "", None, *[None] * len(CRITERIA_KEYS), "")
         with get_conn() as conn:
             row = conn.execute("SELECT * FROM samples WHERE id=?", (sample_id,)).fetchone()
         if not row:
-            return (None, None, *[None] * len(CRITERIA_KEYS), "")
+            return (None, "", None, *[None] * len(CRITERIA_KEYS), "")
+        transcript = row["transcript"] or "— No reference text provided for this sample —"
         existing = get_rating(sess["id"], sample_id) if sess else None
         scores = [existing[k] if existing else None for k in CRITERIA_KEYS]
         comments = existing["comments"] if existing else ""
-        return (row["file_path"], sample_id, *scores, comments)
+        return (row["file_path"], transcript, sample_id, *scores, comments)
     rate_sample.change(
         load_sample, [session, rate_sample],
-        [rate_audio, current_sample, *rate_outputs, rate_comments],
+        [rate_audio, rate_transcript, current_sample, *rate_outputs, rate_comments],
     )
 
     def go_next_unrated(sess, language_id):
@@ -742,11 +768,11 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
 
     def admin_upload(sess, language_id, files, model, is_ref, transcript):
         if not sess or sess["role"] != "admin":
-            return "❌ Admin only.", _samples_table(), gr.update()
+            return "❌ Admin only.", _samples_table(), gr.update(), gr.update()
         if not language_id:
-            return "❌ Choose a language first.", _samples_table(), gr.update()
+            return "❌ Choose a language first.", _samples_table(), gr.update(), gr.update()
         if not files:
-            return "❌ No files selected.", _samples_table(), gr.update()
+            return "❌ No files selected.", _samples_table(), gr.update(), gr.update()
         count = 0
         for f in files:
             path = f if isinstance(f, str) else getattr(f, "name", None)
@@ -754,23 +780,44 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
                 add_sample(language_id, path, model_name=model, is_reference=is_ref, transcript=transcript)
                 count += 1
             except Exception as e:  # noqa
+                ch = _sample_delete_choices()
                 return (f"❌ Error on a file: {e}", _samples_table(),
-                        gr.update(choices=_sample_delete_choices()))
+                        gr.update(choices=ch), gr.update(choices=ch))
+        ch = _sample_delete_choices()
         return (f"✅ Uploaded {count} sample(s).", _samples_table(),
-                gr.update(choices=_sample_delete_choices(), value=None))
+                gr.update(choices=ch, value=None), gr.update(choices=ch, value=None))
     up_btn.click(admin_upload, [session, up_lang, up_files, up_model, up_isref, up_transcript],
-                 [up_msg, samples_tbl, del_sample])
+                 [up_msg, samples_tbl, del_sample, tr_sample])
 
     def admin_delete_sample(sess, sid):
         if not sess or sess["role"] != "admin":
-            return "❌ Admin only.", _samples_table(), gr.update()
+            return "❌ Admin only.", _samples_table(), gr.update(), gr.update()
         if not sid:
+            ch = _sample_delete_choices()
             return ("❌ Select a sample to delete.", _samples_table(),
-                    gr.update(choices=_sample_delete_choices()))
+                    gr.update(choices=ch), gr.update(choices=ch))
         delete_sample(int(sid))
+        ch = _sample_delete_choices()
         return (f"✅ Deleted sample #{int(sid)} (and any ratings it had).", _samples_table(),
-                gr.update(choices=_sample_delete_choices(), value=None))
-    del_btn.click(admin_delete_sample, [session, del_sample], [up_msg, samples_tbl, del_sample])
+                gr.update(choices=ch, value=None), gr.update(choices=ch, value=None))
+    del_btn.click(admin_delete_sample, [session, del_sample],
+                  [up_msg, samples_tbl, del_sample, tr_sample])
+
+    # ---- Admin: per-sample reference text ----
+    def load_transcript(sess, sid):
+        if not sess or sess["role"] != "admin" or not sid:
+            return ""
+        return get_sample_transcript(int(sid))
+    tr_sample.change(load_transcript, [session, tr_sample], [tr_text])
+
+    def save_transcript(sess, sid, text):
+        if not sess or sess["role"] != "admin":
+            return "❌ Admin only."
+        if not sid:
+            return "❌ Select a sample first."
+        set_sample_transcript(int(sid), text)
+        return f"✅ Reference text saved for sample #{int(sid)}."
+    tr_btn.click(save_transcript, [session, tr_sample, tr_text], [tr_msg])
 
     # ---- Admin: users ----
     def _users_table():
@@ -816,11 +863,12 @@ with gr.Blocks(title="Plotweaver AI — TTS MOS Evaluation",
         return export_results(language_id)
     export_btn.click(admin_export, [session, res_lang], [res_file])
 
-    # Populate admin tables + delete picker right after an admin logs in.
+    # Populate admin tables + sample pickers right after an admin logs in.
     li_btn.click(lambda s: (_languages_table(), _samples_table(), _users_table(),
+                            gr.update(choices=_sample_delete_choices()),
                             gr.update(choices=_sample_delete_choices()))
-                 if s and s["role"] == "admin" else ([], [], [], gr.update(choices=[])),
-                 [session], [langs_tbl, samples_tbl, users_tbl, del_sample])
+                 if s and s["role"] == "admin" else ([], [], [], gr.update(choices=[]), gr.update(choices=[])),
+                 [session], [langs_tbl, samples_tbl, users_tbl, del_sample, tr_sample])
 
     # Refresh language-dependent choices on every page load, so newly added
     # languages appear in the signup form (and admin dropdowns) without a restart.