Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import qrcode
 from PIL import Image
+import uuid
+import os
 
 def generate_qr(data):
     if not data:
@@ -11,15 +13,36 @@ def generate_qr(data):
     qr.make(fit=True)
     img = qr.make_image(fill_color="black", back_color="white")
 
-    # Convert from qrcode image (PilImage) to normal PIL.Image
+    img = img.convert("RGB")  # Ensure PIL format
+    return img
+
+def save_qr(data):
+    if not data:
+        raise gr.Error("Please enter text or URL!")
+
+    # Generate QR
+    qr = qrcode.QRCode(box_size=10, border=4)
+    qr.add_data(data)
+    qr.make(fit=True)
+    img = qr.make_image(fill_color="black", back_color="white")
     img = img.convert("RGB")
-    return img   # <-- RETURN PIL IMAGE (✔ Works in Gradio 5)
+
+    # Save with unique file name
+    filename = f"qr_{uuid.uuid4().hex}.png"
+    filepath = f"/tmp/{filename}"
+    img.save(filepath)
+
+    return filepath  # Gradio automatically downloads this file
 
 with gr.Blocks() as demo:
     gr.Markdown("## 📱 QR Code Generator")
 
     text = gr.Textbox(label="Enter Text / URL / Image URL")
-    generate = gr.Button("Generate QR Code")
+    
+    with gr.Row():
+        generate = gr.Button("Generate QR Code")
+        download = gr.Button("Download QR Code")
+
     output = gr.Image(label="QR Code")
 
     with gr.Accordion("Guideline", open=False):
@@ -28,8 +51,10 @@ with gr.Blocks() as demo:
         1. Enter Text or URL  
         2. Click **Generate QR Code**  
         3. Scan the QR code with your mobile  
+        4. Or click **Download QR Code** to save it  
         """)
 
     generate.click(fn=generate_qr, inputs=text, outputs=output)
+    download.click(fn=save_qr, inputs=text, outputs=gr.File(label="Download File"))
 
 demo.launch()