Update app.py

app.py

@@ -36,7 +36,7 @@ def classify_comments(categories):
         sentiments.append(sentiment)
     df['comment_sentiment'] = sentiments
     df['comment_category'] = assigned_categories
-    return df[['customer_id', 'customer_comment', 'comment_sentiment', 'comment_category', 'customer_nps', 'customer_segment']].to_html(index=False)
+    return df.to_html(index=False)  # Return all fields with appended sentiment and category
 
 # Gradio Interface
 with gr.Blocks() as nps:
@@ -79,9 +79,9 @@ with gr.Blocks() as nps:
             else:
                 return "Error: Uploaded file is not a CSV."
             # Check for required columns
-            required_columns = ['customer_id', 'customer_comment', 'customer_nps', 'customer_segment']
+            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 columns: {', '.join(required_columns)}"
+                return f"Error: Uploaded CSV must contain the following column: {', '.join(required_columns)}"
             df = custom_df
             return "Custom CSV loaded successfully!"
         else: