Function main

ScriptAgent/code/director_agent.py:2161–2439

()

def main() -> None:
    import time
    parser = build_argument_parser()
    args = parser.parse_args()
    start_time = time.time()
    logging.basicConfig(
        level=getattr(logging, args.log_level.upper(), logging.INFO),
        format="[%(asctime)s][%(levelname)5s] %(message)s",
    )

    ensure_dependencies()
    
    # Apply model default configuration
    if args.model in MODEL_DEFAULT_CONFIG:
        model_config = MODEL_DEFAULT_CONFIG[args.model]
        # If user hasn't explicitly specified, use model default configuration
        if args.seconds is None:
            args.seconds = model_config["seconds"]
            LOGGER.info(f"Using {args.model} model default duration: {args.seconds} seconds")
        if args.size is None:
            args.size = model_config["size"]
            LOGGER.info(f"Using {args.model} model default resolution: {args.size}")
    else:
        # For models not in MODEL_DEFAULT_CONFIG, use general default values
        if args.seconds is None:
            args.seconds = 12
        if args.size is None:
            args.size = "1792x1024"

    # ✅ Separate parameter checks for batch mode and single script mode
    if args.enable_batch:
        # Batch mode: only needs batch_jsonl, doesn't need script/script_path
        if not args.batch_jsonl:
            raise ValueError("Batch mode requires specifying --batch_jsonl file path.")
        LOGGER.info("Batch mode: Will read scripts from JSONL file")
    else:
        # Single script mode: needs script or script_path
        if not args.script and not args.script_path:
            raise ValueError("Single script mode requires providing either --script or --script_path.")

        script_text = args.script
        if not script_text and args.script_path:
            with open(args.script_path, "r", encoding="utf-8") as file:
                script_text = file.read()

        if not script_text:
            raise ValueError("Read script text is empty.")

        components = extract_story_components(script_text)
        if not components.nodes:
            raise ValueError("Failed to parse valid nodes, please check if script text format is 'number. content'.")
        
        if args.resume and not args.resume_node_dir:
            raise ValueError("When enabling --resume, must specify --resume_node_dir")
        
        if args.resume and args.resume_node_dir:
            if not os.path.exists(args.resume_node_dir):
                raise FileNotFoundError(f"Specified node directory does not exist: {args.resume_node_dir}")