Update src/streamlit_app.py

src/streamlit_app.py

@@ -8,6 +8,10 @@ from get_labels_from_samples import export_all_sensor_frames_and_annotations
 st.title("Copy annotations from one dataset to another")
 st.subheader("For Multi-sensor datasets - Cuboids and bounding boxes")
 
+# Initialize success message in session state
+if "success" not in st.session_state:
+    st.session_state.success = ""
+
 api_key = st.text_input("Enter your API Key", type="password")
 source_uuid = st.text_input("Source UUID (the sample you want to copy from)", value="")
 source_frame_num = st.number_input("Source Frame Number", min_value=1, value=50, step=1)
@@ -15,6 +19,7 @@ target_uuid = st.text_input("Target UUID (the sample you want to copy to)", valu
 target_frame_num = st.number_input("Target Frame Number", min_value=1, value=1, step=1)
 
 if st.button("Overwrite Target Frame Annotations"):
+    st.session_state.success = ""  # Clear previous success message
     if not api_key or not source_uuid or not target_uuid:
         st.error("Please fill in all fields.")
     else:
@@ -62,6 +67,6 @@ if st.button("Overwrite Target Frame Annotations"):
                 labelset="ground-truth",  # Change this if you use a different labelset name
                 attributes=new_target_label.attributes.model_dump()
             )
-            st.success("Target label updated and uploaded to Segments.ai!")
+            st.session_state.success = "Target label updated and uploaded to Segments.ai!"
         except Exception as e:
             st.error(f"Error: {e}")