mallelapreethi commited on
Commit
f66d1ee
·
verified ·
1 Parent(s): 6c7dde0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +32 -26
app.py CHANGED
@@ -46,8 +46,8 @@ st.markdown("""
46
  Upload an image, and get the sketch version instantly!
47
  """)
48
 
49
- # Example conversions and user input side by side
50
- col1, col2 = st.columns(2)
51
 
52
  # Replace 'Dog.jpg' with the path to an example image if running locally
53
  example_image_path = 'Dog.jpg'
@@ -58,40 +58,46 @@ if example_image is not None:
58
  example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
59
  example_sketch = image_to_sketch(example_image)
60
 
 
 
61
  with col1:
62
- st.subheader("Example Conversions")
63
  st.image(example_image_rgb, caption='Original Image', use_column_width=True)
 
64
  st.image(example_sketch, caption='Sketch Image', use_column_width=True)
65
  else:
66
  st.error(f"Failed to load example image from path: {example_image_path}")
67
 
68
- with col2:
69
- st.subheader("Upload Your Image")
70
- uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
71
 
72
- if uploaded_file is not None:
73
- image = np.array(Image.open(uploaded_file))
74
- st.image(image, caption='Uploaded Image', use_column_width=True)
75
-
76
- st.write("")
77
- st.write("Converting...")
 
 
 
78
 
79
- sketch = image_to_sketch(image)
80
-
 
81
  st.image(sketch, caption='Sketch', use_column_width=True)
82
 
83
- # Provide a download link for the sketch image
84
- im_pil = Image.fromarray(sketch)
85
- im_pil.save("sketch.png")
86
- with open("sketch.png", "rb") as file:
87
- btn = st.download_button(
88
- label="Download Sketch",
89
- data=file,
90
- file_name="sketch.png",
91
- mime="image/png"
92
- )
93
- else:
94
- st.info("Please upload an image to convert.")
95
 
96
  # Footer
97
  st.markdown("""
 
46
  Upload an image, and get the sketch version instantly!
47
  """)
48
 
49
+ # Example conversions
50
+ st.subheader("Example Conversions")
51
 
52
  # Replace 'Dog.jpg' with the path to an example image if running locally
53
  example_image_path = 'Dog.jpg'
 
58
  example_image_rgb = cv2.cvtColor(example_image, cv2.COLOR_BGR2RGB)
59
  example_sketch = image_to_sketch(example_image)
60
 
61
+ col1, col2 = st.columns(2)
62
+
63
  with col1:
 
64
  st.image(example_image_rgb, caption='Original Image', use_column_width=True)
65
+ with col2:
66
  st.image(example_sketch, caption='Sketch Image', use_column_width=True)
67
  else:
68
  st.error(f"Failed to load example image from path: {example_image_path}")
69
 
70
+ # User upload section
71
+ st.subheader("Upload Your Image")
72
+ uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
73
 
74
+ if uploaded_file is not None:
75
+ image = np.array(Image.open(uploaded_file))
76
+ st.image(image, caption='Uploaded Image', use_column_width=True)
77
+
78
+ st.write("Converting...")
79
+
80
+ sketch = image_to_sketch(image)
81
+
82
+ col3, col4 = st.columns(2)
83
 
84
+ with col3:
85
+ st.image(image, caption='Uploaded Image', use_column_width=True)
86
+ with col4:
87
  st.image(sketch, caption='Sketch', use_column_width=True)
88
 
89
+ # Provide a download link for the sketch image
90
+ im_pil = Image.fromarray(sketch)
91
+ im_pil.save("sketch.png")
92
+ with open("sketch.png", "rb") as file:
93
+ btn = st.download_button(
94
+ label="Download Sketch",
95
+ data=file,
96
+ file_name="sketch.png",
97
+ mime="image/png"
98
+ )
99
+ else:
100
+ st.info("Please upload an image to convert.")
101
 
102
  # Footer
103
  st.markdown("""