Update app.py

app.py

@@ -48,6 +48,13 @@ with gr.Blocks() as nps:
     category_boxes = gr.State([])  # Store category input boxes as state
     
     def display_categories(categories):
+        category_column.children = []  # Clear previous categories
+        for i, cat in enumerate(categories):
+            with category_column:
+                with gr.Row():
+                    gr.Markdown(f"- {cat}")
+                    remove_btn = gr.Button("X", elem_id=f"remove_{i}", interactive=True)
+                    remove_btn.click(fn=lambda x=cat: remove_category(x, categories), inputs=[], outputs=category_boxes)
         category_column.children = []  # Reset children to clear previous categories
         for i, cat in enumerate(categories):
             with category_column:
@@ -65,6 +72,8 @@ with gr.Blocks() as nps:
     with gr.Row():
         category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
         add_category_btn = gr.Button("Add Category")
+        add_category_btn.click(fn=add_category, inputs=[category_boxes, category_input], outputs=category_boxes)
+        category_boxes.change(fn=display_categories, inputs=category_boxes, outputs=category_column)
     def remove_category(category, category_list):
         category_list.remove(category)  # Remove selected category
         return category_list