Update app.py

app.py

@@ -1,23 +1,7 @@
-import os
 import streamlit as st
 from PIL import Image, ImageEnhance
 import numpy as np
-
-# Check and ensure OpenCV (cv2) is installed with compatible NumPy
-try:
-    import cv2
-except ImportError:
-    st.warning("OpenCV not found or incompatible. Attempting to fix...")
-    os.system("pip install numpy==1.24.3 opencv-python-headless==4.8.0.74")
-    try:
-        import cv2
-    except ImportError:
-        st.error("Failed to fix OpenCV installation. Check requirements.txt for compatibility.")
-        st.stop()
-
-# Verify Versions
-st.sidebar.write(f"OpenCV Version: {cv2.__version__}")
-st.sidebar.write(f"NumPy Version: {np.__version__}")
+import cv2
 
 # Title and Description
 st.title("Image Enhancer App")
@@ -35,7 +19,7 @@ if uploaded_file:
 
     # Sidebar Filters
     st.sidebar.header("Enhancement Options")
-    
+
     # Brightness
     brightness = st.sidebar.slider("Brightness", 0.5, 3.0, 1.0)
     enhancer = ImageEnhance.Brightness(image)
@@ -51,11 +35,18 @@ if uploaded_file:
     enhancer = ImageEnhance.Sharpness(image)
     image = enhancer.enhance(sharpness)
 
-    # Saturation (Using OpenCV for Saturation)
+    # Saturation Adjustment (Using OpenCV for Saturation)
     saturation = st.sidebar.slider("Saturation", 0.5, 3.0, 1.0)
-    image_cv = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2HSV)
-    image_cv[:, :, 1] = np.clip(image_cv[:, :, 1] * saturation, 0, 255)
-    image = Image.fromarray(cv2.cvtColor(image_cv, cv2.COLOR_HSV2RGB))
+
+    # Ensure the image is in a valid NumPy format
+    image_np = np.array(image).astype(np.uint8)
+
+    if len(image_np.shape) == 3:  # Check if the image has color channels
+        image_cv = cv2.cvtColor(image_np, cv2.COLOR_RGB2HSV)
+        image_cv[:, :, 1] = np.clip(image_cv[:, :, 1] * saturation, 0, 255)
+        image = Image.fromarray(cv2.cvtColor(image_cv, cv2.COLOR_HSV2RGB))
+    else:
+        st.warning("Saturation adjustment is only available for color images.")
 
     # Display Enhanced Image
     st.image(image, caption="Enhanced Image", use_column_width=True)
@@ -67,4 +58,4 @@ if uploaded_file:
         st.sidebar.write("Image saved as `enhanced_image.png`!")
 
 # Footer
-st.markdown("**Built with Streamlit**. Deployed on Hugging Face.")
+st.markdown("Built by NS!")