Update app.py

app.py

@@ -2,6 +2,7 @@ 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):
     """
@@ -54,7 +55,14 @@ def generate_qr(url, fill_color=None, back_color=None):
         back_color = back_color.strip() if back_color else 'white'
         
         # Generate QR code with optional color parameters
-        return get_qr_image(url, fill_color, back_color)
+        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
     
     except Exception as e:
         # Error handling for Gradio
@@ -69,11 +77,102 @@ 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"),
+    outputs=gr.Image(label="QR Code", type="file"),
     title="QR Code Generator",
-    description="Generate a QR code from a URL or text with optional color customization."
+    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"]
+    ]
 )
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()
+
+
+
+
+
+
+
+# import gradio as gr
+# import qrcode
+# import qrcode.constants
+# from PIL import Image
+
+# def get_qr_image(data: str, fill_color: str = None, back_color: str = None):
+#     """
+#     Generate a QR code image from the given data and optional color parameters.
+    
+#     Args:
+#         data (str): The data to encode in the QR code
+#         fill_color (str, optional): Color of the QR code modules (default: black)
+#         back_color (str, optional): Background color of the QR code (default: white)
+    
+#     Returns:
+#         PIL.Image: Generated QR code image
+#     """
+#     qr = qrcode.QRCode(
+#         version=1,
+#         error_correction=qrcode.constants.ERROR_CORRECT_H,
+#         box_size=10,
+#         border=4,
+#     )
+    
+#     qr.add_data(data)
+#     qr.make(fit=True)
+    
+#     # Explicitly create a PIL Image and convert to RGB mode
+#     img = qr.make_image(fill_color=fill_color, back_color=back_color)
+#     img = img.convert('RGB')
+    
+#     return img
+
+# def generate_qr(url, fill_color=None, back_color=None):
+#     """
+#     Wrapper function for Gradio interface to generate QR code.
+#     Handles potential errors and provides default behavior.
+    
+#     Args:
+#         url (str): URL or data to encode
+#         fill_color (str, optional): Color of QR code modules
+#         back_color (str, optional): Background color of QR code
+    
+#     Returns:
+#         PIL.Image: Generated QR code image
+#     """
+#     try:
+#         # Validate URL/data is not empty
+#         if not url:
+#             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'
+        
+#         # Generate QR code with optional color parameters
+#         return get_qr_image(url, fill_color, back_color)
+    
+#     except Exception as e:
+#         # Error handling for Gradio
+#         gr.Warning(f"Error generating QR code: {str(e)}")
+#         return None
+
+# # Create Gradio interface
+# iface = gr.Interface(
+#     fn=generate_qr,
+#     inputs=[
+#         gr.Textbox(label="URL or Data"),
+#         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"),
+#     title="QR Code Generator",
+#     description="Generate a QR code from a URL or text with optional color customization."
+# )
+
+# # Launch the app
+# if __name__ == "__main__":
+#     iface.launch()