Update app.py

app.py

@@ -4,7 +4,7 @@ from PIL import Image, ImageDraw
 import io
 
 
-def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
+def generate_qr_code(url, logo, qr_color):
     # Create QR code instance
     qr = qrcode.QRCode(
         version=1,
@@ -20,34 +20,18 @@ def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
     # Create the QR code image
     qr_image = qr.make_image(fill_color=qr_color, back_color="#ffffff")  # Set QR code color here
 
-    # Add gradient if colors are provided
-    if gradient_color1 and gradient_color2:
-        qr_image = qr_image.convert("RGB")
-        qr_image = qr_image.resize((500, 500))
-
-        gradient = Image.new("RGB", qr_image.size)
-        grad = ImageDraw.Gradient(grad.size, "LR", gradient_color1, gradient_color2)
-        qr_image = Image.blend(qr_image, gradient, 0.5)
-
     # Add logo if provided
     if logo:
         logo_image = Image.open(logo).convert("RGBA")
         qr_image.paste(logo_image, (125, 125), logo_image)
 
-    # Save the QR code to a byte stream
-    qr_byte_stream = io.BytesIO()
-    qr_image.save(qr_byte_stream, format="PNG")
-    qr_byte_stream.seek(0)
-
-    return qr_byte_stream.read()
+    return qr_image
 
 
 inputs = [
-    gr.inputs.Textbox(lines=1, label="URL to Site"),
-    gr.ColorPicker(label="Gradient Color 1 (Optional)"),
-    gr.ColorPicker(label="Gradient Color 2 (Optional)"),
+    gr.inputs.Textbox(label="URL to Site"),
     gr.inputs.Image(label="Logo (Optional)"),
-    gr.ColorPicker(label="QR Code Color"),
+    gr.inputs.ColorPicker(label="QR Code Color"),
 ]
 
 outputs = gr.outputs.Image(type="pil")
@@ -59,4 +43,4 @@ gr.Interface(
     title="QR Code Generator",
     description="Generate QR codes with customizable options",
     theme="huggingface",
-).launch()
+).launch(share=True)