Update app.py

app.py

@@ -20,15 +20,14 @@ classifier = pipeline("text-classification", model="distilbert/distilbert-base-u
 generator = pipeline("text2text-generation", model="google/flan-t5-base")
 
 # Function to classify customer comments
-# https://huggingface.co/docs/hub/spaces-zerogpu
 @spaces.GPU
-def classify_comments():
+def classify_comments(category_boxes):
     sentiments = []
     categories = []
     results = []
     for comment in df['customer_comment']:
         sentiment = classifier(comment)[0]['label']
-        category_list = [box.value for box in category_boxes if box.value.strip() != '']
+        category_list = [box for box in category_boxes if box.strip() != '']
         category_str = ', '.join([cat.strip() for cat in category_list])
         prompt = f"What category best describes this comment? '{comment}' Please answer using only the name of the category: {category_str}."
         category = generator(prompt, max_length=30)[0]['generated_text']
@@ -48,54 +47,27 @@ with gr.Blocks() as nps:
     category_boxes = gr.State([])  # Store category input boxes as state
     
     def display_categories(categories):
-        category_column.children = []  # 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_column.children = []  # Reset children to 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():
                 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.append(gr.Row())
         return category_components
+
+    def remove_category(category, category_list):
+        category_list.remove(category)  # Remove selected category
+        return category_list
+
+    category_column = gr.Column()
+    
     with gr.Row():
         category_input = gr.Textbox(label="New Category", placeholder="Enter category name")
         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)
-    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),
-                gr.Button("X", elem_id=f"remove_{i}")
-            ])
-            components.append(row)
-        return components
-    category_column = gr.Column()
-    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")
@@ -128,4 +100,4 @@ with gr.Blocks() as nps:
     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()
+nps.launch()