add spinner for waiting

app.py

@@ -311,7 +311,7 @@ def main():
 
     # Display the uploaded image if the user has uploaded an image
     if uploaded_file is not None:
-        with st.spinner('wait for image...'): 
+        with st.spinner('Waiting for image display...'): 
             original_image = get_image(uploaded_file)
             col1, col2 = st.columns(2)
 
@@ -335,19 +335,20 @@ def main():
                 with st.spinner('Processing...'):
                     perform_inference(model_object, model_arrow, st.session_state.crop_image, score_threshold)
                     st.session_state.prediction = modif_box_pos(st.session_state.prediction, object_dict)                    
-            
+                    st.success('Detection completed!')
                     print('Detection completed!')
 
 
     # If the prediction has been made and the user has uploaded an image, display the options for the user to annotate the image
     if 'prediction' in st.session_state and uploaded_file is not None:
-        st.success('Detection completed!')
-        display_options(st.session_state.crop_image, score_threshold)
+
+        with st.spinner('Waiting for result display...'): 
+            display_options(st.session_state.crop_image, score_threshold)
 
         #if st.session_state.prediction_up==True:
-        st.session_state.bpmn_xml = create_XML(st.session_state.prediction.copy(), st.session_state.text_mapping, st.session_state.scale)
-    
-        display_bpmn_xml(st.session_state.bpmn_xml)
+        with st.spinner('Waiting for BPMN modeler...'):
+            st.session_state.bpmn_xml = create_XML(st.session_state.prediction.copy(), st.session_state.text_mapping, st.session_state.scale)
+            display_bpmn_xml(st.session_state.bpmn_xml)
 
         # Force garbage collection after display
         gc.collect()