Update app.py

app.py

@@ -23,14 +23,6 @@ METADATA_FIELDS = {
     "age": "Age",
 }
 
-COLOR_MAP = {
-    "Communication Barrier": "lightblue",
-    "Emotional Support": "lightgreen",
-    "Future Aspirations": "khaki",
-    "Financial Stress": "lightpink",
-    "Follow-up Needed": "orange",
-}
-
 # --- File processing ---
 def read_docx(path):
     if not docx:
@@ -70,15 +62,18 @@ def process_file(file_obj):
 # --- Apply coding ---
 def apply_code(df, segment, code, file_id, *metadata_values):
     if not file_id:
-        return df, "⚠️ Upload a file first"
+        return df, "⚠️ Upload a file first", gr.update(value="")
     if not segment:
-        return df, "⚠️ Paste a segment first"
+        return df, "⚠️ Paste a segment first", gr.update(value="")
     if not code:
-        return df, "⚠️ Select a code"
+        return df, "⚠️ Select a code", gr.update(value="")
+    
     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}'"
+    
+    # Clear segment box after applying
+    return df, f"✅ Segment coded as '{code}'", gr.update(value="")
 
 # --- Add new code ---
 def add_new_code(new_code, code_list):
@@ -165,7 +160,7 @@ with gr.Blocks() as demo:
     apply_btn.click(
         apply_code,
         inputs=[coded_df_state, segment_box, code_dropdown, file_id] + metadata_inputs,
-        outputs=[coded_df_state, status]
+        outputs=[coded_df_state, status, segment_box]
     )
 
     coded_df_state.change(lambda df: df, inputs=coded_df_state, outputs=table)