Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 from rembg import remove
-from PIL import Image as PILImage, ImageFilter
+from PIL import Image as PILImage, ImageFilter, Image
 from io import BytesIO
 import requests
 import cv2
 import numpy as np
+import webbrowser
 
 def remove_and_replace_background(subject, background, blur_radius, replace_background, use_color_picker, color):
     with open(subject, 'rb') as subject_img_file:
@@ -111,6 +112,27 @@ def rotate_image(img_input, degrees):
 def generate_iopaint_link():
     return "https://huggingface.co/spaces/Pontarids/IOPaint_Runner"
 
+def skew_image(image, horizontal_skew, vertical_skew):
+    image_np = np.array(image)
+    rows, cols, ch = image_np.shape
+
+    horizontal_factor = horizontal_skew / 100.0
+    vertical_factor = vertical_skew / 100.0
+
+    M_horizontal = np.float32([[1, horizontal_factor, 0],
+                               [0, 1, 0],
+                               [0, 0, 1]])
+
+    M_vertical = np.float32([[1, 0, 0],
+                             [vertical_factor, 1, 0],
+                             [0, 0, 1]])
+
+    skewed_image_horizontal = cv2.warpPerspective(image_np, M_horizontal, (cols, rows))
+    skewed_image_vertical = cv2.warpPerspective(skewed_image_horizontal, M_vertical, (cols, rows))
+
+    skewed_image = Image.fromarray(skewed_image_vertical)
+    return skewed_image
+
 with gr.Blocks() as demo:
 
     with gr.Tab("Remove and Replace Background"):
@@ -147,13 +169,21 @@ with gr.Blocks() as demo:
         img_button_contrast.click(adjust_brightness_and_darkness, inputs=[img_input_contrast, brightness_checkbox, brightness_slider, darkness_checkbox, darkness_slider], outputs=img_output_contrast)
     
     with gr.Tab("Rotate Image"):
-        temp_slider = gr.Slider(minimum=0, maximum=360, value=90, step=90, interactive=True, label="Slide me")
+        temp_slider = gr.Slider(minimum=0, maximum=360, interactive=True, label="Slide me")
         img_input_rotate = gr.Image()
         img_output_rotate = gr.Image()
         img_button_rotate = gr.Button("Submit")
         img_button_rotate.click(rotate_image, inputs=[img_input_rotate, temp_slider], outputs=img_output_rotate)
 
-    with gr.Tab("Object Remover"):
+    with gr.Tab("Skew Tool"):
+        image_input = gr.Image(type="pil", label="Upload an image")
+        horizontal_slider = gr.Slider(minimum=-100, maximum=100, value=0, label="Horizontal Skew")
+        vertical_slider = gr.Slider(minimum=-100, maximum=100, value=0, label="Vertical Skew")
+        output_image = gr.Image(type="pil", label="Skewed Image")
+        submit_button_skew = gr.Button("Submit")
+        submit_button_skew.click(skew_image, inputs=[image_input, horizontal_slider, vertical_slider], outputs=output_image)
+
+    with gr.Tab("Object Remover 👑"):
         link_output = gr.Markdown()
         link_button = gr.Button("Generate IOPaint Link")
         link_button.click(generate_iopaint_link, outputs=link_output)