Update app.py

app.py

@@ -4,7 +4,7 @@ import numpy as np
 from PIL import Image
 import io
 
-st.title("🖼️ Image Augmentation Tool (Multi-image Support)")
+st.title(" Image Augmentation Tool (Multi-image Support)")
 
 uploaded_files = st.file_uploader(
     "Upload one or more images", 
@@ -28,9 +28,10 @@ if uploaded_files:
 
     for uploaded_file in uploaded_files:
         st.markdown("---")
-        st.subheader(f"📷 {uploaded_file.name}")
+        st.subheader(f" {uploaded_file.name}")
 
-        image = Image.open(uploaded_file)
+        # Open and convert image
+        image = Image.open(uploaded_file).convert("RGB")
         img_array = np.array(image)
 
         transformed_img = img_array.copy()
@@ -61,18 +62,17 @@ if uploaded_files:
             col1, col2 = st.columns(2)
 
             with col1:
-                st.image(img, caption="Original Image", use_container_width=True)
-
+                st.image(image, caption="Original Image", use_container_width=True)
 
             with col2:
                 if option != "None":
                     if len(transformed_img.shape) == 2:
-                        st.image(transformed_img, caption=f"Transformed: {option}", use_column_width=True, channels="GRAY")
+                        st.image(transformed_img, caption=f"Transformed: {option}", use_container_width=True, channels="GRAY")
                     else:
-                        st.image(transformed_img, caption=f"Transformed: {option}", use_column_width=True)
+                        st.image(transformed_img, caption=f"Transformed: {option}", use_container_width=True)
 
                     # Download button
-                    result = Image.fromarray(transformed_img)
+                    result = Image.fromarray(transformed_img if len(transformed_img.shape) == 3 else cv2.cvtColor(transformed_img, cv2.COLOR_GRAY2RGB))
                     buf = io.BytesIO()
                     result.save(buf, format="PNG")
                     byte_im = buf.getvalue()