def is_image_file(filename):
    image_extensions = [".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp"]
    # 获取文件扩展名并转换为小写
    file_extension = filename.split(".")[-1].lower()
    return file_extension in image_extensions


def is_rtsp_or_video(source):
    if source.startswith("rtsp://"):
        return True

    video_exts = ["mp4", "avi"]
    file_extension = source.split(".")[-1].lower()
    return file_extension in video_exts