Update app.py

app.py

@@ -1,5 +1,8 @@
 import gradio as gr
 import qrcode
+from PIL import Image, ImageDraw
+import io
+
 
 def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
     # Create QR code instance
@@ -9,14 +12,14 @@ def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
         box_size=10,
         border=4,
     )
-    
+
     # Add URL to the QR code
     qr.add_data(url)
     qr.make(fit=True)
 
     # 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")
@@ -25,7 +28,7 @@ def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
         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")
@@ -38,12 +41,13 @@ def generate_qr_code(url, gradient_color1, gradient_color2, logo, qr_color):
 
     return qr_byte_stream.read()
 
+
 inputs = [
-    gr.inputs.Textbox(label="URL to Site"),
-    gr.inputs.ColorPicker(label="Gradient Color 1 (Optional)"),
-    gr.inputs.ColorPicker(label="Gradient Color 2 (Optional)"),
+    gr.inputs.Textbox(lines=1, label="URL to Site"),
+    gr.inputs.Colorpicker(label="Gradient Color 1 (Optional)"),
+    gr.inputs.Colorpicker(label="Gradient Color 2 (Optional)"),
     gr.inputs.Image(label="Logo (Optional)"),
-    gr.inputs.ColorPicker(label="QR Code Color"),
+    gr.inputs.Color(label="QR Code Color"),
 ]
 
 outputs = gr.outputs.Image(type="bytes")