Add clear caches

app_pages/ocr_comparator.py

@@ -155,7 +155,7 @@ def app():
         if 'columns_size' not in st.session_state:
             st.session_state.columns_size = [2] + [1 for x in out_reader_type_list[1:]]
         if 'column_width' not in st.session_state:
-            st.session_state.column_width = [500] + [400 for x in out_reader_type_list[1:]]
+            st.session_state.column_width = [400] + [300 for x in out_reader_type_list[1:]]
         if 'columns_color' not in st.session_state:
             st.session_state.columns_color = ["rgb(228,26,28)"] + \
                                             ["rgb(0,0,0)" for x in out_reader_type_list[1:]]
@@ -851,7 +851,7 @@ def app():
         with show_reco.container():
             # Draw the results, 2 images per line
             reco_lines = math.ceil(len(in_reader_type_list) / 2)
-            column_width = 500
+            column_width = 400
             for ind_lig in range(0, reco_lines+1, 2):
                 cols = st.columns(2)
                 for ind_col in range(2):
@@ -882,10 +882,10 @@ def app():
         """
         with show_detect.container():
             columns_size = [1 for x in reader_type_list]
-            column_width  = [400 for x in reader_type_list]
+            column_width  = [300 for x in reader_type_list]
             columns_color = ["rgb(0,0,0)" for x in reader_type_list]
             columns_size[reader_type_dict[st.session_state.detect_reader]] = 2
-            column_width[reader_type_dict[st.session_state.detect_reader]] = 500
+            column_width[reader_type_dict[st.session_state.detect_reader]] = 400
             columns_color[reader_type_dict[st.session_state.detect_reader]] = "rgb(228,26,28)"
             columns = st.columns(columns_size, ) #gap='medium')
 
@@ -895,7 +895,7 @@ def app():
                 col.markdown(column_title, unsafe_allow_html=True)
                 if isinstance(list_images[ind_col+2], PIL.Image.Image):
                     col.image(list_images[ind_col+2], width=column_width[ind_col], \
-                            use_column_width=True)
+                              use_column_width=True)
                 else:
                     col.write(list_images[ind_col+2], use_column_width=True)
             st.session_state.columns_size = columns_size
@@ -920,7 +920,25 @@ def app():
     ###
     def raz():
         st.session_state.list_coordinates = []
-
+        st.session_state.list_images = []
+        st.session_state.detect_reader = reader_type_list[0]
+
+        st.session_state.columns_size = [2] + [1 for x in reader_type_list[1:]]
+        st.session_state.column_width = [400] + [300 for x in reader_type_list[1:]]
+        st.session_state.columns_color = ["rgb(228,26,28)"] + \
+                                         ["rgb(0,0,0)" for x in reader_type_list[1:]]
+
+        # Clear caches
+        easyocr_detect.clear()
+        ppocr_detect.clear()
+        mmocr_detect.clear()
+        tesserocr_detect.clear()
+        process_detect.clear()
+        get_cropped.clear()
+        easyocr_recog.clear()
+        ppocr_recog.clear()
+        mmocr_recog.clear()
+        tesserocr_recog.clear()
 
 
     ##----------- Initializations ---------------------------------------------------------------------
@@ -1151,7 +1169,7 @@ def app():
             if 'columns_size' not in st.session_state:
                 st.session_state.columns_size = [2] + [1 for x in reader_type_list[1:]]
             if 'column_width' not in st.session_state:
-                st.session_state.column_width = [500] + [400 for x in reader_type_list[1:]]
+                st.session_state.column_width = [400] + [300 for x in reader_type_list[1:]]
             if 'columns_color' not in st.session_state:
                 st.session_state.columns_color = ["rgb(228,26,28)"] + \
                                                 ["rgb(0,0,0)" for x in reader_type_list[1:]]
@@ -1175,7 +1193,7 @@ def app():
                     col.markdown(column_title, unsafe_allow_html=True)
                     if isinstance(list_images[no_col+2], PIL.Image.Image):
                         col.image(list_images[no_col+2], width=st.session_state.column_width[no_col], \
-                                use_column_width=True)
+                                  use_column_width=True)
                         list_ok_detect.append(reader_type_list[no_col])
                     else:
                         col.write(list_images[no_col+2], use_column_width=True)