Update app.py

app.py

@@ -41,13 +41,23 @@ def classify_comments():
 # Gradio Interface
 with gr.Blocks() as nps:
     def add_category(category_list):
-        category_list.append(gr.Textbox(label="Category", placeholder="Enter a category"))
+        category_list.append("")  # Add an empty category field
         return category_list
 
     category_boxes = gr.State([])  # Store category input boxes as state
+    def display_categories(categories):
+        components = []
+        for i, cat in enumerate(categories):
+            row = gr.Row([
+                gr.Textbox(value=cat, label=f"Category {i+1}", interactive=True),
+                gr.Button("X", elem_id=f"remove_{i}")
+            ])
+            components.append(row)
+        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)
+    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")
@@ -75,8 +85,9 @@ with gr.Blocks() as nps:
     uploaded_file.change(fn=load_data, inputs=uploaded_file, outputs=output)
     template_btn.click(fn=lambda: "Using Template Dataset", outputs=output)
     def use_template():
-        return [gr.Textbox.update(value="Product Experience"), gr.Textbox.update(value="Customer Support"), gr.Textbox.update(value="Price of Service"), gr.Textbox.update(value="Other")]
+        return ["Product Experience", "Customer Support", "Price of Service", "Other"]
     template_btn.click(fn=use_template, outputs=category_boxes)
+    category_boxes.change(fn=display_categories, inputs=category_boxes, outputs=category_column)
     classify_btn.click(fn=classify_comments, inputs=category_boxes, outputs=output)
 
 nps.launch()