Update app.py

app.py

@@ -12,6 +12,7 @@ from nltk.tokenize import sent_tokenize
 from better_profanity import profanity
 import tempfile
 import os
+from PIL import Image  # Add PIL for image handling
 
 # Download NLTK data
 nltk.download('punkt', quiet=True)
@@ -112,7 +113,7 @@ def fetch_x_post(x_url, model_name, pos_words, neg_words, intensity, custom_mode
     sample_text = "Sample X post from " + x_url
     return analyze_sentiment(sample_text, model_name, pos_words, neg_words, intensity, custom_model_path, source="X post")
 
-# Generate timeline (return BytesIO directly)
+# Generate timeline (return PIL Image)
 def generate_timeline():
     if not sentiment_scores:
         return None
@@ -128,27 +129,21 @@ def generate_timeline():
     buf = io.BytesIO()
     plt.savefig(buf, format="png")
     buf.seek(0)
+    img = Image.open(buf)  # Convert BytesIO to PIL Image
     plt.close()
-    return buf  # Return the BytesIO object instead of base64 string
+    return img
 
-# Generate word cloud (return BytesIO directly)
+# Generate word cloud (return PIL Image directly)
 def generate_wordcloud(text):
     wordcloud = WordCloud(width=400, height=200, background_color="white").generate(text)
-    buf = io.BytesIO()
-    wordcloud.to_image().save(buf, format="PNG")
-    buf.seek(0)
-    return buf  # Return the BytesIO object instead of base64 string
+    return wordcloud.to_image()  # Directly return PIL Image
 
-# Generate QR code (return BytesIO directly)
+# Generate QR code (return PIL Image)
 def generate_qr(url):
     qr = qrcode.QRCode(version=1, box_size=10, border=4)
     qr.add_data(url)
     qr.make(fit=True)
-    img = qr.make_image(fill="black", back_color="white")
-    buf = io.BytesIO()
-    img.save(buf, format="PNG")
-    buf.seek(0)
-    return buf  # Return the BytesIO object instead of base64 string
+    return qr.make_image(fill="black", back_color="white")  # Directly return PIL Image
 
 # Export history with proper file handling
 def export_history():
@@ -230,6 +225,5 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as interface:
     theme_toggle.change(fn=toggle_theme, inputs=theme_toggle, outputs=theme_status)
     feedback_slider.change(fn=log_feedback, inputs=feedback_slider, outputs=feedback_output)
 
-
 # Launch the app
 interface.launch()