Update eda.py

eda.py

@@ -4,11 +4,21 @@ import matplotlib.pyplot as plt
 from wordcloud import WordCloud
 
 def display_eda(data):
+    # Check if 'sentiment' column exists in the dataset
+    if 'sentiment' not in data.columns:
+        st.error("The dataset does not contain a 'sentiment' column. Please check the data source.")
+        return
+
     # Distribution of sentiments
     st.subheader("Distribution of Sentiments")
     sentiment_counts = data['sentiment'].value_counts()
-    st.bar_chart(sentiment_counts)
-    
+    fig, ax = plt.subplots()
+    sentiment_counts.plot(kind='bar', ax=ax)
+    ax.set_title('Distribution of Sentiments')
+    ax.set_xlabel('Sentiment')
+    ax.set_ylabel('Count')
+    st.pyplot(fig)
+
     # Word cloud for each sentiment
     st.subheader("Word Clouds for Sentiments")
     sentiments = data['sentiment'].unique()