Update app.py

app.py

@@ -34,6 +34,26 @@ def get_sentiment_label(polarity, pos_thresh, neg_thresh):
     else:
         return "Neutral"
 
+# Generate word cloud
+def generate_wordcloud(text_series, title):
+    text = " ".join(text_series.dropna())
+    if not text.strip():
+        fig = plt.figure(figsize=(6, 4))
+        plt.text(0.5, 0.5, f"No data for {title}", fontsize=14, ha='center', va='center')
+        plt.axis("off")
+        plt.title(title)
+        plt.tight_layout()
+        return fig
+
+    wc = WordCloud(width=600, height=400, background_color="white", colormap="tab10").generate(text)
+    fig = plt.figure(figsize=(6, 4))
+    plt.imshow(wc, interpolation="bilinear")
+    plt.axis("off")
+    plt.title(title)
+    plt.tight_layout()
+    return fig
+
+
 # Main processing function
 def analyze_sentiment(file, text_column, pos_thresh, neg_thresh):
     try: