Update app.py

app.py

@@ -1,53 +1,65 @@
 import streamlit as st
-from PIL import Image, ImageDraw, ImageFont, ImageEnhance
+from PIL import Image, ImageEnhance, ImageDraw, ImageFont
 import numpy as np
 import io
 
-# Function to create an image with shapes and text
-def create_image():
-    # Create a blank white image (width=500px, height=500px)
-    img = Image.new('RGB', (500, 500), color='white')
-    draw = ImageDraw.Draw(img)
-    
-    # Draw shapes
-    draw.rectangle([50, 50, 450, 450], outline="black", width=5)  # Rectangle
-    draw.ellipse([100, 100, 400, 400], outline="blue", width=5)   # Circle
-    draw.line([0, 0, 500, 500], fill="red", width=3)               # Diagonal line
-    
-    # Add text to the image
-    font = ImageFont.load_default()
-    text = "Custom Image"
-    textwidth, textheight = draw.textsize(text, font)
-    text_position = ((500 - textwidth) // 2, (500 - textheight) // 2)  # Center the text
-    draw.text(text_position, text, fill="green", font=font)
-    
-    return img
+# Function to apply grayscale filter
+def apply_grayscale(image):
+    return image.convert("L")
 
 # Function to apply brightness adjustment
 def apply_brightness(image, factor):
     enhancer = ImageEnhance.Brightness(image)
     return enhancer.enhance(factor)
 
-# Streamlit UI
-st.title("Create and Edit Custom Images")
-st.write("Create a custom image with shapes and text, and apply filters to it.")
-
-# Create a custom image
-image = create_image()
-
-# Show the original image
-st.image(image, caption="Custom Created Image", use_column_width=True)
-
-# Apply brightness adjustment
-brightness_factor = st.slider("Adjust Brightness", 0.0, 2.0, 1.0)
-image = apply_brightness(image, brightness_factor)
+# Function to add text to the image
+def add_text(image, text):
+    draw = ImageDraw.Draw(image)
+    font = ImageFont.load_default()
+    width, height = image.size
+    text_width, text_height = draw.textsize(text, font=font)
+    
+    # Position text at the bottom center
+    x = (width - text_width) / 2
+    y = height - text_height - 10
+    draw.text((x, y), text, font=font, fill="white")
+    return image
 
-# Show edited image
-st.image(image, caption="Edited Custom Image", use_column_width=True)
+# Streamlit UI
+st.title("Image Editor")
+st.write("Upload an image and apply filters or add text to it.")
 
-# Download button for the edited image
-buffered = io.BytesIO()
-image.save(buffered, format="PNG")
-buffered.seek(0)
-st.download_button("Download Edited Image", buffered, file_name="custom_edited_image.png", mime="image/png")
+# Image upload
+uploaded_image = st.file_uploader("Upload an image", type=["jpg", "jpeg", "png"])
 
+if uploaded_image is not None:
+    # Open the image using PIL
+    image = Image.open(uploaded_image)
+    
+    # Show the original image
+    st.image(image, caption="Original Image", use_column_width=True)
+    
+    # Apply filters and editing options
+    st.sidebar.title("Edit Options")
+    
+    # Grayscale filter
+    if st.sidebar.checkbox("Apply Grayscale"):
+        image = apply_grayscale(image)
+    
+    # Brightness adjustment
+    brightness_factor = st.sidebar.slider("Adjust Brightness", 0.0, 2.0, 1.0)
+    image = apply_brightness(image, brightness_factor)
+    
+    # Add text
+    text = st.sidebar.text_input("Add Text to Image")
+    if text:
+        image = add_text(image, text)
+    
+    # Show edited image
+    st.image(image, caption="Edited Image", use_column_width=True)
+    
+    # Download button
+    buffered = io.BytesIO()
+    image.save(buffered, format="PNG")
+    buffered.seek(0)
+    st.download_button("Download Edited Image", buffered, file_name="edited_image.png", mime="image/png")