Update app.py

app.py

@@ -1,59 +1,61 @@
 import streamlit as st
+from PIL import Image, ImageEnhance
+import io
+
+# Function to show image
+def show_image(img):
+    st.image(img, caption="Edited Image", use_column_width=True)
+
+# Function to reset the image to the original state
+def reset_image():
+    st.session_state["image"] = st.session_state["original_image"]
 
 # Title of the app
-st.title("Temperature Converter")
-
-# Sidebar for user inputs
-st.sidebar.header("Select Conversion")
-from_unit = st.sidebar.selectbox("From Unit", ["Celsius", "Fahrenheit", "Kelvin"])
-to_unit = st.sidebar.selectbox("To Unit", ["Celsius", "Fahrenheit", "Kelvin"])
-
-# Input for temperature
-temperature = st.number_input(f"Enter temperature in {from_unit}", value=0.0)
-
-# Conversion Functions
-def celsius_to_fahrenheit(celsius):
-    return (celsius * 9/5) + 32
-
-def celsius_to_kelvin(celsius):
-    return celsius + 273.15
-
-def fahrenheit_to_celsius(fahrenheit):
-    return (fahrenheit - 32) * 5/9
-
-def fahrenheit_to_kelvin(fahrenheit):
-    return (fahrenheit - 32) * 5/9 + 273.15
-
-def kelvin_to_celsius(kelvin):
-    return kelvin - 273.15
-
-def kelvin_to_fahrenheit(kelvin):
-    return (kelvin - 273.15) * 9/5 + 32
-
-# Logic to perform conversion
-if from_unit == "Celsius":
-    if to_unit == "Fahrenheit":
-        converted_temp = celsius_to_fahrenheit(temperature)
-    elif to_unit == "Kelvin":
-        converted_temp = celsius_to_kelvin(temperature)
-    else:
-        converted_temp = temperature  # No conversion needed if the same unit
-
-elif from_unit == "Fahrenheit":
-    if to_unit == "Celsius":
-        converted_temp = fahrenheit_to_celsius(temperature)
-    elif to_unit == "Kelvin":
-        converted_temp = fahrenheit_to_kelvin(temperature)
-    else:
-        converted_temp = temperature  # No conversion needed if the same unit
-
-elif from_unit == "Kelvin":
-    if to_unit == "Celsius":
-        converted_temp = kelvin_to_celsius(temperature)
-    elif to_unit == "Fahrenheit":
-        converted_temp = kelvin_to_fahrenheit(temperature)
-    else:
-        converted_temp = temperature  # No conversion needed if the same unit
-
-# Display the result
-st.write(f"{temperature} {from_unit} is equal to {converted_temp:.2f} {to_unit}")
+st.title("Image Editor and Reset Tool")
+
+# Sidebar for uploading the original image
+st.sidebar.header("Upload Image")
+uploaded_image = st.sidebar.file_uploader("Upload your image", type=["jpg", "jpeg", "png"])
+
+# Save the original image in session state
+if uploaded_image is not None:
+    img = Image.open(uploaded_image)
+    
+    # Store the original image in session state
+    if "original_image" not in st.session_state:
+        st.session_state["original_image"] = img.copy()
+
+    st.session_state["image"] = img.copy()
+
+    show_image(st.session_state["image"])
+
+# Options to edit the image
+if uploaded_image is not None:
+    st.sidebar.header("Edit Image")
+    
+    # Rotation
+    rotate_angle = st.sidebar.slider("Rotate image", 0, 360, 0)
+    if rotate_angle != 0:
+        st.session_state["image"] = st.session_state["image"].rotate(rotate_angle)
+    
+    # Brightness Adjustment
+    brightness_factor = st.sidebar.slider("Brightness", 0.0, 2.0, 1.0)
+    enhancer = ImageEnhance.Brightness(st.session_state["image"])
+    st.session_state["image"] = enhancer.enhance(brightness_factor)
+
+    # Crop Image (User specifies the cropping box)
+    st.sidebar.subheader("Crop Image")
+    crop_x1 = st.sidebar.slider("Crop X1", 0, st.session_state["image"].width, 0)
+    crop_y1 = st.sidebar.slider("Crop Y1", 0, st.session_state["image"].height, 0)
+    crop_x2 = st.sidebar.slider("Crop X2", 0, st.session_state["image"].width, st.session_state["image"].width)
+    crop_y2 = st.sidebar.slider("Crop Y2", 0, st.session_state["image"].height, st.session_state["image"].height)
+    if crop_x1 < crop_x2 and crop_y1 < crop_y2:
+        st.session_state["image"] = st.session_state["image"].crop((crop_x1, crop_y1, crop_x2, crop_y2))
+    
+    # Reset Image
+    if st.sidebar.button("Reset to Original"):
+        reset_image()
+        st.sidebar.success("Image reset to original.")
+
+    # Display the edited image
+    show_image(st.session_state["image"])