Update app.py

app.py

@@ -40,13 +40,24 @@ def classify_comments():
 
 # Gradio Interface
 with gr.Blocks() as nps:
-    def add_category(category_list):
-        category_list.append("")  # Add an empty category field
+    def add_category(category_list, new_category):
+        if new_category.strip() != "":
+            category_list.append(new_category.strip())  # Add new category
         return category_list
 
     category_boxes = gr.State([])  # Store category input boxes as state
-    def display_categories(categories):
+    category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
+    def remove_category(category, category_list):
+        category_list.remove(category)  # Remove selected category
+        return category_list
         components = []
+        for i, cat in enumerate(categories):
+            row = gr.Row([
+                gr.Markdown(f"- {cat}"),
+                gr.Button("X", elem_id=f"remove_{i}", interactive=True).click(fn=lambda x=cat: remove_category(x, categories), inputs=[], outputs=category_boxes)
+            ])
+            components.append(row)
+        return components
         for i, cat in enumerate(categories):
             row = gr.Row([
                 gr.Textbox(value=cat, label=f"Category {i+1}", interactive=True),
@@ -56,7 +67,7 @@ with gr.Blocks() as nps:
         return components
     category_column = gr.Row()
     add_category_btn = gr.Button("Add Category")
-    add_category_btn.click(fn=add_category, inputs=[category_boxes], outputs=category_boxes)
+    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)
     uploaded_file = gr.File(label="Upload CSV", type="filepath")
     template_btn = gr.Button("Use Template")