Update app.py

app.py

@@ -2,12 +2,13 @@ import streamlit as st
 import cv2
 import numpy as np
 from PIL import Image
+import io
 
 # Function to convert image to sketch
 def image_to_sketch(image):
     gray_image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
     inverted_image = 255 - gray_image
-    blurred_image = cv2.GaussianBlur(inverted_image, (21, 21), 0)
+    blurred_image = cv2.GaussianBlur(inverted_image, (31, 31), 0)
     inverted_blurred = 255 - blurred_image
     sketch = cv2.divide(gray_image, inverted_blurred, scale=256.0)
     return sketch
@@ -37,12 +38,13 @@ st.markdown("""
         margin-bottom: 70px;
     }
     .spacing {
-        margin: 10px 0;
+        margin: 10px 10px;
     }
     .centered-button {
         display: flex;
         justify-content: center;
         align-items: center;
+        gap: 10px;
     }
     </style>
     """, unsafe_allow_html=True)
@@ -51,7 +53,7 @@ st.markdown("""
 st.markdown('<p class="title">🎨 Image to Sketch Converter</p>', unsafe_allow_html=True)
 st.markdown("""
     Convert your images into beautiful sketches with this simple app.
-    Upload an image, and get the sketch version instantly!
+    Upload an image, and get the sketch version instantly! You can even download the sketch.
     """)
 
 # Example conversions
@@ -80,11 +82,17 @@ st.subheader("Upload Your Image")
 uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
 
 if uploaded_file is not None:
-    image = np.array(Image.open(uploaded_file))
+    # Load the image
+    image = Image.open(uploaded_file)
+    image_np = np.array(image)
+    
+    # Determine the format of the uploaded image
+    image_format = image.format.lower()
     
     st.write("Converting...")
 
-    sketch = image_to_sketch(image)
+    # Convert the image to a sketch
+    sketch = image_to_sketch(image_np)
     
     col3, col4 = st.columns(2)
 
@@ -96,13 +104,19 @@ if uploaded_file is not None:
     # Add some space before the button
     st.markdown('<div class="spacing"></div>', unsafe_allow_html=True)
     
+    # Convert the sketch to an image and save to an in-memory file object
+    sketch_image = Image.fromarray(sketch)
+    buf = io.BytesIO()
+    sketch_image.save(buf, format=image_format.upper())
+    byte_im = buf.getvalue()
+    
     # Provide a download link for the sketch image in the center
     st.markdown('<div class="centered-button">', unsafe_allow_html=True)
     btn = st.download_button(
         label="Download Sketch",
-        data=Image.fromarray(sketch).tobytes(),
-        file_name="sketch.png",
-        mime="image/png"
+        data=byte_im,
+        file_name=f"sketch.{image_format}",
+        mime=f"image/{image_format}"
     )
     st.markdown('</div>', unsafe_allow_html=True)
 else: