allow the user to rotate the image and change the light

modules/streamlit_utils.py

@@ -304,19 +304,59 @@ def load_user_image(img_selected, is_mobile):
     return uploaded_file
 
 def display_image(uploaded_file, screen_width, is_mobile):
-    
+    if 'rotation_angle' not in st.session_state:
+        st.session_state.rotation_angle = 0  # Initialize the rotation angle in session state
+
+    if 'brightness' not in st.session_state:
+        st.session_state.brightness = 1.0  # Initialize brightness in session state
+
+    def rotate_image(angle):
+        st.session_state.rotation_angle += angle
+
+    def adjust_brightness(image, brightness):
+        enhancer = ImageEnhance.Brightness(image)
+        return enhancer.enhance(brightness)
+
     with st.spinner('Waiting for image display...'):
         original_image = get_image(uploaded_file)
         resized_image = original_image.resize((screen_width // 2, int(original_image.height * (screen_width // 2) / original_image.width)))
 
+        with st.expander("Rotate and adjust brightness"):
+            if not is_mobile:
+                col1, col2 = st.columns([1.5, 1])
+                with col1:  
+                    st.session_state.brightness = st.slider("Adjust Brightness", min_value=0.2, max_value=2.0, value=1.0, step=0.1)
+            else:
+                st.session_state.brightness = st.slider("Adjust Brightness", min_value=0.2, max_value=2.0, value=1.0, step=0.1)
+
+            # Add buttons to rotate the image next to each other
+            col1, col2 = st.columns([1, 1])
+            with col1:
+                if st.button("Rotate Left"):
+                    rotate_image(90)
+            with col2:
+                if st.button("Rotate Right"):
+                    rotate_image(-90)
+
+        # Apply the rotation angle from session state
+        rotated_image = resized_image.rotate(st.session_state.rotation_angle, expand=True)
+        original_image = original_image.rotate(st.session_state.rotation_angle, expand=True)
+
+        # Apply the brightness adjustment
+        adjusted_image = adjust_brightness(rotated_image, st.session_state.brightness)
+        original_image = adjust_brightness(original_image, st.session_state.brightness)
+
         if not is_mobile:
-            cropped_image = crop_image(resized_image, original_image)
+            cropped_image = crop_image(adjusted_image, original_image)
         else:
-            st.image(resized_image, caption="Image", use_column_width=False, width=int(4/5 * screen_width))
+            st.image(adjusted_image, caption="Image", use_column_width=False, width=int(4/5 * screen_width))
             cropped_image = original_image
 
     return cropped_image
 
+
+
+
 def crop_image(resized_image, original_image):
     marge = 10
     cropped_box = st_cropper(