Update app.py

app.py

@@ -1,6 +1,7 @@
 import streamlit as st
 from PIL import Image
 from texture_transfer import create_image_tile, create_layover, create_image_cutout, paste_image
+from create_print_layover import create_image_with_feather_tile, control_texture
 
 st.title("Texture Correction App")
 
@@ -21,6 +22,13 @@ with product:
     st.header("Input product texture patch")
     product_image = st.file_uploader("Patch of the Garment(Cut out an area from the garment)",
                                      type=["png", "jpg", "jpeg"])
+    overlap_width = st.slider("Overlap Width", min_value=0, max_value=100, step=1, value=50)
+    what_overlap = st.selectbox(
+        'how you want to overlap?',
+        ('vertical', 'horizontal'),
+        index=None,
+        placeholder="please choose an option"
+    )
     if product_image:
         st.image(product_image)
 
@@ -39,7 +47,12 @@ with texture_correct:
             product_image_dim = product_image_n.size
             x_dim = product_image_dim[0]
             y_dim = product_image_dim[1]
-            create_image_tile(product_image, x_dim, y_dim)
+            if choice == "Direct Texture":
+                create_image_tile(product_image, x_dim, y_dim)
+            elif choice == "Feathered Texture":
+                create_image_with_feather_tile(product_image, x_dim, y_dim, overlap_width)
+            elif choice == "Direction Print":
+                control_texture(product_image, what_overlap, overlap_width, x_dim/2, y_dim/2)
             overlay = create_layover(generated_image, 'tiled_image.png', opacity)
             create_image_cutout('lay_over_image.png', gen_image_mask)
             paste_image(generated_image, 'cut_out_image.png', gen_image_mask)