Update app.py

app.py

@@ -12,7 +12,7 @@ import spaces
 import numpy as np
 import cv2
 from pyzxing import BarCodeReader
-from PIL import ImageOps
+from PIL import ImageOps, ImageEnhance, ImageFilter
 from huggingface_hub import hf_hub_download, snapshot_download
 from PIL import ImageEnhance
 
@@ -338,6 +338,42 @@ def invert_init_image_display(image):
         return Image.fromarray(inverted)
     return inverted
 
+def adjust_color_balance(image, r, g, b):
+    # Convert image to RGB if it's not already
+    image = image.convert('RGB')
+    
+    # Split the image into its RGB channels
+    r_channel, g_channel, b_channel = image.split()
+    
+    # Adjust each channel
+    r_channel = r_channel.point(lambda i: i + (i * r))
+    g_channel = g_channel.point(lambda i: i + (i * g))
+    b_channel = b_channel.point(lambda i: i + (i * b))
+    
+    # Merge the channels back
+    return Image.merge('RGB', (r_channel, g_channel, b_channel))
+
+def apply_qr_overlay(image, original_qr, overlay, opacity):
+    if not overlay or original_qr is None:
+        return image
+    
+    # Resize original QR to match the generated image
+    original_qr = original_qr.resize(image.size)
+    
+    # Create a new image blending the generated image and the QR code
+    return Image.blend(image, original_qr, opacity)
+
+def apply_edge_enhancement(image, strength):
+    if strength == 0:
+        return image
+    
+    # Apply edge enhancement
+    enhanced = image.filter(ImageFilter.EDGE_ENHANCE)
+    
+    # Blend the original and enhanced images based on strength
+    return Image.blend(image, enhanced, strength / 5.0)
+
+
 css = """
 h1, h2, h3, h4, h5, h6, p, li, ul, ol, a, .centered-image {
     text-align: center;
@@ -389,12 +425,13 @@ with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True
         )
         
         # Add login form
-        with gr.Column():
-                username = gr.Textbox(label="Username", placeholder="Enter your username", value="UDG")
-                password = gr.Textbox(label="Password", type="password", placeholder="Enter your password", value="UDG!")
-        with gr.Column():
-                login_button = gr.Button("Login")
-                login_error = gr.Markdown("Invalid username or password. Please try again.", visible=False)
+        with gr.Row():
+            with gr.Column():
+                    username = gr.Textbox(label="Username", placeholder="Enter your username", value="UDG")
+                    password = gr.Textbox(label="Password", type="password", placeholder="Enter your password", value="UDG!")
+            with gr.Column():
+                    login_button = gr.Button("Login")
+                    login_error = gr.Markdown("Invalid username or password. Please try again.", visible=False)
 
 
     with gr.Tab("QR Code and Prompt Input", visible=False) as app_container:
@@ -626,6 +663,17 @@ with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True
             image_selector = gr.Dropdown(label="Select Image to Edit", choices=[], interactive=True, visible=False)
             image_to_edit = gr.Image(label="Your Artistic QR Code", show_download_button=True, show_fullscreen_button=True)
 
+            with gr.Row():
+                qr_overlay = gr.Checkbox(label="Overlay Original QR Code", value=False)
+                qr_opacity = gr.Slider(minimum=0.1, maximum=1.0, step=0.1, value=0.5, label="QR Overlay Opacity")
+                edge_enhance = gr.Slider(minimum=0.0, maximum=5.0, step=0.1, value=0.0, label="Edge Enhancement")
+
+            with gr.Row():
+                red_balance = gr.Slider(minimum=-1.0, maximum=1.0, step=0.1, value=0.0, label="Red Balance")
+                green_balance = gr.Slider(minimum=-1.0, maximum=1.0, step=0.1, value=0.0, label="Green Balance")
+                blue_balance = gr.Slider(minimum=-1.0, maximum=1.0, step=0.1, value=0.0, label="Blue Balance")
+
+
             with gr.Row():
                 brightness = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=1.0, label="Brightness")
                 contrast = gr.Slider(minimum=0.1, maximum=2.0, step=0.1, value=1.0, label="Contrast")
@@ -635,8 +683,8 @@ with gr.Blocks(theme='Hev832/Applio', css=css, fill_width=True, fill_height=True
 
             with gr.Row():
                 edited_image = gr.Image(label="Edited QR Code", show_download_button=True, show_fullscreen_button=True, visible=False)
-                scan_button = gr.Button("Verify QR Code Works", size="sm")
-                scan_result = gr.Textbox(label="Validation Result of QR Code", interactive=False)
+                scan_button = gr.Button("Verify QR Code Works", size="sm", visible=False)
+                scan_result = gr.Textbox(label="Validation Result of QR Code", interactive=False, visible=False)
                 
             used_seed = gr.Number(label="Seed Used", interactive=False)