prepis.py

import whisper
import argparse
import os
from datetime import timedelta

def format_timestamp(seconds: float) -> str:
    return str(timedelta(seconds=int(seconds)))

def transcribe_file(input_file: str, model_name: str, output_file: str):
    print(f"🧠 Načítám model '{model_name}'...")
    model = whisper.load_model(model_name)

    print(f"🎧 Přepisuji soubor: {input_file}")
    result = model.transcribe(input_file, language="cs", verbose=False)

    segments = result.get("segments", [])
    formatted = []

    for seg in segments:
        start = format_timestamp(seg["start"])
        end = format_timestamp(seg["end"])
        text = seg["text"].strip()
        line = f"[{start} - {end}] {text}"
        print(line)
        formatted.append(line)

    with open(output_file, "w", encoding="utf-8") as f:
        f.write("\n".join(formatted))

    print(f"\n✅ Hotovo. Přepis uložen do souboru: {output_file}")

if __name__ == "__main__":
    parser = argparse.ArgumentParser(description="Přepis audio/video souboru pomocí Whisper.")
    parser.add_argument("input", nargs="?", help="Cesta k audio/video souboru (např. schuzka.mp4)")
    parser.add_argument("-m", "--model", default="medium", help="Název Whisper modelu (tiny, base, small, medium, large)")
    parser.add_argument("-o", "--output", default="prepis.txt", help="Výstupní soubor (default: prepis.txt)")

    args = parser.parse_args()

    if not args.input:
        args.input = input("Zadej cestu k audio/video souboru: ").strip()

    if not os.path.exists(args.input):
        print("❌ Soubor neexistuje.")
        exit(1)

    transcribe_file(args.input, args.model, args.output)