Update app.py

app.py

@@ -40,33 +40,42 @@ def classify_comments(categories):
 
 # Gradio Interface
 with gr.Blocks() as nps:
-    # State to store categories
-    categories = gr.State([])
+    # App title
+    gr.Markdown("# Customer Comment Classifier")
 
-    # 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])
+    # Short explanation
+    gr.Markdown("""
+    This app classifies customer comments into categories and assigns sentiment labels (Positive/Negative). 
+    You can upload your own dataset or use the provided template. The app will append the generated 
+    `comment_sentiment` and `comment_category` fields to your dataset.
+    """)
 
-    # Function to reset categories
-    def reset_categories():
-        return [], "**Categories:**\n- None"
-
-    # UI for adding categories
+    # File upload and instructions
     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")
+        with gr.Column(scale=1):
+            uploaded_file = gr.File(label="Upload CSV", type="filepath", scale=1)
+        with gr.Column(scale=2):
+            gr.Markdown("""
+            **Instructions:**
+            - Upload a CSV file with at least one column: `customer_comment`.
+            - If you don't have your own data, click **Use Template** to load a sample dataset.
+            """)
 
-    # File upload and template buttons
-    uploaded_file = gr.File(label="Upload CSV", type="filepath")
+    # Template button
     template_btn = gr.Button("Use Template")
-    gr.Markdown("# NPS Comment Categorization")
+
+    # Category widgets
+    with gr.Row():
+        with gr.Column():
+            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")
 
     # Classify button
-    classify_btn = gr.Button("Classify Comments")
+    classify_btn = gr.Button("Classify", size="sm")
+
+    # Output
     output = gr.HTML()
 
     # Function to load data from uploaded CSV