.

app.py

@@ -6,9 +6,6 @@ import numpy as np
 from modules.streamlit_utils import *
 
 
-
-
-
 def main():
     is_mobile, screen_width = configure_page()
     display_banner(is_mobile)
@@ -23,34 +20,46 @@ def main():
     if uploaded_file is not None:
         cropped_image = display_image(uploaded_file, screen_width, is_mobile)
 
-    if cropped_image is not None:
+    if uploaded_file is not None:
         get_score_threshold(is_mobile)
+    
         if st.button("🚀 Launch Prediction"):
             st.session_state.image = launch_prediction(cropped_image, st.session_state.score_threshold, is_mobile, screen_width)
             st.session_state.original_prediction = st.session_state.prediction.copy()
             st.rerun()
-    
-    if 'prediction' in st.session_state and uploaded_file:
-        #if st.button("🔄 Refresh image"):
-            #st.rerun()
 
+    # Create placeholders for all sections
+    prediction_result_placeholder = st.empty()
+    additional_options_placeholder = st.empty()
+    modeler_placeholder = st.empty()
+
+
+    if 'prediction' in st.session_state and uploaded_file:
         if st.session_state.image != cropped_image:
             print('Image has changed')
-            #delete the prediction
+            # Delete the prediction
             del st.session_state.prediction
             return
 
-        with st.expander("Show result of prediction"):
-            with st.spinner('Waiting for result display...'):
-                display_options(st.session_state.crop_image, st.session_state.score_threshold, is_mobile, int(5/6 * screen_width))
+        with prediction_result_placeholder.container():
+            with st.expander("Show result of prediction"):
+                with st.spinner('Waiting for result display...'):
+                    display_options(st.session_state.crop_image, st.session_state.score_threshold, is_mobile, int(5/6 * screen_width))
 
         if not is_mobile:
-            modify_results()
+            with additional_options_placeholder.container():
+                modify_results()
 
-        modeler_options(is_mobile)
-                
-        
-        display_bpmn_modeler(is_mobile, screen_width)
+        with modeler_placeholder.container():
+            modeler_options(is_mobile)
+            display_bpmn_modeler(is_mobile, screen_width)
+    else:
+        prediction_result_placeholder.empty()
+        additional_options_placeholder.empty()
+        modeler_placeholder.empty()
+        # Create a lot of space for scrolling
+        for _ in range(50):
+            st.text("")
 
     gc.collect()