Update app.py

app.py

@@ -1,33 +1,31 @@
 import streamlit as st
-import cv2
 import numpy as np
+import cv2
 
-# Function to create a Blake-style image with a circular center
 def create_blake_image(input_image):
-    # Create a circular mask
-    height, width, _ = input_image.shape
-    mask = np.zeros((height, width, 3), dtype=np.uint8)
-    center = (width // 2, height // 2)
-    radius = min(center)
-    cv2.circle(mask, center, radius, (255, 255, 255), thickness=-1)
-
-    # Apply the circular mask to the input image
-    circular_image = cv2.bitwise_and(input_image, mask)
-
-    return circular_image
-
-# Streamlit app
-st.title("Blake Image with Circular Center")
-picture = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
+    # Load the image using OpenCV
+    img = cv2.imread(input_image)
+    
+    # Get the shape of the original image
+    height, width, _ = img.shape
+    
+    # Create a circular mask with the same shape
+    mask = np.zeros((height, width), dtype=np.uint8)
+    circle_center = (width // 2, height // 2)
+    circle_radius = min(width, height) // 2
+    cv2.circle(mask, circle_center, circle_radius, 255, thickness=-1)
+    
+    # Apply the mask to the original image
+    result = cv2.bitwise_and(img, img, mask=mask)
+    
+    return result
+
+st.title("Blake-Style Image Converter")
+picture = st.file_uploader("Upload an image", type=["jpg", "png", "jpeg"])
 
 if picture:
-    # Read the image
-    input_image = cv2.imdecode(np.fromstring(picture.read(), np.uint8), 1)
-
-    # Create the Blake-style image
-    blake_image = create_blake_image(input_image)
-
-    # Display the original and Blake-style images
-    st.image([input_image, blake_image], caption=["Original Image", "Blake Image"], use_container_width=True)
-
-st.text("Upload an image to see the Blake-style transformation.")
+    # Convert the uploaded image to a Blake-style image
+    blake_image = create_blake_image(picture)
+    
+    # Display the result
+    st.image(blake_image, caption="Blake-Style Image", use_column_width=True)