Update app.py

app.py

@@ -43,7 +43,8 @@ def classify_comments(categories):
 def visualize_output():
     # Ensure the required columns exist
     if 'comment_sentiment' not in df.columns or 'comment_category' not in df.columns:
-        return None, None, None, "Error: Please classify comments before visualizing."
+        # Return 5 values (None for plots and an error message for markdown)
+        return None, None, None, "Error: Please classify comments before visualizing.", None
 
     # Pie Chart of Sentiment
     sentiment_counts = df['comment_sentiment'].value_counts()