Update app.py

app.py

@@ -5,116 +5,108 @@ import io
 # Set Streamlit page configuration
 st.set_page_config(page_title="Image Transformer", page_icon="🖼️", layout="wide")
 
+# Sidebar for navigation and options
+st.sidebar.title("Transformation Options")
+operation = st.sidebar.selectbox(
+    "Choose an operation to perform:",
+    ("Mirror", "Resize", "Crop", "Rotate", "Black/White", "Pixelate", "Compress")
+)
+
+# Main title for the app
 st.title("Image Manipulation with Generative AI")
-st.write("Select an option and upload an image to apply the transformation")
-
-# Create columns for layout
-col1, col2 = st.columns([1, 2])
-
-with col1:
-    # Transformation options visible before image upload
-    st.header("Transformation Options")
-    operation = st.selectbox(
-        "Choose an operation to perform:",
-        ("Mirror", "Resize", "Crop", "Rotate", "Black/White", "Pixelate", "Compress")
-    )
-
-    # Show relevant sliders or options based on the chosen operation
-    if operation == "Resize":
+st.write("Upload an image and apply transformations based on your selection")
+
+# Upload image
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+
+if uploaded_file is not None:
+    # Load the image
+    image = Image.open(uploaded_file)
+    
+    # Show the original image (smaller size)
+    st.image(image, caption="Original Image", use_column_width=False, width=250)
+
+    # Apply transformations based on the selected operation
+    if operation == "Mirror":
+        st.header("Mirror Image")
+        if st.button("Apply Mirror"):
+            mirrored_image = ImageOps.mirror(image)
+            st.image(mirrored_image, caption="Mirrored Image", use_column_width=False, width=250)
+            
+            # Download button for mirrored image
+            buffered = io.BytesIO()
+            mirrored_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="mirrored_image.png", mime="image/png")
+
+    elif operation == "Resize":
+        st.header("Resize Image")
         new_width = st.slider("Resize Image - Width", 50, 500, 100)
         new_height = st.slider("Resize Image - Height", 50, 500, 100)
+        if st.button("Apply Resize"):
+            resized_image = image.resize((new_width, new_height))
+            st.image(resized_image, caption="Resized Image", use_column_width=False, width=250)
+
+            # Download button for resized image
+            buffered = io.BytesIO()
+            resized_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="resized_image.png", mime="image/png")
 
     elif operation == "Crop":
+        st.header("Crop Image")
         crop_values = st.slider("Crop Image - (left, upper, right, lower)", 0, 100, (10, 10, 90, 90))
+        if st.button("Apply Crop"):
+            cropped_image = image.crop(crop_values)
+            st.image(cropped_image, caption="Cropped Image", use_column_width=False, width=250)
+
+            # Download button for cropped image
+            buffered = io.BytesIO()
+            cropped_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="cropped_image.png", mime="image/png")
 
     elif operation == "Rotate":
+        st.header("Rotate Image")
         angle = st.slider("Rotate Image", 0, 360, 0)
+        if st.button("Apply Rotate"):
+            rotated_image = image.rotate(angle)
+            st.image(rotated_image, caption="Rotated Image", use_column_width=False, width=250)
+
+            # Download button for rotated image
+            buffered = io.BytesIO()
+            rotated_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="rotated_image.png", mime="image/png")
+
+    elif operation == "Black/White":
+        st.header("Convert to Black/White")
+        if st.button("Apply Black/White"):
+            bw_image = image.convert("L")
+            st.image(bw_image, caption="Black & White Image", use_column_width=False, width=250)
+
+            # Download button for black-and-white image
+            buffered = io.BytesIO()
+            bw_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="bw_image.png", mime="image/png")
 
     elif operation == "Pixelate":
+        st.header("Pixelate Image")
         pixel_size = st.slider("Pixelate Image", 1, 20, 10)
+        if st.button("Apply Pixelate"):
+            small = image.resize((image.width // pixel_size, image.height // pixel_size), Image.NEAREST)
+            pixelated_image = small.resize((image.width, image.height), Image.NEAREST)
+            st.image(pixelated_image, caption="Pixelated Image", use_column_width=False, width=250)
+
+            # Download button for pixelated image
+            buffered = io.BytesIO()
+            pixelated_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="pixelated_image.png", mime="image/png")
 
     elif operation == "Compress":
+        st.header("Compress Image")
         quality = st.slider("Compress Image (Quality)", 10, 100, 80)
-
-with col2:
-    # Upload image
-    uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
-
-    if uploaded_file is not None:
-        # Load image
-        image = Image.open(uploaded_file)
-        
-        # Display smaller uploaded image
-        st.image(image, caption="Original Image", use_column_width=False, width=250)
-
-        # Apply transformation based on user selection
-        if operation == "Mirror":
-            if st.button("Apply Mirror"):
-                mirrored_image = ImageOps.mirror(image)
-                st.image(mirrored_image, caption="Mirrored Image", use_column_width=False, width=250)
-
-                # Download button for mirrored image
-                buffered = io.BytesIO()
-                mirrored_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="mirrored_image.png", mime="image/png")
-
-        elif operation == "Resize":
-            if st.button("Apply Resize"):
-                resized_image = image.resize((new_width, new_height))
-                st.image(resized_image, caption="Resized Image", use_column_width=False, width=250)
-
-                # Download button for resized image
-                buffered = io.BytesIO()
-                resized_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="resized_image.png", mime="image/png")
-
-        elif operation == "Crop":
-            if st.button("Apply Crop"):
-                cropped_image = image.crop(crop_values)
-                st.image(cropped_image, caption="Cropped Image", use_column_width=False, width=250)
-
-                # Download button for cropped image
-                buffered = io.BytesIO()
-                cropped_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="cropped_image.png", mime="image/png")
-
-        elif operation == "Rotate":
-            if st.button("Apply Rotate"):
-                rotated_image = image.rotate(angle)
-                st.image(rotated_image, caption="Rotated Image", use_column_width=False, width=250)
-
-                # Download button for rotated image
-                buffered = io.BytesIO()
-                rotated_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="rotated_image.png", mime="image/png")
-
-        elif operation == "Black/White":
-            if st.button("Apply Black/White"):
-                bw_image = image.convert("L")
-                st.image(bw_image, caption="Black & White Image", use_column_width=False, width=250)
-
-                # Download button for black-and-white image
-                buffered = io.BytesIO()
-                bw_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="bw_image.png", mime="image/png")
-
-        elif operation == "Pixelate":
-            if st.button("Apply Pixelate"):
-                small = image.resize((image.width // pixel_size, image.height // pixel_size), Image.NEAREST)
-                pixelated_image = small.resize((image.width, image.height), Image.NEAREST)
-                st.image(pixelated_image, caption="Pixelated Image", use_column_width=False, width=250)
-
-                # Download button for pixelated image
-                buffered = io.BytesIO()
-                pixelated_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="pixelated_image.png", mime="image/png")
-
-        elif operation == "Compress":
-            if st.button("Apply Compress"):
-                compressed_image = image.copy()
-                st.image(compressed_image, caption=f"Compressed Image (Quality: {quality})", use_column_width=False, width=250)
-
-                # Download button for compressed image
-                buffered = io.BytesIO()
-                compressed_image.save(buffered, format="PNG")
-                st.download_button("Download Image", data=buffered.getvalue(), file_name="compressed_image.png", mime="image/png")
+        if st.button("Apply Compress"):
+            compressed_image = image.copy()
+            st.image(compressed_image, caption=f"Compressed Image (Quality: {quality})", use_column_width=False, width=250)
+
+            # Download button for compressed image
+            buffered = io.BytesIO()
+            compressed_image.save(buffered, format="PNG")
+            st.download_button("Download Image", data=buffered.getvalue(), file_name="compressed_image.png", mime="image/png")