bigsub.py

import os
import math

def time_to_ms(time_str):
    """Конвертирует время формата MM:SS в миллисекунды"""
    try:
        parts = list(map(int, time_str.split(':')))
        
        if len(parts) == 2:  # MM:SS
            minutes, seconds = parts
            return (minutes * 60 + seconds) * 1000
        elif len(parts) == 3:  # HH:MM:SS
            hours, minutes, seconds = parts
            return (hours * 3600 + minutes * 60 + seconds) * 1000
        else:
            raise ValueError("Неверный формат времени")
    except:
        return 120000  # 2 минуты по умолчанию

def ms_to_srt_time(ms):
    """Конвертирует миллисекунды в SRT формат: HH:MM:SS,mmm"""
    # Корректировка как у автора (-1 мс для ровного времени)
    if ms % 1000 == 0:
        ms -= 1
    
    hours = ms // 3600000
    ms %= 3600000
    minutes = ms // 60000
    ms %= 60000
    seconds = ms // 1000
    milliseconds = ms % 1000
    
    return f"{hours:02d}:{minutes:02d}:{seconds:02d},{milliseconds:03d}"

def generate_youtube_drawing_srt():
    """Генерирует SRT для рисования на YouTube с эмодзи карандаша"""
    print("=" * 60)
    print("🎨 YOUTUBE DRAWING SRT GENERATOR ✐")
    print("   (Эмодзи карандаша в первые 15 секунд)")
    print("=" * 60)
    
    # Константы
    STEP_MS = 100            # Шаг 100 мс
    PENCIL_EMOJI = "✐"      # Эмодзи карандаша U+2710
    DOT_SYMBOL = "ㆍ"        # Символ для точек
    
    print("\n⚙️  ПАРАМЕТРЫ:")
    
    # 1. Длительность видео
    while True:
        duration_input = input("Введите длительность видео (MM:SS, например 02:00): ").strip()
        if ':' in duration_input:
            total_ms = time_to_ms(duration_input)
            if total_ms >= 15000:  # Минимум 15 секунд для эмодзи
                break
            else:
                print("⚠️  Минимум 15 секунд для отображения эмодзи!")
        else:
            print("⚠️  Используйте формат MM:SS")
    
    # 2. Начало точек
    while True:
        start_input = input("Начало точек в секундах (рекомендуется 5): ").strip()
        if start_input.replace('.', '').isdigit():
            start_sec = float(start_input)
            START_MS = int(start_sec * 1000)
            if START_MS >= 0 and START_MS < total_ms:
                break
        print("⚠️  Введите число (например 5)")
    
    # 3. Эмодзи карандаша (первые 15 секунд)
    PENCIL_DURATION = 15000  # 15 секунд
    pencil_start = 0
    pencil_end = min(PENCIL_DURATION, total_ms)
    
    # Рассчитываем
    end_time = ms_to_srt_time(total_ms)
    num_dots = (total_ms - START_MS) // STEP_MS
    
    print("\n📊 РАСЧЕТ:")
    print(f"  • Длительность видео: {total_ms/1000:.1f} сек")
    print(f"  • Эмодзи карандаша: {pencil_start}-{pencil_end/1000:.1f} сек")
    print(f"  • Начало точек: {START_MS/1000:.1f} сек")
    print(f"  • Шаг точек: {STEP_MS} мс")
    print(f"  • Всего точек: {num_dots:,}")
    print(f"  • Эмодзи: '{PENCIL_EMOJI}'")
    print(f"  • Символ точек: '{DOT_SYMBOL}'")
    
    # Создаем SRT контент
    srt_lines = []
    
    print("\n🔄 Генерация SRT файла...")
    
    # 1. Эмодзи карандаша (первые 15 секунд)
    srt_lines.append("1\n")
    srt_lines.append(f"{ms_to_srt_time(pencil_start)} --> {ms_to_srt_time(pencil_end)}\n")
    srt_lines.append(f"{PENCIL_EMOJI}\n")
    srt_lines.append("<div></div>\n\n")
    
    # 2. Генерация точек с шагом 100мс
    index = 2
    current_ms = START_MS
    
    # Если начало точек раньше конца эмодзи, корректируем
    if current_ms < pencil_end:
        current_ms = pencil_end
    
    dots_generated = 0
    
    while current_ms < total_ms:
        start_time = ms_to_srt_time(current_ms)
        
        srt_lines.append(f"{index}\n")
        srt_lines.append(f"{start_time} --> {end_time}\n")
        srt_lines.append(f"{DOT_SYMBOL}\n")
        srt_lines.append("<div></div>\n\n")
        
        current_ms += STEP_MS
        index += 1
        dots_generated += 1
        
        # Прогресс
        if dots_generated % 500 == 0:
            progress = (current_ms - START_MS) / (total_ms - START_MS) * 100
            print(f"  Сгенерировано точек: {dots_generated:,} ({progress:.1f}%)")
    
    # 3. Добавляем завершающий эмодзи в конце (если есть место)
    if total_ms - current_ms > 2000:  # Если осталось больше 2 секунд
        final_start = max(current_ms, total_ms - 5000)  # Последние 5 секунд
        if final_start < total_ms:
            srt_lines.append(f"{index}\n")
            srt_lines.append(f"{ms_to_srt_time(final_start)} --> {end_time}\n")
            srt_lines.append(f"{PENCIL_EMOJI}\n")
            srt_lines.append("<div></div>\n\n")
            index += 1
            print(f"  Добавлен завершающий эмодзи в конце")
    
    # Сохранение файла
    print("\n💾 СОХРАНЕНИЕ ФАЙЛА...")
    
    default_name = f"youtube_drawing_{total_ms//1000}sec"
    filename = input(f"Имя файла (без .srt, Enter для '{default_name}'): ").strip()
    if not filename:
        filename = default_name
    
    filename = filename.replace(".srt", "") + ".srt"
    
    desktop = os.path.join(os.path.expanduser("~"), "Desktop")
    save_path = os.path.join(desktop, filename)
    
    try:
        with open(save_path, "w", encoding="utf-8") as f:
            f.writelines(srt_lines)
        
        file_size = os.path.getsize(save_path) / 1024
        
        print("\n" + "=" * 60)
        print("✅ SRT ФАЙЛ УСПЕШНО СОЗДАН! ✐")
        print("=" * 60)
        print(f"📂 Файл: {save_path}")
        print(f"📊 Размер: {file_size:.1f} KB")
        print(f"🔢 Всего записей: {index - 1}")
        print(f"  • Эмодзи карандаша: 1 запись")
        print(f"  • Точек: {dots_generated:,} записей")
        
        print("\n📋 СТРУКТУРА ФАЙЛА:")
        print("-" * 40)
        
        # Показываем начало файла
        preview_count = min(5, len(srt_lines) // 4)
        for i in range(preview_count):
            start_idx = i * 4
            if start_idx + 3 < len(srt_lines):
                print(f"Запись {i+1}:")
                print(f"  Время: {srt_lines[start_idx+1].strip()}")
                print(f"  Текст: {srt_lines[start_idx+2].strip()}")
                print()
        
        if len(srt_lines) > 20:
            print(f"...\n[и еще {len(srt_lines)//4 - preview_count} записей]\n")
        
        print("\n🎬 КАК ИСПОЛЬЗОВАТЬ В YOUTUBE:")
        print("1. Загрузите видео на YouTube")
        print("2. В Творческой студии → Субтитры → Добавить")
        print("3. Загрузите файл:", filename)
        print("4. Включите субтитры в плеере")
        print("5. Двигайте ползунок - увидите:")
        print("   • ✐ в первые 15 секунд")
        print("   • ••• точки при движении ползунка")
        
        print("\n⚠️  РЕКОМЕНДАЦИИ:")
        print("   • Оптимально: 60-90 секунд видео")
        print("   • YouTube может тормозить при >10,000 субтитров")
        print("   • Эмодзи ✐ будет виден в начале видео")
        
    except Exception as e:
        print(f"\n❌ Ошибка при сохранении: {e}")
        return None
    
    return save_path

def generate_quick_example():
    """Быстрый пример: 1:40, эмодзи ✐ 15 сек, точки с 5 сек"""
    print("\n🧪 БЫСТРЫЙ ПРИМЕР (01:40)")
    print("-" * 40)
    
    total_ms = 100000  # 1:40
    start_ms = 5000    # 5 сек
    step_ms = 100      # 100 мс
    pencil_emoji = "✐"
    dot_symbol = "ㆍ"
    pencil_duration = 15000  # 15 сек
    
    srt_lines = []
    
    # 1. Эмодзи карандаша
    srt_lines.append("1\n")
    srt_lines.append(f"00:00:00,000 --> 00:00:14,999\n")
    srt_lines.append(f"{pencil_emoji}\n")
    srt_lines.append("<div></div>\n\n")
    
    # 2. Точки (начинаем после эмодзи или с 5 сек)
    index = 2
    current_ms = max(start_ms, pencil_duration)
    
    dots_count = 0
    while current_ms < total_ms:
        start_str = ms_to_srt_time(current_ms)
        srt_lines.append(f"{index}\n")
        srt_lines.append(f"{start_str} --> 00:01:39,999\n")
        srt_lines.append(f"{dot_symbol}\n")
        srt_lines.append("<div></div>\n\n")
        
        current_ms += step_ms
        index += 1
        dots_count += 1
    
    desktop = os.path.join(os.path.expanduser("~"), "Desktop")
    example_file = os.path.join(desktop, "youtube_example_140.srt")
    
    with open(example_file, "w", encoding="utf-8") as f:
        f.writelines(srt_lines)
    
    print(f"📁 Создан файл: {example_file}")
    print(f"📊 Записей: {index-1}")
    print(f"  • Эмодзи ✐: 15 секунд")
    print(f"  • Точек: {dots_count:,} (шаг 100мс)")
    print(f"⏱️  Длительность: 1:40")
    print("\n✅ Готово! Файл на Рабочем столе.")

if __name__ == "__main__":
    print("YOUTUBE DRAWING SRT GENERATOR ✐")
    print("=" * 40)
    print("Создает SRT для 'рисования' точками на YouTube")
    print("с эмодзи карандаша в начале видео.")
    print()
    
    print("Выберите вариант:")
    print("1. 🎨 Настроить все параметры")
    print("2. 🧪 Быстрый пример (01:40, ✐ 15 сек, точки с 5 сек)")
    print("3. ⚡ Максимальный (02:00)")
    
    choice = input("\nВведите номер: ").strip()
    
    if choice == "1":
        generate_youtube_drawing_srt()
    elif choice == "2":
        generate_quick_example()
    elif choice == "3":
        # Быстрая генерация на 2 минуты
        print("\n⚡ Генерация на 2 минуты...")
        # Просто запускаем с параметрами по умолчанию
        import sys
        sys.argv = [sys.argv[0], "02:00"]
        generate_youtube_drawing_srt()
    else:
        print("Используется вариант 2")
        generate_quick_example()
    
    input("\nНажмите Enter для выхода...")