fixed extra iteration while uploading new files

app.py

@@ -23,7 +23,7 @@ from app_utils import get_model_bin
 
 
 SESSION_STATE_VARIABLES = [
-    'model_folder','max_img_size','uploaded_file_key'
+    'model_folder','max_img_size','uploaded_file_key','uploaded_files'
 ]
 for i in SESSION_STATE_VARIABLES:
     if i not in st.session_state:
@@ -254,13 +254,20 @@ if colorizer is not None:
                                             )
 
     if uploaded_files:
-        
+        # Copy images to a session state
+        st.session_state['uploaded_files'] = uploaded_files
+                
+        st.session_state['uploaded_file_key'] = str(randint(1000, 100000000))  # remove the uploaded file from the UI
+        st.experimental_rerun()  # Force rerun to reload the file_uploader object with new key
+    
+    # If session state is not empty, we will process stored images
+    if not st.session_state['uploaded_files']:
         if len(uploaded_files) == 1:
             display_single_image(uploaded_files[0], st.session_state.max_img_size)
         elif len(uploaded_files) > 1:
             process_multiple_images(uploaded_files, st.session_state.max_img_size)
-        
-        st.session_state['uploaded_file_key'] = str(randint(1000, 100000000))  # remove the uploaded file from the UI
-        st.experimental_rerun()  # Force rerun to reload the file_uploader object with new key
+            
+        # Reset session state variable
+        st.session_state['uploaded_files'] = None