Update app.py

app.py

@@ -112,7 +112,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
+# Generate timeline (return BytesIO directly)
 def generate_timeline():
     if not sentiment_scores:
         return None
@@ -128,20 +128,18 @@ def generate_timeline():
     buf = io.BytesIO()
     plt.savefig(buf, format="png")
     buf.seek(0)
-    img_str = "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode()
     plt.close()
-    return img_str
+    return buf  # Return the BytesIO object instead of base64 string
 
-# Generate word cloud
+# Generate word cloud (return BytesIO 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)
-    img_str = "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode()
-    return img_str
+    return buf  # Return the BytesIO object instead of base64 string
 
-# Generate QR code
+# Generate QR code (return BytesIO directly)
 def generate_qr(url):
     qr = qrcode.QRCode(version=1, box_size=10, border=4)
     qr.add_data(url)
@@ -150,8 +148,7 @@ def generate_qr(url):
     buf = io.BytesIO()
     img.save(buf, format="PNG")
     buf.seek(0)
-    img_str = "data:image/png;base64," + base64.b64encode(buf.getvalue()).decode()
-    return img_str
+    return buf  # Return the BytesIO object instead of base64 string
 
 # Export history with proper file handling
 def export_history():
@@ -173,7 +170,7 @@ def toggle_theme(light_mode):
 
 # Gradio interface
 with gr.Blocks(theme=gr.themes.Monochrome()) as interface:
-    gr.Markdown("# Sentiment Analysis App")
+    gr.Markdown("# Sentify")
     gr.Markdown("Next-level sentiment analysis with context, comparison, and more!")
     
     with gr.Row():