Update app.py

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 import qrcode
 import qrcode.constants
 from PIL import Image
-import io
 
 def get_qr_image(data: str, fill_color: str = None, back_color: str = None):
     """
@@ -51,18 +50,11 @@ def generate_qr(url, fill_color=None, back_color=None):
             raise ValueError("URL or data cannot be empty")
         
         # Convert empty strings to None
-        fill_color = fill_color.strip() if fill_color else 'black'
-        back_color = back_color.strip() if back_color else 'white'
+        fill_color = fill_color.strip() if fill_color else None
+        back_color = back_color.strip() if back_color else None
         
         # Generate QR code with optional color parameters
-        img = get_qr_image(url, fill_color, back_color)
-        
-        # Save to a bytes buffer in PNG format
-        buffer = io.BytesIO()
-        img.save(buffer, format='PNG')
-        buffer.seek(0)
-        
-        return buffer
+        return get_qr_image(url, fill_color, back_color)
     
     except Exception as e:
         # Error handling for Gradio
@@ -77,10 +69,9 @@ iface = gr.Interface(
         gr.Textbox(label="Fill Color (optional)", placeholder="e.g., 'blue', '#FF0000'"),
         gr.Textbox(label="Background Color (optional)", placeholder="e.g., 'white', '#FFFFFF'")
     ],
-    outputs=gr.Image(label="QR Code", type="numpy"),
+    outputs=gr.Image(label="QR Code"),
     title="QR Code Generator",
     description="Generate a QR code from a URL or text with optional color customization.",
-    # Optional: Specify file type
     examples=[
         ["https://example.com"],
         ["Hello World", "blue", "white"]
@@ -96,7 +87,6 @@ if __name__ == "__main__":
 
 
 
-
 # import gradio as gr
 # import qrcode
 # import qrcode.constants