add extract_frames_with_labels function

app.py

@@ -14,6 +14,42 @@ hf_hub_download(
     local_dir="checkpoint"
 )
 
+def extract_frames_with_labels(video_path, output_dir="frames"):
+    # Ensure output directory exists
+    os.makedirs(output_dir, exist_ok=True)
+    
+    # Open the video file
+    video_capture = cv2.VideoCapture(video_path)
+    if not video_capture.isOpened():
+        raise ValueError(f"Cannot open video file: {video_path}")
+    
+    frame_data = []
+    frame_index = 0
+    
+    # Loop through the video frames
+    while True:
+        ret, frame = video_capture.read()
+        if not ret:
+            break  # Exit the loop if there are no frames left to read
+
+        # Zero-padded frame index for filename and label
+        frame_label = f"{frame_index:04}"
+        frame_filename = os.path.join(output_dir, f"frame_{frame_label}.jpg")
+        
+        # Save the frame as a .jpg file
+        cv2.imwrite(frame_filename, frame)
+        
+        # Append the tuple (filename, label) to the list
+        frame_data.append((frame_filename, frame_label))
+        
+        # Increment frame index
+        frame_index += 1
+    
+    # Release the video capture object
+    video_capture.release()
+    
+    return frame_data
+
 # Define a function to run your script with selected inputs
 def run_xportrait(
     model_config,