Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,40 +1,42 @@
 import altair as alt
-import numpy as np
 import pandas as pd
 import streamlit as st
+from PIL import Image
+import cv2
+import numpy as np
+import io
+import zipfile
+
+st.title("Image Augmentation App")
+uploaded_image = st.file_uploader("Upload an Image", type=["jpg", "jpeg", "png"])
+
+if uploaded_image:
+    img = np.array(Image.open(uploaded_image))
+    st.image(img, caption="Uploaded Image", use_column_width=True)
+    
+    st.write("**Applying augmentations...**")
+    
+    transformations = {
+        "Translated": cv2.warpAffine(img, np.float32([[1, 0, 50], [0, 1, 50]]), (img.shape[1], img.shape[0])),
+        "Scaled": cv2.resize(img, None, fx=1.2, fy=1.2),
+        "Rotated": cv2.warpAffine(img, cv2.getRotationMatrix2D((img.shape[1] // 2, img.shape[0] // 2), 45, 1), (img.shape[1], img.shape[0])),
+        "Cropped": img[50:img.shape[0] - 50, 50:img.shape[1] - 50]
+    }
+    
+    augmented_images = []
+    zip_buffer = io.BytesIO()
+    
+    with zipfile.ZipFile(zip_buffer, "w") as zipf:
+        for i, (name, transformed_img) in enumerate(transformations.items()):
+            img_pil = Image.fromarray(cv2.cvtColor(transformed_img, cv2.COLOR_BGR2RGB))
+            img_bytes = io.BytesIO()
+            img_pil.save(img_bytes, format="JPEG")
+            zipf.writestr(f"aug_{i + 1}_{name}.jpg", img_bytes.getvalue())
+            augmented_images.append(img_pil)
+    
+    st.write("**Augmentation completed!**")
+    st.image(augmented_images, width=150)
+    
+    zip_buffer.seek(0)
+    st.download_button("Download Augmented Images", data=zip_buffer, file_name="augmented_images.zip", mime="application/zip")
 
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))