Update app.py

app.py

@@ -110,13 +110,17 @@ def build_html_transcript(text, coded_df):
 # ------------------------------
 # APPLY CODE
 # ------------------------------
-def apply_code(coded_df, file_id, full_text, segment, code, *metadata_values):
+def apply_code(coded_df, file_id, full_text, segment, code, code_list, *metadata_values):
     if not segment:
-        return coded_df, "⚠️ Select a segment first."
+        return coded_df, code_list, "⚠️ Select a segment first."
     if not code:
-        return coded_df, "⚠️ Enter or select a code first."
+        return coded_df, code_list, "⚠️ Enter or select a code first."
     if not file_id:
-        return coded_df, "⚠️ Upload a file first."
+        return coded_df, code_list, "⚠️ Upload a file first."
+
+    # Update code list dynamically
+    if code not in code_list:
+        code_list.append(code)
 
     meta_dict = dict(zip(METADATA_FIELDS.keys(), metadata_values))
     context = "Context unavailable"
@@ -136,7 +140,7 @@ def apply_code(coded_df, file_id, full_text, segment, code, *metadata_values):
         **meta_dict
     }
     new_df = pd.concat([coded_df, pd.DataFrame([new_row])], ignore_index=True)
-    return new_df, f"✅ Segment coded as '{code}'!"
+    return new_df, code_list, f"✅ Segment coded as '{code}'!"
 
 # ------------------------------
 # EXPORT XLSX
@@ -154,7 +158,7 @@ def export_excel(df):
 with gr.Blocks(title="Interactive Qualitative Coding") as demo:
     gr.Markdown("# 📑 Interactive Qualitative Coding Tool")
 
-    # states
+    # States
     file_id = gr.State("")
     full_text = gr.State("")
     coded_df_state = gr.State(get_empty_df())
@@ -201,12 +205,17 @@ with gr.Blocks(title="Interactive Qualitative Coding") as demo:
     # Fill code input when selecting dropdown
     code_dropdown.change(lambda x: x, inputs=code_dropdown, outputs=code_input)
 
-    # Apply code
+    # Apply code (updates coded data and code list)
     code_btn.click(
         fn=apply_code,
-        inputs=[coded_df_state, file_id, full_text, segment_box, code_input] + metadata_inputs,
-        outputs=[coded_df_state, status]
+        inputs=[coded_df_state, file_id, full_text, segment_box, code_input, code_categories_state] + metadata_inputs,
+        outputs=[coded_df_state, code_categories_state, status]
     )
+
+    # Update dropdown when code list changes
+    code_categories_state.change(lambda codes: gr.update(choices=codes), inputs=code_categories_state, outputs=code_dropdown)
+
+    # Update table when coded_df changes
     coded_df_state.change(lambda x: x, inputs=coded_df_state, outputs=table)
 
     # Export