Add shortcut to examples

app_inpaint.py

@@ -11,6 +11,8 @@ import cv2
 import numpy as np
 import shutil
 import base64
+import logging
+
 
 @st.cache(show_spinner=False, allow_output_mutation=True, suppress_st_warning=True)
 def load_models():
@@ -53,7 +55,7 @@ def inpaint_text(img, text_coordinates):
 def file_selector(folder_path='.'):
     filenames = os.listdir(folder_path)
     selected_filename = st.selectbox('Select a file', filenames)
-    return os.path.join(folder_path, selected_filename)
+    return os.path.join(folder_path, selected_filename), selected_filename
 
 
 
@@ -130,16 +132,17 @@ with st.expander("Project Description", expanded=False):
 
         
         
-filename_example = None        
+file_example_path = None        
 if st.checkbox('Select a example'):
     folder_path = './Examples/'
     # if st.checkbox('Change directory'):
     #     folder_path = st.text_input('Enter folder path', '.')
-    filename_example = file_selector(folder_path=folder_path)
-    st.write('You selected `%s`' % filename_example)        
+    file_example_path, example_file_name  = file_selector(folder_path=folder_path)
+    st.write('You selected `%s`' % file_example_path)        
+     
         
 uploaded_file = st.file_uploader(label="Upload image", 
-                 type="jpg", 
+                 type=["jpg", "jpeg"], 
                  accept_multiple_files=False, 
                  key=None,
                  help=None, 
@@ -154,30 +157,35 @@ col1, col2, col3 = st.columns([2, 0.5, 2])
 
 
 
-if filename_example:
+if file_example_path and not uploaded_file:
     with col1:
-        st.header("Original")
-        img = Image.open( filename_example )
+        st.subheader("Original")
+        st.write(uploaded_file)
+        # st.write(f"./Examples_inpainted/{example_file_name.strip(".jpg")}_inpainted.jpeg")
+        img = Image.open( file_example_path )
         st.image(img, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
 
     with col3:
-        st.header("Inpainted")
+        st.subheader("Inpainted")
         with st.spinner('Wait for it...'):
-            img_array =  np.array(Image.open( filename_example ))
-            # detect text
-            bounds = reader.readtext(img_array, detail=1) #detail=1 # [(coordinates, detected text, confidence threshold)]
-            text_coordinates = [ bound[0] for bound in bounds]
-            # inpaint text coordinates
-            inpaint_image = inpaint_text(img_array, text_coordinates)
+            time.sleep(1)
+            example_file_name = example_file_name.strip(".jpg")
+            inpaint_image = f"./Examples_inpainted/{example_file_name}_inpaint.jpeg"
+            # img_array =  np.array(Image.open( file_example_path ))
+            # # detect text
+            # bounds = reader.readtext(img_array, detail=1) #detail=1 # [(coordinates, detected text, confidence threshold)]
+            # text_coordinates = [ bound[0] for bound in bounds]
+            # # inpaint text coordinates
+            # inpaint_image = inpaint_text(img_array, text_coordinates)
             st.image(inpaint_image, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto") 
         
 if uploaded_file:
         with col1:
-            st.header("Original")
+            st.subheader("Original")
             st.image(uploaded_file, caption=None, width=None, use_column_width=None, clamp=False, channels="RGB", output_format="auto")
 
         with col3:
-            st.header("Inpainted")
+            st.subheader("Inpainted")
             with st.spinner('Wait for it...'):
                 # Transform loaded file to bytes
                 bytes_data = uploaded_file.getvalue()