Update app.py

app.py

@@ -1,13 +1,12 @@
 import gradio as gr
 import qrcode
 from PIL import Image
-import io
 
 def generate_qr(link):
     if link.strip() == "":
         return None
 
-    # Create QR
+    # Create QR code object
     qr = qrcode.QRCode(
         version=1,
         box_size=10,
@@ -16,21 +15,19 @@ def generate_qr(link):
     qr.add_data(link)
     qr.make(fit=True)
 
+    # Make the image (this is already a PIL image)
     img = qr.make_image(fill_color="black", back_color="white")
 
-    # Convert to displayable image
-    buf = io.BytesIO()
-    img.save(buf, format="PNG")
-    buf.seek(0)
+    # Ensure it's in RGB mode for Gradio
+    img = img.convert("RGB")
 
-    return buf
+    return img
 
-# Gradio UI
 with gr.Blocks(title="Simple QR Code Maker") as demo:
     gr.Markdown("### 🔳 Simple QR Code Maker\nEnter any link to generate a QR code instantly!")
 
     link_input = gr.Textbox(label="Enter URL", placeholder="https://example.com")
-    output_image = gr.Image(label="Generated QR Code")
+    output_image = gr.Image(label="Generated QR Code", type="pil")
 
     generate_btn = gr.Button("Generate QR Code")
     generate_btn.click(fn=generate_qr, inputs=link_input, outputs=output_image)