Update app.py

app.py

@@ -98,7 +98,6 @@ with gr.Blocks() as nps:
             """)
             template_btn = gr.Button("Use Template", size="sm")
 
-    # Visual separator
     gr.Markdown("---")
 
     # Category section
@@ -111,7 +110,7 @@ with gr.Blocks() as nps:
                 reset_btn = gr.Button("Reset Categories", size="sm")
             # Category display
             category_status = gr.Markdown("**Categories:**\n- None")
-        with gr.Column(scale=2):
+        with gr.Column(scale=1):
             gr.Markdown("""
             **Instructions:**
             - Enter a category name and click **Add Category** to add it to the list.
@@ -119,13 +118,15 @@ with gr.Blocks() as nps:
             - The `customer_comment` field will be categorized based on the categories you provide.
             """)
 
+    gr.Markdown("---")
+
     # Classify button
+    with gr.Column(scale=1):
     classify_btn = gr.Button("Classify", size="sm")
 
-    # Output container with loading spinner
-    with gr.Row():
-        with gr.Column(scale=3):  # Center the container and make it 75% of the window width
-            output = gr.HTML()
+        with gr.Row():
+            with gr.Column(scale=3):  # Center the container and make it 75% of the window width
+                output = gr.HTML()
 
     # Event handlers
     add_category_btn.click(