Update app.py

app.py

@@ -12,45 +12,46 @@ st.markdown("Upload an image and change its color dynamically!")
 uploaded_file = st.file_uploader("Upload an Image", type=["png", "jpg", "jpeg", "webp"])
 
 if uploaded_file is not None:
-    # Convert image to OpenCV format
+    # Load image using PIL and convert to NumPy array
     image = Image.open(uploaded_file)
     image = np.array(image)
-    
-# Ensure the image has 3 channels (RGB)
-if len(image.shape) == 2:  # Grayscale image
-    image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
-elif image.shape[-1] == 4:  # RGBA image
-    image = cv2.cvtColor(image, cv2.COLOR_RGBA2RGB)
-    
+
+    # Ensure the image has 3 channels (RGB)
+    if len(image.shape) == 2:  # Grayscale image (Single Channel)
+        image = cv2.cvtColor(image, cv2.COLOR_GRAY2RGB)
+    elif image.shape[-1] == 4:  # RGBA image (4 channels)
+        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
-    
+
     # 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))
-    
+
     # Intensity Slider
     intensity = st.slider("Color Intensity", 0, 100, 50)
-    
-    # Change Color Function
+
+    # 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)
-    
+
     # Apply Color Change
     modified_image = change_image_color(image, new_color, intensity)
     st.image(modified_image, caption="Modified Image", use_column_width=True)
-    
+
     # Download Button
     img_byte_arr = io.BytesIO()
     Image.fromarray(modified_image).save(img_byte_arr, format="PNG")
     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")