Update engine.py

engine.py

@@ -67,6 +67,59 @@ class Engine:
 
         logger.info("✅ FacePoke Engine initialized successfully.")
 
+    async def process_video(self, video, params):
+        cap = cv2.VideoCapture(video_path)
+        video_writer = None
+        frames = []
+        
+        while True:
+            ret, frame = cap.read()
+            
+            if not ret:
+                break
+            
+            if video_writer is None:
+                height, width, _ = frame.shape
+                output_file = "output_video.mp4"
+                fourcc = cv2.VideoWriter_fourcc(*'mp4v')
+                video_writer = cv2.VideoWriter(output_file, fourcc, 24.0, (width, height))
+            
+            frames.append(frame)
+        
+        for frame in loader(frames):
+            image = Image.fromarray(frame)
+            image = image.convert('RGB')
+
+            img_rgb = np.array(image)
+
+            inference_cfg = self.live_portrait.live_portrait_wrapper.cfg
+            img_rgb = await asyncio.to_thread(resize_to_limit, img_rgb, inference_cfg.ref_max_shape, inference_cfg.ref_shape_n)
+            crop_info = await asyncio.to_thread(self.live_portrait.cropper.crop_single_image, img_rgb)
+            img_crop_256x256 = crop_info['img_crop_256x256']
+
+            I_s = await asyncio.to_thread(self.live_portrait.live_portrait_wrapper.prepare_source, img_crop_256x256)
+            x_s_info = await asyncio.to_thread(self.live_portrait.live_portrait_wrapper.get_kp_info, I_s)
+            f_s = await asyncio.to_thread(self.live_portrait.live_portrait_wrapper.extract_feature_3d, I_s)
+            x_s = await asyncio.to_thread(self.live_portrait.live_portrait_wrapper.transform_keypoint, x_s_info)
+
+            processed_data = {
+                'img_rgb': img_rgb,
+                'crop_info': crop_info,
+                'x_s_info': x_s_info,
+                'f_s': f_s,
+                'x_s': x_s,
+                'inference_cfg': inference_cfg
+            }
+            
+            _, frame = await self.transform_frame(processed_data,  params)
+            bgr_frame = cv2.cvtColor(np.array(frame), cv2.COLOR_BGR2RGB)
+            video_writer.write(cv2.cvtColor(bgr_frame, cv2.COLOR_BGR2RGB))
+            
+        video_writer.release()
+        cap.release()
+
+        return output_file
+
     async def load_frames(self, frames):
       uid = str(uuid.uuid4())
       for frame in loader(frames):
@@ -397,4 +450,4 @@ class Engine:
             return [buffered.getvalue(), result_image]
 
         except Exception as e:
-            raise ValueError(f"Failed to modify image: {str(e)}")
+            raise ValueError(f"Failed to modify image: {str(e)}")