Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -92,48 +92,48 @@ def get_frames_from_video(video_input, video_state):
                 frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
                 status_ok = False
             else:
-            while cap.isOpened():
-                ret, frame = cap.read()
-                if ret == True:
-                    # resize input image
-                    original_h, original_w = frame.shape[:2]
-                    frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+                while cap.isOpened():
+                    ret, frame = cap.read()
+                    if ret == True:
+                        # resize input image
+                        original_h, original_w = frame.shape[:2]
+                        frames.append(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+                    else:
+                        break
+                t = len(frames)
+                if t > 0:
+                    print(f'Inp video shape: t_{t}, s_{original_h}x{original_w}')
                 else:
-                    break
-            t = len(frames)
-            if t > 0:
-                print(f'Inp video shape: t_{t}, s_{original_h}x{original_w}')
-            else:
-                print(f'Inp video shape: t_{t}, no input video!!!')
-    except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
-        status_ok = False
-        print("read_frame_source:{} error. {}\n".format(video_path, str(e)))
+                    print(f'Inp video shape: t_{t}, no input video!!!')
+        except (OSError, TypeError, ValueError, KeyError, SyntaxError) as e:
+            status_ok = False
+            print("read_frame_source:{} error. {}\n".format(video_path, str(e)))
+        
+        # initialize video_state
+        if frames[0].shape[0] > 720 or frames[0].shape[1] > 720:
+             operation_log = [(f"Video uploaded! Try to click the image shown in step2 to add masks. (You uploaded a video with a size of {original_w}x{original_h}, and the length of its longest edge exceeds 720 pixels. We may resize the input video during processing.)", "Normal")]
     
-    # initialize video_state
-    if frames[0].shape[0] > 720 or frames[0].shape[1] > 720:
-         operation_log = [(f"Video uploaded! Try to click the image shown in step2 to add masks. (You uploaded a video with a size of {original_w}x{original_h}, and the length of its longest edge exceeds 720 pixels. We may resize the input video during processing.)", "Normal")]
-
-    video_state = {
-        "user_name": user_name,
-        "video_name": os.path.split(video_path)[-1],
-        "origin_images": frames,
-        "painted_images": frames.copy(),
-        "masks": [np.zeros((original_h, original_w), np.uint8)]*len(frames),
-        "logits": [None]*len(frames),
-        "select_frame_number": 0,
-        "fps": fps
-        }
-    video_info = "Video Name: {},\nFPS: {},\nTotal Frames: {},\nImage Size:{}".format(video_state["video_name"], round(video_state["fps"], 0), length, (original_w, original_h))
-    model.samcontroler.sam_controler.reset_image() 
-    model.samcontroler.sam_controler.set_image(video_state["origin_images"][0])
-    return video_state, video_info, video_state["origin_images"][0], gr.update(visible=status_ok, maximum=len(frames), value=1), gr.update(visible=status_ok, maximum=len(frames), value=len(frames)), \
-                        gr.update(visible=status_ok), gr.update(visible=status_ok), \
-                        gr.update(visible=status_ok), gr.update(visible=status_ok),\
-                        gr.update(visible=status_ok), gr.update(visible=status_ok), \
-                        gr.update(visible=status_ok), gr.update(visible=status_ok), \
-                        gr.update(visible=status_ok), gr.update(visible=status_ok), \
-                        gr.update(visible=status_ok), gr.update(visible=status_ok, choices=[], value=[]), \
-                        gr.update(visible=True, value=operation_log), gr.update(visible=status_ok, value=operation_log)
+        video_state = {
+            "user_name": user_name,
+            "video_name": os.path.split(video_path)[-1],
+            "origin_images": frames,
+            "painted_images": frames.copy(),
+            "masks": [np.zeros((original_h, original_w), np.uint8)]*len(frames),
+            "logits": [None]*len(frames),
+            "select_frame_number": 0,
+            "fps": fps
+            }
+        video_info = "Video Name: {},\nFPS: {},\nTotal Frames: {},\nImage Size:{}".format(video_state["video_name"], round(video_state["fps"], 0), length, (original_w, original_h))
+        model.samcontroler.sam_controler.reset_image() 
+        model.samcontroler.sam_controler.set_image(video_state["origin_images"][0])
+        return video_state, video_info, video_state["origin_images"][0], gr.update(visible=status_ok, maximum=len(frames), value=1), gr.update(visible=status_ok, maximum=len(frames), value=len(frames)), \
+                            gr.update(visible=status_ok), gr.update(visible=status_ok), \
+                            gr.update(visible=status_ok), gr.update(visible=status_ok),\
+                            gr.update(visible=status_ok), gr.update(visible=status_ok), \
+                            gr.update(visible=status_ok), gr.update(visible=status_ok), \
+                            gr.update(visible=status_ok), gr.update(visible=status_ok), \
+                            gr.update(visible=status_ok), gr.update(visible=status_ok, choices=[], value=[]), \
+                            gr.update(visible=True, value=operation_log), gr.update(visible=status_ok, value=operation_log)
 
 def select_template(image_selection_slider, video_state, interactive_state, mask_dropdown):