srt_to_smpte.py

#!/usr/bin/env python3
import sys
import os
from moviepy.editor import VideoFileClip
import pandas as pd
import re
from datetime import datetime, timedelta

# Convert a time string into a timedelta object
def str_to_timedelta(time_str):
    return datetime.strptime(time_str, '%H:%M:%S,%f') - datetime(1900, 1, 1)

# Convert a timedelta object to SMPTE timecode format
def timedelta_to_smpte_timecode(t_delta, fps):
    total_seconds = int(t_delta.total_seconds())
    hours, remainder = divmod(total_seconds, 3600)
    minutes, seconds = divmod(remainder, 60)
    frames = int(((t_delta.total_seconds() - total_seconds) * fps))
    return f"{hours:02}:{minutes:02}:{seconds:02}:{frames:02}"

# Extract the frame rate from the given video file
def get_frame_rate(video_file_path):
    try:
        with VideoFileClip(video_file_path) as clip:
            return clip.fps
    except Exception as e:
        print(f"Error extracting frame rate from video: {e}")
        sys.exit(1)

# Convert SRT file to CSV format
def convert_srt_to_csv(srt_file_path, video_file_path):
    if not os.path.exists(srt_file_path):
        print(f"The SRT file does not exist: {srt_file_path}")
        sys.exit(1)
    if not os.path.exists(video_file_path):
        print(f"The video file does not exist: {video_file_path}")
        sys.exit(1)

    fps = get_frame_rate(video_file_path)
    csv_file_path = srt_file_path.replace('.srt', '.csv')

    try:
        with open(srt_file_path, 'r', encoding='utf-8') as file:
            srt_content = file.read()
    except Exception as e:
        print(f"Error reading SRT file: {e}")
        sys.exit(1)

    try:
        entries = re.split(r'\n\n', srt_content.strip())
        csv_data = []
        for entry in entries:
            lines = entry.split('\n')
            if len(lines) >= 3:
                sequence = lines[0]
                times = re.split(r' --> ', lines[1])
                start_time = str_to_timedelta(times[0])
                end_time = str_to_timedelta(times[1])
                text = ' '.join(lines[2:]).replace('\n', ' ')
                csv_data.append({
                    'Line Number': sequence,
                    'Timecode IN': timedelta_to_smpte_timecode(start_time, fps),
                    'Timecode OUT': timedelta_to_smpte_timecode(end_time, fps),
                    'Text': text
                })
        pd.DataFrame(csv_data).to_csv(csv_file_path, index=False, sep=',', header=True)
        print(f"Converted SRT file saved to {csv_file_path}")
        print(f"The frame rate of the video is: {fps} fps")
    except Exception as e:
        print(f"Error processing SRT file: {e}")
        sys.exit(1)

# Get user input with a prompt
def get_user_input(prompt):
    return input(prompt)

# Main script execution
if __name__ == "__main__":
    if len(sys.argv) == 3:
        srt_file_path = sys.argv[1]
        video_file_path = sys.argv[2]
    else:
        print("You did not provide the required SRT and video file paths.")
        srt_file_path = get_user_input("Please enter the full path to the SRT file: ")
        video_file_path = get_user_input("Please enter the full path to the video file: ")

    convert_srt_to_csv(srt_file_path, video_file_path)