Update app.py

app.py

@@ -4,48 +4,71 @@ import numpy as np
 from PIL import Image
 import io
 
-st.set_page_config(page_title="Image Color Changer", page_icon="🎨", layout="centered")
+st.set_page_config(page_title="🎨 Image Color Changer", page_icon="🎨", layout="centered")
 st.title("🎨 Image Color Changer")
-st.markdown("Upload an image and change its color dynamically!")
+st.markdown("Upload an image, select a color, and change its appearance dynamically!")
 
 # Upload Image
 uploaded_file = st.file_uploader("Upload an Image", type=["png", "jpg", "jpeg", "webp"])
 
 if uploaded_file is not None:
-    # Load image using PIL and convert to NumPy array
+    # Convert image to OpenCV format
     image = Image.open(uploaded_file)
     image = np.array(image)
 
     # Ensure the image has 3 channels (RGB)
-    if len(image.shape) == 2:  # Grayscale image (Single Channel)
+    if len(image.shape) == 2:  # Convert grayscale to RGB
         image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
-    elif image.shape[-1] == 4:  # RGBA image (4 channels)
+    elif image.shape[-1] == 4:  # Convert RGBA to RGB
         image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
 
-    # Display Original Image
     st.image(image, caption="Original Image", use_column_width=True)
 
     # Color Picker
-    st.subheader("Select Target Color")
-    target_color = st.color_picker("Choose a color", "#ff0000")  # Default red
+    st.subheader("Select Target Color to Replace")
+    selected_color = st.color_picker("Pick a color to change", "#ff0000")  # Default Red
 
     # Convert Hex to RGB
-    hex_color = target_color.lstrip("#")
-    new_color = tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
+    hex_color = selected_color.lstrip("#")
+    target_rgb = tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))
 
-    # Intensity Slider
-    intensity = st.slider("Color Intensity", 0, 100, 50)
+    # Color Tolerance
+    tolerance = st.slider("Color Tolerance (Higher = More Area)", 0, 100, 30)
 
-    # Function to Change Image Color
-    def change_image_color(img, new_rgb, intensity=50):
-        img_hsv = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
-        img_hsv[:, :, 0] = new_rgb[0] * (intensity / 100)  # Adjust Hue
-        img_hsv[:, :, 1] = new_rgb[1] * (intensity / 100)  # Adjust Saturation
-        img_hsv[:, :, 2] = new_rgb[2] * (intensity / 100)  # Adjust Brightness
-        return cv2.cvtColor(img_hsv, cv2.COLOR_HSV2RGB)
+    # Replace Color Function
+    def replace_color(img, target_rgb, new_rgb, tolerance):
+        # Convert to HSV for better color detection
+        hsv_img = cv2.cvtColor(img, cv2.COLOR_RGB2HSV)
+
+        # Convert target RGB to HSV
+        target_hsv = cv2.cvtColor(np.uint8([[target_rgb]]), cv2.COLOR_RGB2HSV)[0][0]
+
+        # Define HSV range for mask
+        lower_bound = np.array([max(0, target_hsv[0] - tolerance), 50, 50])
+        upper_bound = np.array([min(179, target_hsv[0] + tolerance), 255, 255])
+
+        # Create a mask for selected color
+        mask = cv2.inRange(hsv_img, lower_bound, upper_bound)
+
+        # Convert new RGB to HSV
+        new_hsv = cv2.cvtColor(np.uint8([[new_rgb]]), cv2.COLOR_RGB2HSV)[0][0]
+
+        # Apply color change where the mask is active
+        hsv_img[mask > 0] = new_hsv
+
+        # Convert back to RGB
+        return cv2.cvtColor(hsv_img, cv2.COLOR_HSV2RGB)
+
+    # Select Replacement Color
+    st.subheader("Choose New Color")
+    new_color = st.color_picker("Pick the replacement color", "#00ff00")  # Default Green
+
+    # Convert New Color to RGB
+    hex_new_color = new_color.lstrip("#")
+    new_rgb = tuple(int(hex_new_color[i:i+2], 16) for i in (0, 2, 4))
 
     # Apply Color Change
-    modified_image = change_image_color(image, new_color, intensity)
+    modified_image = replace_color(image, target_rgb, new_rgb, tolerance)
     st.image(modified_image, caption="Modified Image", use_column_width=True)
 
     # Download Button
@@ -54,4 +77,4 @@ if uploaded_file is not None:
     st.download_button("Download Modified Image", img_byte_arr.getvalue(), file_name="modified_image.png", mime="image/png")
 
 st.markdown("---")
-st.markdown("💡 Developed with ❤️ using Streamlit, OpenCV, and NumPy")
+st.markdown("💡 Developed with ❤️ using Streamlit, OpenCV, and NumPy")