Anustup commited on
Commit
505d5eb
·
verified ·
1 Parent(s): f40fc3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
  from PIL import Image
 
3
  from texture_transfer import create_image_tile, create_layover, create_image_cutout, paste_image
4
  from create_print_layover import create_image_with_feather_tile, control_texture
5
 
@@ -58,7 +59,9 @@ with texture_correct:
58
  elif choice == "Feathered Texture":
59
  create_image_with_feather_tile(product_image, x_dim, y_dim, overlap_width)
60
  elif choice == "Direction Print":
61
- control_texture(product_image, what_overlap, overlap_width, x_dim/2, y_dim/2)
 
 
62
  overlay = create_layover(generated_image, 'tiled_image.png', opacity)
63
  create_image_cutout('lay_over_image.png', gen_image_mask)
64
  paste_image(generated_image, 'cut_out_image.png', gen_image_mask)
 
1
  import streamlit as st
2
  from PIL import Image
3
+ import cv2
4
  from texture_transfer import create_image_tile, create_layover, create_image_cutout, paste_image
5
  from create_print_layover import create_image_with_feather_tile, control_texture
6
 
 
59
  elif choice == "Feathered Texture":
60
  create_image_with_feather_tile(product_image, x_dim, y_dim, overlap_width)
61
  elif choice == "Direction Print":
62
+ control_texture(product_image, what_overlap, overlap_width, x_dim, y_dim)
63
+ resized_image = cv2.resize('tiled_image_2.png', (x_dim, y_dim))
64
+ cv2.imwrite('tiled_image.png', resized_image)
65
  overlay = create_layover(generated_image, 'tiled_image.png', opacity)
66
  create_image_cutout('lay_over_image.png', gen_image_mask)
67
  paste_image(generated_image, 'cut_out_image.png', gen_image_mask)