Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import gradio as gr
 from transformers import pipeline
 from datetime import datetime
@@ -12,7 +13,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
+from PIL import Image  # For PIL image handling
 
 # Download NLTK data
 nltk.download('punkt', quiet=True)
@@ -129,21 +130,25 @@ def generate_timeline():
     buf = io.BytesIO()
     plt.savefig(buf, format="png")
     buf.seek(0)
-    img = Image.open(buf)  # Convert BytesIO to PIL Image
+    img = Image.open(buf)  # Standard PIL Image
     plt.close()
     return img
 
-# Generate word cloud (return PIL Image directly)
+# Generate word cloud (return PIL Image)
 def generate_wordcloud(text):
     wordcloud = WordCloud(width=400, height=200, background_color="white").generate(text)
-    return wordcloud.to_image()  # Directly return PIL Image
+    return wordcloud.to_image()  # Standard PIL Image
 
-# Generate QR code (return PIL Image)
+# Generate QR code (return standard PIL Image)
 def generate_qr(url):
     qr = qrcode.QRCode(version=1, box_size=10, border=4)
     qr.add_data(url)
     qr.make(fit=True)
-    return qr.make_image(fill="black", back_color="white")  # Directly return PIL Image
+    qr_img = qr.make_image(fill="black", back_color="white")  # Returns qrcode.image.pil.PilImage
+    buf = io.BytesIO()
+    qr_img.save(buf, format="PNG")
+    buf.seek(0)
+    return Image.open(buf)  # Convert to standard PIL Image
 
 # Export history with proper file handling
 def export_history():
@@ -226,4 +231,4 @@ with gr.Blocks(theme=gr.themes.Monochrome()) as interface:
     feedback_slider.change(fn=log_feedback, inputs=feedback_slider, outputs=feedback_output)
 
 # Launch the app
-interface.launch() 
+interface.launch()