Update app.py

app.py

@@ -10,7 +10,7 @@ except ImportError:
 # ------------------------------
 # CONFIG
 # ------------------------------
-CODES = [
+DEFAULT_CODES = [
     "Communication Barrier",
     "Emotional Support",
     "Future Aspirations",
@@ -91,11 +91,10 @@ def build_transcript_html(text, df):
     transcript.addEventListener('mouseup', function() {{
         const sel = window.getSelection().toString();
         if(sel.length>0){{
-            // store in hidden input
             const state_input = document.querySelector('#selected_segment_state');
             if(state_input) {{
                 state_input.value = sel;
-                state_input.dispatchEvent(new Event("input",{ {bubbles:true} }));
+                state_input.dispatchEvent(new Event("input", {{bubbles:true}}));
             }}
         }}
     }});
@@ -114,6 +113,14 @@ def apply_code(df, segment, code, file_id, *metadata_values):
     df = pd.concat([df, pd.DataFrame([new_row])], ignore_index=True)
     return df, f"✅ Segment coded as '{code}'"
 
+# ------------------------------
+# ADD NEW CODE CATEGORY
+# ------------------------------
+def add_new_category(new_code, code_list):
+    if new_code and new_code not in code_list:
+        code_list.append(new_code)
+    return code_list
+
 # ------------------------------
 # EXPORT XLSX
 # ------------------------------
@@ -134,6 +141,7 @@ with gr.Blocks() as demo:
     file_id = gr.State("")
     coded_df_state = gr.State(get_empty_df())
     selected_segment_state = gr.State("")
+    code_categories_state = gr.State(DEFAULT_CODES)
 
     # ---------------- Metadata Top ----------------
     with gr.Row():
@@ -146,17 +154,16 @@ with gr.Blocks() as demo:
         # Left: transcript
         with gr.Column(scale=3):
             transcript_html = gr.HTML()
-            # Hidden state to store selected segment
-            selected_segment = gr.Textbox(label="Selected segment (hidden)", interactive=False, visible=False, elem_id="selected_segment_state")
+            selected_segment_hidden = gr.Textbox(label="hidden", interactive=False, visible=False, elem_id="selected_segment_state")
 
         # Right: code buttons + table
         with gr.Column(scale=2):
-            gr.Markdown("## 🏷️ Code Categories")
-            code_buttons = []
-            for c in CODES:
-                btn = gr.Button(c)
-                code_buttons.append(btn)
-            gr.Markdown("## 📊 Coded Segments")
+            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 Categories (Click to apply)")
+            code_buttons_box = gr.Column()
             table = gr.Dataframe(interactive=False)
 
             export_btn = gr.Button("Export XLSX")
@@ -175,15 +182,26 @@ 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
-    for btn, code_name in zip(code_buttons, CODES):
-        btn.click(
-            apply_code,
-            inputs=[coded_df_state, selected_segment_state, gr.State(code_name), file_id] + metadata_inputs,
-            outputs=[coded_df_state, status]
-        )
-
-    # Update table
+    def make_apply_func(code_name):
+        return lambda df, segment, fid, *metas: apply_code(df, segment, code_name, fid, *metas)
+
+    # Table update
     coded_df_state.change(lambda df: df, inputs=coded_df_state, outputs=table)
 
     # Export