import os

os.environ["CUDA_VISIBLE_DEVICES"] = "4"

import json
from suno_utils.audio import Audio
from suno_utils.utils.text import read_jsonl
from suno_utils.utils.s3 import list_s3_dir, read_from_s3

import torch
import numpy as np

from suno_utils.diffusion.generation import (
    preload_dit_model,
    preload_tokenizer,
    TOKENIZER_FILEPATH,
    SEMANTIC_MODEL_FILEPATH,
    SEMANTIC_CLUSTERS_FILEPATH,
    _retrieve_models,
)

from suno_utils.tasks.mert_25 import (
    preload_models as preload_semantic_models,
    encode as encode_semantic,
)

from suno_utils.tasks.dac_vae_fixed_25hz import (
    preload_models as preload_codec_models,
    decode as codec_decode,
    decode_stream_to_full_audio,
)

from suno_utils.diffusion import generation as diffusion_gen
from suno_utils.tasks.upsample_engine_old import UpsampleEngine, Request, Job

def main()