add draft title cards

plus title card dir

ami_scripts/prepend_title_cards.py

@@ -0,0 +1,219 @@
+#!/usr/bin/env python3
+
+import argparse
+import subprocess
+import json
+import os
+import re
+import glob
+
+def get_title_cards_for_group(prefix):
+    # Define the paths for the title card images
+    script_dir = os.path.dirname(os.path.realpath(__file__))
+    title_card_dir = os.path.join(os.path.dirname(script_dir), 'title_cards')
+
+    nypl_title_card = os.path.join(title_card_dir, 'Title_Card_NYPL.png')
+    schomburg_title_card = os.path.join(title_card_dir, 'Title_Card_Schomburg.png')
+    lpa_title_card = os.path.join(title_card_dir, 'Title_Card_LPA.png')
+
+    # Group logic
+    if prefix in ['scb', 'scd']:
+        return [nypl_title_card, schomburg_title_card]
+    elif prefix in ['myd', 'myt', 'mym', 'myh']:
+        return [nypl_title_card, lpa_title_card]
+    elif prefix == 'mao':
+        return [nypl_title_card]
+    else:
+        return []
+
+
+def process_video(video_path, asset_flag):
+
+    # Extract prefix from filename
+    filename = os.path.basename(video_path)
+    prefix = filename.split('_')[0]
+
+    # Get the appropriate title cards based on the group
+    title_cards = get_title_cards_for_group(prefix)
+
+    # Get info about the video file
+    ffmpeg_probe_cmd = ['ffprobe', '-v', 'quiet', '-print_format', 'json', '-show_format', '-show_streams', video_path]
+    video_info_json = subprocess.check_output(ffmpeg_probe_cmd)
+    video_info = json.loads(video_info_json)
+
+    video_streams = [stream for stream in video_info['streams'] if stream['codec_type'] == 'video']
+    assert len(video_streams) > 0, "No video streams found in the input video file."
+    video_stream = video_streams[0]
+
+    audio_streams = [stream for stream in video_info['streams'] if stream['codec_type'] == 'audio']
+
+    # Get the video specs
+    width = video_stream['width']
+    height = video_stream['height']
+    pixel_format = video_stream['pix_fmt']
+    video_codec = video_stream['codec_name']
+    frame_rate = video_stream['avg_frame_rate']
+    interlaced = video_stream['field_order'] in ['tb', 'bt']
+    display_aspect_ratio = video_stream['display_aspect_ratio']
+
+    # Prepare output file path
+    base_name, extension = os.path.splitext(video_path)
+    output_file = base_name + '_with_title' + extension
+
+    image_video_files = []
+
+    for idx, image in enumerate(title_cards):
+        # Create a temp file for the image video
+        image_video_file_name = f"{base_name}_temp_{idx}.mp4"
+        image_video_files.append(image_video_file_name)
+
+        # Create silent video
+        filters = [
+            f"fade=t=in:st=0:d=1",
+            f"fade=t=out:st=4:d=1:color=black",
+            f"format={pixel_format}"
+        ]
+
+        # Check if the video is HD and pillarboxed
+        if width == 1920 and height == 1080:
+            # Calculate new dimensions for 4:3 aspect ratio within 1920x1080
+            new_width = 1440  # 1080p height with 4:3 aspect ratio
+            new_height = 1080
+            pad_x = (1920 - new_width) // 2
+            pad_y = 0
+
+            # Scale and pad the image to fit 1920x1080 without stretching
+            filters.extend([
+                f"scale={new_width}:{new_height}",
+                f"pad=1920:1080:{pad_x}:{pad_y}:black"
+            ])
+        elif width == 720 and height == 480:
+            # Check if the video is SD widescreen (16:9 aspect ratio)
+            if video_stream.get('display_aspect_ratio') == '16:9':
+                # Calculate new dimensions for 4:3 aspect ratio within 720x480
+                new_width_4_3 = int(480 * (4 / 3))  # 640 for 4:3 aspect ratio
+                pad_x = (720 - new_width_4_3) // 2  # Padding for left and right
+
+                # Scale and pad the image to fit 720x480 without stretching
+                filters.extend([
+                    f"scale={new_width_4_3}:{height}",
+                    f"pad=720:480:{pad_x}:0:black",
+                    "setdar=16/9"
+                ])
+            else:
+                # For other 720x480 videos, scale as usual
+                filters.extend([
+                    f"scale={width}:{height}",
+                    "setsar=1",
+                    "setdar=4/3"
+                ])
+        else:
+            # For other resolutions, scale as usual
+            filters.extend([
+                f"scale={width}:{height}",
+                "setsar=1",
+                "setdar=4/3"
+            ])
+
+        ffmpeg_image_to_video_cmd = [
+            'ffmpeg', '-y', '-loop', '1', '-i', image, '-t', '5', '-c:v', video_codec, '-r', frame_rate, 
+            '-vf', ','.join(filters)
+        ]
+
+        if interlaced:
+            ffmpeg_image_to_video_cmd += ['-flags', '+ilme+ildct']
+
+        ffmpeg_image_to_video_cmd.append(image_video_file_name)
+        print(ffmpeg_image_to_video_cmd)
+        subprocess.check_output(ffmpeg_image_to_video_cmd)
+
+        if audio_streams:  # Add silent audio to each title card video
+            audio_stream = audio_streams[0]
+            audio_codec = audio_stream['codec_name']
+            channel_layout = audio_stream.get('channel_layout', 'stereo')
+            sample_rate = audio_stream.get('sample_rate', '48000')
+            bitrate = audio_stream.get('bit_rate', '320k')
+
+            ffmpeg_add_audio_cmd = [
+                'ffmpeg', '-i', image_video_file_name, '-f', 'lavfi', 
+                '-t', '5', '-i', f'anullsrc=channel_layout={channel_layout}:sample_rate={sample_rate}',
+                '-c:v', 'copy', '-c:a', audio_codec, '-b:a', bitrate, '-shortest', f"{base_name}_temp_with_audio_{idx}.mp4"
+            ]
+
+            print(ffmpeg_add_audio_cmd)
+            subprocess.check_output(ffmpeg_add_audio_cmd)
+            os.rename(f"{base_name}_temp_with_audio_{idx}.mp4", image_video_file_name)
+
+    # Create a list file for concat
+    concat_list = 'concat_list.txt'
+    with open(concat_list, 'w') as f:
+        for file in image_video_files:
+            f.write(f"file '{file}'\n")
+        f.write(f"file '{video_path}'\n")
+
+    # Concatenate the title card videos and the original video
+    temp_output_path = os.path.join(os.path.dirname(video_path), 'temp_concat_output.mp4')
+    ffmpeg_concat_cmd = ['ffmpeg', '-y', '-f', 'concat', '-safe', '0', '-i', concat_list, '-c', 'copy', temp_output_path]
+    subprocess.check_output(ffmpeg_concat_cmd)
+
+    # Apply text and timecode to the main video after title cards
+    if asset_flag:
+        title_cards_duration = len(title_cards) * 5  # Assuming each title card is 5 seconds
+        output_path = os.path.join(os.path.dirname(video_path), base_name + '_with_title' + extension)
+
+        # Extract asset ID
+        match = re.search(r'_(\d{6})_', video_path)
+        asset_id = match.group(1) if match else ''
+
+        # Determine bitrate based on resolution
+        if width == 1920 and height == 1080:  # HD content
+            video_bitrate = "8000000"
+            bufsize = "8000000"
+            maxrate = "8000000"
+        else:  # SD content
+            video_bitrate = "3500000"
+            bufsize = "1750000"
+            maxrate = "3500000"
+
+        ffmpeg_drawtext_cmd = [
+            'ffmpeg', '-i', temp_output_path, '-c:v', 'libx264', '-b:v', video_bitrate, '-bufsize', bufsize, '-maxrate', maxrate, '-vf',
+            f"drawtext=fontfile=/System/Library/Fonts/Helvetica.ttc:fontsize=25:text='{asset_id}':x=10:y=10:fontcolor=white:enable='gte(t,{title_cards_duration})'," +
+            f"drawtext=fontfile=/System/Library/Fonts/Helvetica.ttc:fontsize=20:text='%{{pts\\:hms\\: - {title_cards_duration}}}':box=1:[email protected]:boxborderw=5:x=(w-tw)/2:y=h-th-10:fontcolor=white:enable='gte(t,{title_cards_duration})',format=yuv420p",
+            '-c:a', 'copy', output_path
+        ]
+        subprocess.check_output(ffmpeg_drawtext_cmd)
+        os.remove(temp_output_path)
+    else:
+        output_path = base_name + '_with_title' + extension
+        os.rename(temp_output_path, output_path)
+
+    # Remove temporary files
+    for file in image_video_files:
+        try:
+            os.remove(file)
+        except OSError as e:
+            print(f"Error: {file} : {e.strerror}")
+
+    os.remove(concat_list)
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Prepend title card to a video.")
+    parser.add_argument('-v', '--video', help='Path to the video file.')
+    parser.add_argument('-d', '--directory', help='Path to the directory containing video files.')
+    parser.add_argument('-a', '--asset', action='store_true', help='Extract and add asset ID from the filename.')
+    args = parser.parse_args()
+
+    if args.directory:
+        # Process each video file in the directory
+        for video_file in sorted(glob.glob(os.path.join(args.directory, '*.mp4'))):
+            process_video(video_file, args.asset)
+    elif args.video:
+        # Process a single video file
+        process_video(args.video, args.asset)
+    else:
+        print("Error: No video file or directory specified.")
+        parser.print_help()
+
+if __name__ == "__main__":
+    main()