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.clear()  # 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_components = []
         for i, cat in enumerate(categories):
             with gr.Row():
@@ -55,7 +62,9 @@ with gr.Blocks() as nps:
                 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)
         return category_components
-    category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
+    with gr.Row():
+        category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
+        add_category_btn = gr.Button("Add Category")
     def remove_category(category, category_list):
         category_list.remove(category)  # Remove selected category
         return category_list
@@ -78,6 +87,7 @@ with gr.Blocks() as nps:
     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)
+    category_boxes.change(fn=display_categories, inputs=category_boxes, outputs=category_column)
     uploaded_file = gr.File(label="Upload CSV", type="filepath")
     template_btn = gr.Button("Use Template")
     gr.Markdown("# NPS Comment Categorization")