Update app.py

app.py

@@ -21,6 +21,7 @@ import streamlit.components.v1 as components
 from PIL import ImageOps, Image
 from ultralytics import YOLO
 from src.tiles import *
+from src.image_transform import *
 
 
 
@@ -117,64 +118,56 @@ if st.session_state.screen == 1:
     
     image_path = "/home/user/app/test_images/"
     with row3_col1:
-        img_1 = Image.open(image_path + "IMG_5356.JPG")
-        img_1 = ImageOps.exif_transpose(img_1)
+        img_1 = image_transform("IMG_5356.JPG", image_path)
         st.image(img_1, use_column_width=True)
         if st.button("Select Image ↑", key="button_1", use_container_width=True):
             st.session_state.image = img_1
             st.session_state.screen = 2
             st.rerun()
     with row3_col2:
-        img_2 = Image.open(image_path + "IMG_5357.JPG")
-        img_2 = ImageOps.exif_transpose(img_2)
+        img_2 = image_transform("IMG_5357.JPG", image_path)
         st.image(img_2, use_column_width=True)
         if st.button("Select Image ↑", key="button_2", use_container_width=True):
             st.session_state.image = img_2
             st.session_state.screen = 2
             st.rerun()
     with row3_col3:
-        img_3 = Image.open(image_path + "IMG_5368.JPG")
-        img_3 = ImageOps.exif_transpose(img_3)
+        img_3 = image_transform("IMG_5368.JPG", image_path)
         st.image(img_3, use_column_width=True)
         if st.button("Select Image ↑", key="button_3", use_container_width=True):
             st.session_state.image = img_3
             st.session_state.screen = 2
             st.rerun()
     with row3_col4:
-        img_4 = Image.open(image_path + "IMG_5369.JPG", use_container_width=True)
-        img_4 = ImageOps.exif_transpose(img_4)
+        img_4 = image_transform("IMG_5369.JPG", image_path)
         st.image(img_4, use_column_width=True)
         if st.button("Select Image ↑", key="button_4", use_container_width=True):
             st.session_state.image = img_4
             st.session_state.screen = 2
             st.rerun()
     with row4_col1:
-        img_5 = Image.open(image_path + "IMG_5382.JPG")
-        img_5 = ImageOps.exif_transpose(img_5)
+        img_5 = image_transform("IMG_5382.JPG", image_path)
         st.image(img_5, use_column_width=True)
         if st.button("Select Image ↑", key="button_5", use_container_width=True):
             st.session_state.image = img_5
             st.session_state.screen = 2
             st.rerun()
     with row4_col2:
-        img_6 = Image.open(image_path + "IMG_5383.JPG")
-        img_6 = ImageOps.exif_transpose(img_6)
+        img_6 = image_transform("IMG_5383.JPG", image_path)
         st.image(img_6, use_column_width=True)
         if st.button("Select Image ↑", key="button_6", use_container_width=True):
             st.session_state.image = img_6
             st.session_state.screen = 2
             st.rerun()
     with row4_col3:
-        img_7 = Image.open(image_path + "IMG_5408.JPG")
-        img_7 = ImageOps.exif_transpose(img_7)
+        img_7 = image_transform("IMG_5408.JPG", image_path)
         st.image(img_7, use_column_width=True)
         if st.button("Select Image ↑", key="button_7", use_container_width=True):
             st.session_state.image = img_7
             st.session_state.screen = 2
             st.rerun()
     with row4_col4:
-        img_8 = Image.open(image_path + "IMG_5409.JPG")
-        img_8 = ImageOps.exif_transpose(img_8)
+        img_8 = image_transform("IMG_5409.JPG", image_path)
         st.image(img_8, use_column_width=True)
         if st.button("Select Image ↑", key="button_8", use_container_width=True):
             st.session_state.image = img_8