Update app.py

app.py

@@ -114,9 +114,9 @@ def apply_code(df, segment, code, file_id, *metadata_values):
     return df, f"✅ Segment coded as '{code}'"
 
 # ------------------------------
-# ADD NEW CODE CATEGORY
+# ADD NEW CODE TO DROPDOWN
 # ------------------------------
-def add_new_category(new_code, code_list):
+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
@@ -143,7 +143,7 @@ with gr.Blocks() as demo:
     selected_segment_state = gr.State("")
     code_categories_state = gr.State(DEFAULT_CODES)
 
-    # ---------------- Metadata Top ----------------
+    # ---------------- Metadata ----------------
     with gr.Row():
         metadata_inputs = []
         for k,lbl in METADATA_FIELDS.items():
@@ -151,19 +151,21 @@ with gr.Blocks() as demo:
 
     # ---------------- Transcript + Coding ----------------
     with gr.Row():
-        # Left: transcript
+        # Left
         with gr.Column(scale=3):
             transcript_html = gr.HTML()
-            selected_segment_hidden = gr.Textbox(label="hidden", interactive=False, visible=False, elem_id="selected_segment_state")
+            hidden_segment = gr.Textbox(interactive=False, visible=False, elem_id="selected_segment_state")
 
-        # Right: code buttons + table
+        # Right
         with gr.Column(scale=2):
-            gr.Markdown("## 🏷️ Add New Code Category")
-            new_code_input = gr.Textbox(label="New code name")
-            add_code_btn = gr.Button("Add Code Category")
+            gr.Markdown("## 🏷️ Code Category")
+            code_dropdown = gr.Dropdown(label="Select code", choices=DEFAULT_CODES)
+            code_input = gr.Textbox(label="Or type new code")
+            add_code_btn = gr.Button("Add new code")
 
-            gr.Markdown("## 🏷️ Code Categories (Click to apply)")
-            code_buttons_box = gr.Column()
+            apply_btn = gr.Button("Apply code to selected text")
+
+            gr.Markdown("## 📊 Coded Segments")
             table = gr.Dataframe(interactive=False)
 
             export_btn = gr.Button("Export XLSX")
@@ -173,7 +175,6 @@ with gr.Blocks() as demo:
             status = gr.Textbox(label="Status", value="Ready")
 
     # ---------------- Callbacks ----------------
-    # Load file
     file_input.change(fn=process_file, inputs=file_input, outputs=[full_text, file_id, coded_df_state])
 
     # Update transcript HTML
@@ -182,26 +183,14 @@ with gr.Blocks() as demo:
     full_text.change(update_transcript, inputs=[full_text, coded_df_state], outputs=transcript_html)
     coded_df_state.change(update_transcript, inputs=[full_text, coded_df_state], outputs=transcript_html)
 
-    # Add new code category
-    add_code_btn.click(add_new_category, inputs=[new_code_input, code_categories_state], outputs=[code_categories_state])
-
-    # Dynamically create code buttons
-    def create_buttons(codes):
-        buttons = []
-        for c in codes:
-            b = gr.Button(c)
-            buttons.append(b)
-        return buttons
-    code_buttons_state = gr.State()
-    def update_buttons(codes):
-        return gr.update(value=codes)
-    code_categories_state.change(update_buttons, inputs=code_categories_state, outputs=code_buttons_state)
-
-    # Apply code buttons
-    def make_apply_func(code_name):
-        return lambda df, segment, fid, *metas: apply_code(df, segment, code_name, fid, *metas)
-
-    # Table update
+    # Add new code
+    add_code_btn.click(add_new_code, inputs=[code_input, code_categories_state], outputs=[code_categories_state])
+    code_categories_state.change(lambda codes: gr.update(choices=codes), inputs=code_categories_state, outputs=code_dropdown)
+
+    # Apply code
+    apply_btn.click(apply_code, inputs=[coded_df_state, selected_segment_state, code_dropdown, file_id] + metadata_inputs, outputs=[coded_df_state, status])
+
+    # Update table
     coded_df_state.change(lambda df: df, inputs=coded_df_state, outputs=table)
 
     # Export