Update app.py

app.py

@@ -176,7 +176,32 @@ def fetch_youtube_data(keyword, limit=30):
         return pd.DataFrame()
 
 # --------------------------
-# Prediction Functions (Rewritten to Fix Error)
+# Visualization Functions
+# --------------------------
+
+def generate_wordcloud(text):
+    """Fast word cloud generation"""
+    try:
+        if not text.strip():
+            return ""
+        
+        wordcloud = WordCloud(
+            width=800, 
+            height=400, 
+            background_color='white',
+            collocations=False,  # Faster processing
+            stopwords=nltk.corpus.stopwords.words('english')
+        ).generate(text)
+        
+        img = BytesIO()
+        wordcloud.to_image().save(img, format='PNG')
+        return base64.b64encode(img.getvalue()).decode()
+    except Exception as e:
+        st.error(f"Word cloud generation error: {str(e)}")
+        return ""
+
+# --------------------------
+# Prediction Functions
 # --------------------------
 
 def prepare_data_for_prediction(data):
@@ -410,7 +435,10 @@ def main():
                 
                 st.subheader("📊 Text Visualization")
                 wordcloud_img = f'data:image/png;base64,{generate_wordcloud(user_input)}'
-                st.image(wordcloud_img, use_column_width=True)
+                if wordcloud_img:
+                    st.image(wordcloud_img, use_column_width=True)
+                else:
+                    st.info("No word cloud generated due to insufficient text")
                 
         else:  # Live Data Analysis
             if not keyword.strip():
@@ -464,7 +492,10 @@ def main():
                 st.subheader("📊 Content Visualization")
                 all_text = " ".join(combined_data['text'])
                 wordcloud_img = f'data:image/png;base64,{generate_wordcloud(all_text)}'
-                st.image(wordcloud_img, use_column_width=True)
+                if wordcloud_img:
+                    st.image(wordcloud_img, use_column_width=True)
+                else:
+                    st.info("No word cloud generated due to insufficient text")
                 
                 # Filter recent data
                 combined_data['date'] = pd.to_datetime(combined_data['date'])