Update app.py

app.py

@@ -6,7 +6,7 @@ import gradio as gr
 
 sb_logo = Image.open("./SB_logo_horizontal.png")
 
-def freemium_watermark(img, sb_logo):
+def freemium_watermark(img, sb_logo, alpha):
     width_px, height_px = img.size
     # Convert the logo to "RGBA" mode and resize it to 80% of the image width
     sb_logo = sb_logo.convert("RGBA")
@@ -16,7 +16,7 @@ def freemium_watermark(img, sb_logo):
     sb_logo = sb_logo.resize((new_logo_width, new_logo_height))
     # Make the logo semi-translucent
     transparent_img = Image.new('RGBA', (sb_logo.width, sb_logo.height), (0, 0, 0, 0))  # Create a transparent image of the same size
-    sb_logo = Image.blend(transparent_img, sb_logo, alpha=0.3)  # Blend the logo with the transparent image
+    sb_logo = Image.blend(transparent_img, sb_logo, alpha=alpha)  # Blend the logo with the transparent image
     # Calculate the position to paste the logo at the center of the image
     paste_x_position = (width_px - new_logo_width) // 2
     paste_y_position = int(height_px * 0.4) - (new_logo_height // 2)
@@ -37,7 +37,8 @@ def lighter(input_path):
     dilated_edges = cv2.dilate(edges, structuring_element, iterations=2)
     line_drawing = 255 - dilated_edges
     line_img = Image.fromarray(line_drawing)
-    return freemium_watermark(line_img, sb_logo)
+    alpha = 0.3
+    return freemium_watermark(line_img, sb_logo, alpha)
     
 def darker(input_path):
     print('darker')
@@ -54,7 +55,8 @@ def darker(input_path):
     dilated_edges = cv2.dilate(edges, structuring_element, iterations=2)
     line_drawing = 255 - dilated_edges
     line_img = Image.fromarray(line_drawing)
-    return freemium_watermark(line_img, sb_logo)
+    alpha = 0.5
+    return freemium_watermark(line_img, sb_logo, alpha)
 
 with gr.Blocks(analytics_enabled=False, theme=gr.themes.Soft()) as demo:
     with gr.Row():
@@ -67,6 +69,7 @@ with gr.Blocks(analytics_enabled=False, theme=gr.themes.Soft()) as demo:
             with gr.Row():
                 lighter_btn=gr.Button("Lighter")
                 darker_btn=gr.Button("Darker")
+                
             gr.Markdown("<p style='text-align: center; font-size: 20px;'>Want to remove the watermark?</p>\n")
             gr.Markdown("<p style='text-align: center; font-size: 20px;'>Subscribe or <a href='https://skyebrowse.com/pricing'>purchase a model</a> starting at $3.<p>")
             lighter_btn.click(lighter, inputs=[input], outputs=[output], show_progress="minimal")
@@ -84,6 +87,7 @@ with gr.Blocks(analytics_enabled=False, theme=gr.themes.Soft()) as demo:
         dilated_edges = cv2.dilate(edges, structuring_element, iterations=2)
         line_drawing = 255 - dilated_edges
         line_img = Image.fromarray(line_drawing)
-        return freemium_watermark(line_img, sb_logo)
+        alpha = 0.3
+        return freemium_watermark(line_img, sb_logo, alpha)
 
 demo.queue(default_concurrency_limit=10, api_open=False).launch(show_api=False)