Update app.py

app.py

@@ -3,7 +3,9 @@ import numpy as np
 from PIL import Image
 Image.MAX_IMAGE_PIXELS = None
 import gradio as gr
-        
+
+sb_logo = Image.open("./SB_logo_horizontal.png")
+
 def freemium_watermark(img, sb_logo):
     width_px, height_px = img.size
     # Convert the logo to "RGBA" mode and resize it to 80% of the image width
@@ -22,8 +24,38 @@ def freemium_watermark(img, sb_logo):
     # Save the image
     return img
 
+def lighter(input_path):
+    print('lighter')
+    if input_path is None:
+        return None
+    img_array = np.array(input_path)
+    # Line processing
+    blurred = cv2.GaussianBlur(img_array, (7, 7), 0)
+    #lighter
+    edges = cv2.Canny(blurred, 100, 180)
+    structuring_element = cv2.getStructuringElement(cv2.MORPH_RECT, (2,2))
+    dilated_edges = cv2.dilate(edges, structuring_element, iterations=1)
+    line_drawing = 255 - dilated_edges
+    line_img = Image.fromarray(line_drawing)
+    return freemium_watermark(line_img, sb_logo)
     
-sb_logo = Image.open("./SB_logo_horizontal.png")
+def darker(input_path):
+    print('darker')
+    if input_path is None:
+        return None
+    img_array = np.array(input_path)
+    # Line processing
+    blurred = cv2.GaussianBlur(img_array, (5, 5), 0)
+    #darker
+    edges = cv2.Canny(blurred, 60, 90) #darker
+    #lighter
+    # edges = cv2.Canny(blurred, 100, 180)
+    structuring_element = cv2.getStructuringElement(cv2.MORPH_RECT, (2,2))
+    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)
+
 with gr.Blocks(analytics_enabled=False, theme=gr.themes.Soft()) as demo:
     with gr.Row():
         with gr.Column():
@@ -32,21 +64,26 @@ with gr.Blocks(analytics_enabled=False, theme=gr.themes.Soft()) as demo:
             gr.Examples(examples=["hotel.jpg", "road_small.jpg", "crash_small.jpg", "trailer.jpg"], inputs=input)
         with gr.Column():
             output = gr.Image(type = "filepath", label="Sketch Drawing", show_share_button=False)
+            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")
+            darker_btn.click(darker, inputs=[input], outputs=[output], show_progress="minimal")
     
     @gr.on(inputs=[input], outputs=[output], show_progress="minimal")
-    def sketch(input_path):   
+    def sketch(input_path):
         if input_path is None:
             return None
         img_array = np.array(input_path)
         # Line processing
         blurred = cv2.GaussianBlur(img_array, (7, 7), 0)
         edges = cv2.Canny(blurred, 100, 180)
-        structuring_element = cv2.getStructuringElement(cv2.MORPH_RECT, (4,3))
-        dilated_edges = cv2.dilate(edges, structuring_element, iterations=1)
+        structuring_element = cv2.getStructuringElement(cv2.MORPH_RECT, (2,2))
+        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)
 
-demo.queue(default_concurrency_limit=10).launch()
+demo.queue(default_concurrency_limit=10, api_open=False).launch(show_api=False)