Update eda.py

eda.py

@@ -3,11 +3,25 @@ import pandas as pd
 import matplotlib.pyplot as plt
 from wordcloud import WordCloud
 
+def label_sentiment(rating):
+    """Label sentiment based on the rating."""
+    if rating in [1, 2]:
+        return 'negative'
+    elif rating == 3:
+        return 'neutral'
+    elif rating in [4, 5]:
+        return 'positive'
+    else:
+        return 'unknown'
+
 def display_eda(data):
-    # Check if 'sentiment' column exists in the dataset
+    # Derive the 'sentiment' column from 'rating' if it doesn't exist
     if 'sentiment' not in data.columns:
-        st.error("The dataset does not contain a 'sentiment' column. Please check the data source.")
-        return
+        if 'rating' not in data.columns:
+            st.error("The dataset does not contain a 'rating' or 'sentiment' column. Please check the data source.")
+            return
+        else:
+            data['sentiment'] = data['rating'].apply(label_sentiment)
 
     # Distribution of sentiments
     st.subheader("Distribution of Sentiments")