Update app.py

app.py

@@ -30,14 +30,12 @@ COLOR_MAP = {
     "Follow-up Needed": "orange",
 }
 
-
 def read_docx(path):
     if not docx:
         return "Error: python-docx not installed."
     d = docx.Document(path)
     return "\n".join([p.text for p in d.paragraphs])
 
-
 def read_vtt(path):
     with open(path, "r", encoding="utf-8") as f:
         lines = f.read().split("\n")
@@ -48,13 +46,11 @@ def read_vtt(path):
     ]
     return " ".join(cleaned)
 
-
 def get_empty_df():
     return pd.DataFrame(
         columns=["File ID", "Coded Segment", "Code"] + list(METADATA_FIELDS.keys())
     )
 
-
 def process_file(file_obj):
     if file_obj is None:
         return "", "", get_empty_df()
@@ -69,7 +65,6 @@ def process_file(file_obj):
             text = f.read()
     return text, name, get_empty_df()
 
-
 def build_transcript_html(text, df):
     display_text = text
     if df is not None and not df.empty:
@@ -102,22 +97,23 @@ def build_transcript_html(text, df):
     """
     return html
 
-
 def apply_code(df, segment, code, file_id, *metadata_values):
-    if not segment or not code or not file_id:
-        return df, "⚠️ Select text and file first"
+    if not file_id:
+        return df, "⚠️ Upload a file first"
+    if not segment:
+        return df, "⚠️ Select text in transcript"
+    if not code:
+        return df, "⚠️ Select a code"
     meta_dict = dict(zip(METADATA_FIELDS.keys(), metadata_values))
     new_row = {"File ID": file_id, "Coded Segment": segment, "Code": code, **meta_dict}
     df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
     return df, f"✅ Segment coded as '{code}'"
 
-
 def add_new_code(new_code, code_list):
     if new_code and new_code not in code_list:
         code_list.append(new_code)
     return code_list
 
-
 def export_excel(df):
     if df.empty:
         return None, "Nothing to export"
@@ -125,8 +121,8 @@ def export_excel(df):
     df.to_excel(path, index=False)
     return path, "Excel ready"
 
-
 with gr.Blocks() as demo:
+
     # States
     full_text = gr.State("")
     file_id = gr.State("")
@@ -140,7 +136,7 @@ with gr.Blocks() as demo:
         for k, lbl in METADATA_FIELDS.items():
             metadata_inputs.append(gr.Textbox(label=lbl))
 
-    # Transcript + coding
+    # Main row: transcript left, coding right
     with gr.Row():
         # Left: transcript
         with gr.Column(scale=3):
@@ -193,10 +189,10 @@ with gr.Blocks() as demo:
         outputs=code_dropdown,
     )
 
+    # Apply button uses hidden textbox as input
     apply_btn.click(
         apply_code,
-        inputs=[coded_df_state, selected_segment_state, code_dropdown, file_id]
-        + metadata_inputs,
+        inputs=[coded_df_state, hidden_segment, code_dropdown, file_id] + metadata_inputs,
         outputs=[coded_df_state, status],
     )