sdc

App.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from utilitis import Draw,Add_Results
+from utilitis import Draw,Add_Results,Change_Image
 from PIL import Image
 
 st.title("Bienvenue à Textra Web App")
@@ -8,13 +8,16 @@ st.write("(PNG, JPG, JPEG)")
 uploaded_file = st.file_uploader("Ou selectioner une image:", type=["png", "jpg", "jpeg"], accept_multiple_files=False)
 
 if uploaded_file is not None:
+    image_initiale = Image.open(uploaded_file)
+    image_initiale = image_initiale.convert("RGB")
+    
     image = Image.open(uploaded_file)
     image = image.convert("RGB")
     image,Results = Draw(image)
     #st.write("Predicted Text:",type(image))
     st.image(image, caption="Uploaded Image", use_column_width=True)
     st.sidebar.title('Results')
-    st.sidebar.table(Results)
+    Change_Image(image,image_initiale)
     Add_Results(Results)
 
 

utilitis.py

@@ -160,3 +160,20 @@ def Add_Results(data):
 
 #############################################################################
 #############################################################################
+def Change_Image(image1,image2):
+        # Initialize session state
+        if 'current_image' not in st.session_state:
+            st.session_state.current_image = 'image1'
+
+        # Button to switch between images
+        if st.button('Image'):
+            if st.session_state.current_image == 'image1':
+                st.session_state.current_image = 'image2'
+            else:
+                st.session_state.current_image = 'image1'
+
+        # Display the selected image
+        if st.session_state.current_image == 'image1':
+            st.image(image1, caption='Output', use_column_width=True)
+        else:
+            st.image(image2, caption='Image initiale', use_column_width=True)