import logging
from pathlib import Path
from src.runner import run
from src.config import presets


def main():
    import argparse

    logging.basicConfig(
        level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s"
    )

    parser = argparse.ArgumentParser()
    parser.add_argument("-b", "--base_path", help="Base path", default="output")
    parser.add_argument(
        "-v", "--video_path", help="Video path", default="example/video.mp4"
    )
    parser.add_argument(
        "-o",
        "--output",
        help="Output video name (example: 'interpolated_video.mp4')",
        default="interpolated_video.mp4",
    )
    parser.add_argument(
        "-p",
        "--preset",
        help="Model preset",
        choices=["small", "large", "global"],
        default="global",
    )
    args = parser.parse_args()
    run(
        base_path=Path(args.base_path),
        video_path=Path(args.video_path),
        output_video=args.output,
        preset=getattr(presets, args.preset.upper()),
    )


if __name__ == "__main__":
    main()