Update pages/4_Image Augmentation.py

pages/4_Image Augmentation.py

@@ -1 +1,33 @@
-import streamlit as st
+import streamlit as st
+from PIL import Image, ImageEnhance
+
+# Title of the app
+st.title("Image Data Augmentation")
+
+# File uploader to upload images
+uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "png", "jpeg"])
+
+if uploaded_file is not None:
+    # Open the uploaded image
+    image = Image.open(uploaded_file)
+    st.image(image, caption='Uploaded Image', use_column_width=True)
+    
+    st.write("Augment your image below:")
+
+    # Slider for adjusting brightness
+    brightness = st.slider("Brightness", 0.5, 3.0, 1.0)
+    enhancer = ImageEnhance.Brightness(image)
+    img_brightness = enhancer.enhance(brightness)
+    st.image(img_brightness, caption='Brightness Adjusted', use_column_width=True)
+
+    # Slider for adjusting contrast
+    contrast = st.slider("Contrast", 0.5, 3.0, 1.0)
+    enhancer = ImageEnhance.Contrast(image)
+    img_contrast = enhancer.enhance(contrast)
+    st.image(img_contrast, caption='Contrast Adjusted', use_column_width=True)
+
+    # Slider for adjusting sharpness
+    sharpness = st.slider("Sharpness", 0.5, 3.0, 1.0)
+    enhancer = ImageEnhance.Sharpness(image)
+    img_sharpness = enhancer.enhance(sharpness)
+    st.image(img_sharpness, caption='Sharpness Adjusted', use_column_width=True)