Update app.py

app.py

@@ -48,36 +48,17 @@ with gr.Blocks() as nps:
             categories.append(new_category.strip())
         return categories, "", f"**Categories:**\n" + "\n".join([f"- {cat}" for cat in 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
-    def display_categories(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
+    # Function to reset categories
+    def reset_categories():
+        return [], "**Categories:**\n- None"
 
     # UI for adding categories
     with gr.Row():
         category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
         add_category_btn = gr.Button("Add Category")
+        reset_btn = gr.Button("Reset Categories")
         category_status = gr.Markdown("**Categories:**\n- None")
 
-    # Display added categories
-    category_display = gr.Column(visible=False)
-
     # File upload and template buttons
     uploaded_file = gr.File(label="Upload CSV", type="filepath")
     template_btn = gr.Button("Use Template")
@@ -114,10 +95,9 @@ with gr.Blocks() as nps:
         inputs=[categories, category_input],
         outputs=[categories, category_input, category_status]
     )
-    categories.change(
-        fn=display_categories,
-        inputs=categories,
-        outputs=category_display
+    reset_btn.click(
+        fn=reset_categories,
+        outputs=[categories, category_status]
     )
     uploaded_file.change(
         fn=load_data,