Update app.py

app.py

@@ -46,11 +46,28 @@ with gr.Blocks() as nps:
     def add_category(categories, new_category):
         if new_category.strip() != "" and len(categories) < 5:  # Limit to 5 categories
             categories.append(new_category.strip())
-        return categories, "", f"**Categories:**\n" + "\n".join([f"- {cat}" for cat in categories])
+        return categories, "", f"**Categories:**\n" + "\n".join([f"- {cat} 🗑️" for cat in categories])
 
-    # Function to display categories
+    # Function to remove a category
+    def remove_category(categories, category_to_remove):
+        if category_to_remove in categories:
+            categories.remove(category_to_remove)
+        return categories, f"**Categories:**\n" + "\n".join([f"- {cat} 🗑️" for cat in categories])
+
+    # Function to display categories with remove buttons
     def display_categories(categories):
-        return gr.Column.update(visible=True, value=[gr.Markdown(f"- {cat}") for cat in categories])
+        category_rows = []
+        for cat in categories:
+            with gr.Row():
+                gr.Markdown(f"- {cat}")
+                remove_btn = gr.Button("🗑️", elem_id=f"remove_{cat}")
+                remove_btn.click(
+                    fn=remove_category,
+                    inputs=[categories, gr.State(cat)],
+                    outputs=[categories, category_status]
+                )
+            category_rows.append(gr.Row())
+        return category_rows
 
     # UI for adding categories
     with gr.Row():
@@ -89,7 +106,7 @@ with gr.Blocks() as nps:
     # Function to use template categories
     def use_template():
         template_categories = ["Product Experience", "Customer Support", "Price of Service", "Other"]
-        return template_categories, f"**Categories:**\n" + "\n".join([f"- {cat}" for cat in template_categories])
+        return template_categories, f"**Categories:**\n" + "\n".join([f"- {cat} 🗑️" for cat in template_categories])
 
     # Event handlers
     add_category_btn.click(