Update app.py

app.py

@@ -329,7 +329,12 @@ def update_ui(vis_frame_type):
         return gr.update(visible=False), gr.update(visible=True)
 
 def switch_working_frame(working_frame, scanned_frames, video_frames_dir):
-    # Use a regular expression to find the integer
+    new_working_frame = None
+    if working_frame == None:
+        new_working_frame = os.path.join(video_frames_dir, scanned_frames[0])
+        return new_working_frame, gr.State([]), gr.State([]), new_working_frame, new_working_frame, new_working_frame
+    else:
+        # Use a regular expression to find the integer
         match = re.search(r'frame_(\d+)', working_frame)
         if match:
             # Extract the integer from the match