Update web-demos/hugging_face/app.py

web-demos/hugging_face/app.py

@@ -78,7 +78,6 @@ def get_frames_from_video(video_input, video_state):
     user_name = time.time()
     status_ok = True
     operation_log = [("[Must Do]", "Click image"), (": Video uploaded! Try to click the image shown in step2 to add masks.\n", None)]
-
     try:
         cap = cv2.VideoCapture(video_path)
         fps = cap.get(cv2.CAP_PROP_FPS)
@@ -90,29 +89,24 @@ def get_frames_from_video(video_input, video_state):
             if ret == True:
                 original_h, original_w = frame.shape[:2]
                 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))
-                    else:
-                        break
-                t = len(frames)
-                if t > 0:
-                    print(f'Inp video shape: t_{t}, s_{original_h}x{original_w}')
+            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))
                 else:
-                    print(f'Inp video shape: t_{t}, no input video!!!')
+                    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)))
-        
-    if not frames:
-        raise ValueError("Не удалось извлечь кадры из видео. Проверьте файл.")
-        print("Загружено кадров:", len(frames))
-        print("Тип frames:", type(frames))
 
     # initialize video_state
     if frames[0].shape[0] > 720 or frames[0].shape[1] > 720: