import argparse
import re
from paddle import utils

def is_valid_url(url):
    regex = re.compile(
    r'^(?:http|ftp|https)s?://',
    re.IGNORECASE,
    )
    return re.match(regex, url) is not None

def is_valid_path(path):
    return path.startswith("/tmp")

def main():
    parser = argparse.ArgumentParser(description='Download AI Model.')
    parser.add_argument('url', type=str, help='URL to download from')
    parser.add_argument('path', type=str, help='Path to save the file, must start with /tmp')
    args parser.parse_args()
    if not is_valid_url(args.url):
        raise ValueError("The URL is not valid.")
    if not is_valid_path(args.path):
        raise ValueError("The path must start with /tmp.")
    utils.download._wget_download(args.url, args.path)

if __name__ == '__main__':
    main()