Update app.py

app.py

@@ -38,6 +38,39 @@ def classify_comments(categories):
     df['comment_category'] = assigned_categories
     return df.to_html(index=False)  # Return all fields with appended sentiment and category
 
+# Function to add a category
+def add_category(categories, new_category):
+    if new_category.strip() != "" and len(categories) < 5:  # Limit to 5 categories
+        categories.append(new_category.strip())
+    return categories, "", f"**Categories:**\n" + "\n".join([f"- {cat}" for cat in categories])
+
+# Function to reset categories
+def reset_categories():
+    return [], "**Categories:**\n- None"
+
+# Function to load data from uploaded CSV
+def load_data(file):
+    global df  # Ensure we're modifying the global DataFrame
+    if file is not None:
+        file.seek(0)  # Reset file pointer
+        if file.name.endswith('.csv'):
+            custom_df = pd.read_csv(file, encoding='utf-8')
+        else:
+            return "Error: Uploaded file is not a CSV."
+        # Check for required columns
+        required_columns = ['customer_comment']
+        if not all(col in custom_df.columns for col in required_columns):
+            return f"Error: Uploaded CSV must contain the following column: {', '.join(required_columns)}"
+        df = custom_df
+        return "Custom CSV loaded successfully!"
+    else:
+        return "No file uploaded."
+
+# Function to use template categories
+def use_template():
+    template_categories = ["Product Experience", "Customer Support", "Price of Service", "Other"]
+    return template_categories, f"**Categories:**\n" + "\n".join([f"- {cat}" for cat in template_categories])
+
 # Gradio Interface
 with gr.Blocks() as nps:
     # App title
@@ -78,29 +111,6 @@ with gr.Blocks() as nps:
     # Output
     output = gr.HTML()
 
-    # Function to load data from uploaded CSV
-    def load_data(file):
-        global df  # Ensure we're modifying the global DataFrame
-        if file is not None:
-            file.seek(0)  # Reset file pointer
-            if file.name.endswith('.csv'):
-                custom_df = pd.read_csv(file, encoding='utf-8')
-            else:
-                return "Error: Uploaded file is not a CSV."
-            # Check for required columns
-            required_columns = ['customer_comment']
-            if not all(col in custom_df.columns for col in required_columns):
-                return f"Error: Uploaded CSV must contain the following column: {', '.join(required_columns)}"
-            df = custom_df
-            return "Custom CSV loaded successfully!"
-        else:
-            return "No file uploaded."
-
-    # Function to use template categories
-    def use_template():
-        template_categories = ["Product Experience", "Customer Support", "Price of Service", "Other"]
-        return template_categories, f"**Categories:**\n" + "\n".join([f"- {cat}" for cat in template_categories])
-
     # Event handlers
     add_category_btn.click(
         fn=add_category,