| |
|
| | """
|
| | Extract Source Images from Video Files
|
| |
|
| | This utility extracts source images that were embedded in video files
|
| | generated by Wan2GP with source image embedding enabled.
|
| | Supports MKV (attachments) and MP4 (cover art) formats.
|
| | """
|
| |
|
| | import sys
|
| | import os
|
| | import argparse
|
| | import glob
|
| | from shared.utils.audio_video import extract_source_images
|
| |
|
| |
|
| | def main():
|
| | parser = argparse.ArgumentParser(
|
| | description="Extract source images from video files with embedded images (MKV attachments or MP4 cover art)",
|
| | formatter_class=argparse.RawDescriptionHelpFormatter,
|
| | epilog=__doc__
|
| | )
|
| |
|
| | parser.add_argument(
|
| | 'video_files',
|
| | nargs='+',
|
| | help='Video file(s) to extract source images from (MKV or MP4)'
|
| | )
|
| |
|
| | parser.add_argument(
|
| | '-o', '--output-dir',
|
| | help='Output directory for extracted images (default: same as video directory)'
|
| | )
|
| |
|
| | parser.add_argument(
|
| | '-v', '--verbose',
|
| | action='store_true',
|
| | help='Enable verbose output'
|
| | )
|
| |
|
| | args = parser.parse_args()
|
| |
|
| |
|
| | video_files = []
|
| | for pattern in args.video_files:
|
| | if '*' in pattern or '?' in pattern:
|
| | video_files.extend(glob.glob(pattern))
|
| | else:
|
| | video_files.append(pattern)
|
| |
|
| | if not video_files:
|
| | print("No video files found matching the specified patterns.")
|
| | return 1
|
| |
|
| | total_extracted = 0
|
| |
|
| | for video_file in video_files:
|
| | if not os.path.exists(video_file):
|
| | print(f"Warning: File not found: {video_file}")
|
| | continue
|
| |
|
| | if not (video_file.lower().endswith('.mkv') or video_file.lower().endswith('.mp4')):
|
| | print(f"Warning: Skipping unsupported file: {video_file} (only MKV and MP4 supported)")
|
| | continue
|
| |
|
| | if args.verbose:
|
| | print(f"\nProcessing: {video_file}")
|
| |
|
| |
|
| | if args.output_dir:
|
| | output_dir = args.output_dir
|
| | else:
|
| |
|
| | video_dir = os.path.dirname(video_file) or '.'
|
| | video_name = os.path.splitext(os.path.basename(video_file))[0]
|
| | output_dir = os.path.join(video_dir, f"{video_name}_sources")
|
| |
|
| | try:
|
| | extracted_files = extract_source_images(video_file, output_dir)
|
| |
|
| | if extracted_files:
|
| | print(f"✓ Extracted {len(extracted_files)} source image(s) from {video_file}")
|
| | if args.verbose:
|
| | for img_file in extracted_files:
|
| | print(f" → {img_file}")
|
| | total_extracted += len(extracted_files)
|
| | else:
|
| | print(f"ℹ No source images found in {video_file}")
|
| |
|
| | except Exception as e:
|
| | print(f"✗ Error processing {video_file}: {e}")
|
| |
|
| | print(f"\nTotal: Extracted {total_extracted} source image(s) from {len(video_files)} video file(s)")
|
| | return 0
|
| |
|
| |
|
| | if __name__ == "__main__":
|
| | sys.exit(main())
|
| |
|
