import torch, joblib, json
from speechbrain.pretrained import EncoderClassifier
from helper_classes import Config, AudioProcessor, LanguageIdentifier

# Load classifier + config
custom_classifier = joblib.load("custom_classifier.pkl")
label_map = json.load(open("label_map.json"))

config = Config()
config.label_map = label_map

identifier = LanguageIdentifier(config)
identifier.load_vox_model()
identifier.custom_classifier = custom_classifier

def predict(audio_path: str):
    return identifier.predict(audio_path)