handle categories better

app.py

@@ -43,11 +43,12 @@ with gr.Blocks() as nps:
     def add_category():
         new_box = gr.Textbox(label="Category", placeholder="Enter a category")
         category_boxes.append(new_box)
-        return new_box
+        return category_boxes
 
     category_boxes = []  # Store category input boxes
+    category_column = gr.Column()
     add_category_btn = gr.Button("Add Category")
-    add_category_btn.click(fn=add_category, outputs=category_boxes)
+    add_category_btn.click(fn=add_category, outputs=category_column)
     uploaded_file = gr.File(label="Upload CSV", type="filepath")
     template_btn = gr.Button("Use Template")
     gr.Markdown("# NPS Comment Categorization")
@@ -74,6 +75,7 @@ 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)
+    template_btn.click(fn=lambda: [gr.Textbox.update(value="Product Experience"), gr.Textbox.update(value="Customer Support"), gr.Textbox.update(value="Price of Service"), gr.Textbox.update(value="Other")], outputs=category_column)
     classify_btn.click(fn=classify_comments, inputs=category_boxes, outputs=output)
 
 nps.launch()