Update app.py

app.py

@@ -12,33 +12,41 @@ remover = Remover()
 def doo(video):
     cap = cv2.VideoCapture(video)
     fps = cap.get(cv2.CAP_PROP_FPS)
-    frame_width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
-    frame_height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
-    total_frames = int(cap.get(cv2.CAP_PROP_FRAME_COUNT))
     
-    writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, (frame_width, frame_height))
-    
-    batch_size = 30  # Process 30 frames at a time
+    writer = None
+    frame_count = 0
+    max_frames = 400  # Maximum number of frames to process at a time
     
     while cap.isOpened():
-        frames = []
-        for _ in range(batch_size):
-            ret, frame = cap.read()
-            if ret is False:
-                break
-            frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) 
-            img = Image.fromarray(frame).convert('RGB')
-            out = remover.process(img, type='green')
-            frames.append(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
-        
-        if not frames:
+        ret, frame = cap.read()
+    
+        if ret is False:
             break
             
-        for frame in frames:
-            writer.write(frame)
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) 
+        img = Image.fromarray(frame).convert('RGB')
+    
+        if writer is None:
+            writer = cv2.VideoWriter('output.mp4', cv2.VideoWriter_fourcc(*'mp4v'), fps, img.size)
+    
+        out = remover.process(img, type='green')
+        writer.write(cv2.cvtColor(np.array(out), cv2.COLOR_BGR2RGB))
+        
+        frame_count += 1
+        if frame_count >= max_frames:
+            cap.release()
+            writer.release()
+            
+            # Process the next segment
+            cap = cv2.VideoCapture(video)
+            for _ in range(frame_count):
+                cap.read()  # Skip frames that have already been processed
+            writer = None
+            frame_count = 0
     
     cap.release()
-    writer.release()
+    if writer is not None:
+        writer.release()
     return 'output.mp4'
 
 iface = gr.Interface(fn=doo, inputs="video", outputs="video")