Update app.py

app.py

@@ -67,10 +67,6 @@ except Exception as ex:
     st.error(f"Unable to load model. Check the specified path: {model_path}")
     st.error(ex)
 
-# We'll use a session_state variable to remember the current slider value.
-if "frame_slider" not in st.session_state:
-    st.session_state.frame_slider = 0
-
 # A container to display the currently viewed frame.
 viewer_slot = st.empty()
 
@@ -135,26 +131,24 @@ if st.sidebar.button("Let's Detect Wildfire"):
                 else:
                     progress_text.text(f"Processing frame {frame_count}")
 
-                # Only update slider if we have at least one processed frame.
+                # Update slider only if we have at least one processed frame.
                 if len(processed_frames) > 0:
-                    # Clear the previous slider widget.
+                    # Clear previous slider widget.
                     slider_container.empty()
-                    # Determine the current slider value.
-                    curr_slider_val = st.session_state.get("frame_slider", len(processed_frames)-1)
-                    # Ensure the slider value is within the new bounds.
-                    if curr_slider_val > len(processed_frames)-1:
-                        curr_slider_val = len(processed_frames)-1
-                    # Create a new slider. This slider's key is fixed because we cleared the container beforehand.
+                    # Use the slider widget's own value from session_state, or default to the last frame.
+                    default_val = st.session_state.get("frame_slider", len(processed_frames)-1)
+                    # Ensure default value is within bounds.
+                    if default_val > len(processed_frames)-1:
+                        default_val = len(processed_frames)-1
+                    # Create the slider. The key "frame_slider" will automatically update st.session_state.
                     slider_val = slider_container.slider(
                         "Frame Viewer", 
                         min_value=0, 
                         max_value=len(processed_frames)-1, 
-                        value=curr_slider_val, 
+                        value=default_val, 
                         step=1, 
                         key="frame_slider"
                     )
-                    st.session_state.frame_slider = slider_val
-
                     # If the user is at the most recent frame, update the viewer.
                     if slider_val == len(processed_frames)-1:
                         viewer_slot.image(processed_frames[-1], caption=f"Frame {len(processed_frames)-1}", use_column_width=True)