Update app.py

app.py

@@ -3,84 +3,65 @@ from PIL import Image
 import base64
 import gradio as gr
 
-# Function to process text and generate QR code
-def txt2qrcode(text: str):
-    # Encode the text as URL-safe base64
-    encoded_text = base64.b64encode(text.encode('utf-8')).decode('utf-8')  # URL encoding the input text
+# Function to encode text or file content in base64 and generate a QR code
+def generate_qr(data: str, error_correction: int = qrcode.constants.ERROR_CORRECT_L, box_size: int = 10, border: int = 4, fill_color: str = "black", back_color: str = "white", logo_path: str = None):
+    encoded_data = base64.b64encode(data.encode('utf-8')).decode('utf-8')
+    final_url = f"https://flowly-ai.vercel.app/tools/qr/qr-code/read/{encoded_data}"
 
-    # Construct the final URL to embed into the QR code
-    final_url = f"https://flowly-ai.vercel.app/tools/qr/pro-qr-codes/read/{encoded_text}"
-
-    # Create the QR code
     qr = qrcode.QRCode(
         version=1,
-        error_correction=qrcode.constants.ERROR_CORRECT_L,
-        box_size=10,
-        border=4,
+        error_correction=error_correction,
+        box_size=box_size,
+        border=border,
     )
     qr.add_data(final_url)
     qr.make(fit=True)
+    img_qr = qr.make_image(fill_color=fill_color, back_color=back_color)
 
-    # Create an image for the QR code
-    img_qr = qr.make_image(fill='black', back_color='white')
-
-    # Optionally, add a logo to the center of the QR code
-    logo = Image.open("logo.png")  # Replace with your logo path
-    logo = logo.convert("RGBA")
-    
-    qr_width, qr_height = img_qr.size
-    logo_width, logo_height = logo.size
-    
-    # Resize logo to fit in the QR code
-    logo = logo.resize((qr_width // 5, qr_height // 5))  # Resize to be 20% of the QR code size
-
-    # Calculate position to place logo at the center
-    logo_position = ((qr_width - logo_width) // 2, (qr_height - logo_height) // 2)
+    if logo_path:
+        try:
+            logo = Image.open(logo_path).convert("RGBA")
+            qr_width, qr_height = img_qr.size
+            logo = logo.resize((qr_width // 5, qr_height // 5))
+            logo_position = ((qr_width - logo.size[0]) // 2, (qr_height - logo.size[1]) // 2)
+            img_qr.paste(logo, logo_position, logo)
+        except Exception as e:
+            print(f"Error loading logo: {e}")
 
-    # Paste the logo onto the QR code image
-    img_qr.paste(logo, logo_position, logo)
-    
-    # Save the QR code image
-    output_path = "qr_code_with_logo.png"
+    output_path = "qr_code.png"
     img_qr.save(output_path)
-
     return output_path
 
-# Function to process a file (e.g., text file) and generate QR code
-def file2qrcode(file: str):
-    # Read the content of the file
-    with open(file, 'r') as f:
-        text = f.read()
+# Function to handle text or file input
+def qr_code_interface(input_type, text, file, error_correction, box_size, border, fill_color, back_color, logo):
+    logo_path = logo.name if logo else None
     
-    # Generate QR code with the file's text content
-    return txt2qrcode(text)
-
-# Gradio Interface for the QR code generation
-def qr_code_interface(text: str):
-    result = txt2qrcode(text)
-    return result
-
-def qr_code_file_interface(file: str):
-    result = file2qrcode(file)
-    return result
+    if input_type == "Text" and text:
+        return generate_qr(text, error_correction, box_size, border, fill_color, back_color, logo_path)
+    elif input_type == "File" and file:
+        with open(file.name, 'r') as f:
+            file_data = f.read()
+        return generate_qr(file_data, error_correction, box_size, border, fill_color, back_color, logo_path)
+    else:
+        return None
 
-# Create a Gradio interface
+# Gradio Interface
 interface = gr.Interface(
     fn=qr_code_interface,
-    inputs=gr.Textbox(label="Enter Text to Encode", placeholder="Enter your text here..."),
+    inputs=[
+        gr.Radio(["Text", "File"], label="Input Type", value="Text"),
+        gr.Textbox(label="Enter Text", visible=True),
+        gr.File(label="Upload Text File", type="file", visible=False),
+        gr.Slider(0, 3, 1, label="Error Correction", value=0),
+        gr.Slider(1, 20, 1, label="Box Size", value=10),
+        gr.Slider(1, 10, 1, label="Border Size", value=4),
+        gr.ColorPicker(label="QR Fill Color", value="#000000"),
+        gr.ColorPicker(label="QR Background Color", value="#FFFFFF"),
+        gr.File(label="Upload Logo (Optional)")
+    ],
     outputs=gr.Image(type="filepath", label="Generated QR Code"),
-    description="Generate a QR Code from your text. The input text will be URL encoded and embedded into a QR code pointing to a specific URL.",
-    css="footer {visibility: hidden}"
-)
-
-# Create a second Gradio interface for file input
-file_interface = gr.Interface(
-    fn=qr_code_file_interface,
-    inputs=gr.File(label="Upload Text File"),
-    outputs=gr.Image(type="filepath", label="Generated QR Code from File"),
-    description="Upload a text file, and generate a QR code based on the file's content."
+    description="Generate a QR Code from either text or file with custom styles, colors, and an optional logo."
 )
 
-# Launch the Gradio apps
+# Launch the app
 interface.launch(share=True)
-file_interface.launch(share=True)