import librosa
from scripts.predict import predict_multimodal, predict_unimodal


def predict_multimodal_runner(sample: str):
    # Load test audio and lyrics
    audio_path = f"data/external/{sample}.mp3"
    lyrics_path = f"data/external/{sample}.txt"

    # Load audio
    audio_data, sr = librosa.load(audio_path)

    # Load lyrics
    with open(lyrics_path, "r", encoding="utf-8") as f:
        lyrics_text = f.read()

    print("Running multimodal prediction pipeline...")
    prediction = predict_multimodal(audio_data, lyrics_text)

    print("\n=== MULTIMODAL PREDICTION RESULT ===")
    print(f"Prediction: {prediction}")


def predict_unimodal_runner(sample: str):
    # Load test audio
    audio_path = f"data/external/{sample}.mp3"

    # Load audio
    audio_data, sr = librosa.load(audio_path)

    print("Running audio-only prediction pipeline...")
    prediction = predict_unimodal(audio_data)

    print("\n=== AUDIO-ONLY PREDICTION RESULT ===")
    print(f"Prediction: {prediction}")


if __name__ == "__main__":
    sample = "sample"

    # Run both predictions
    predict_multimodal_runner(sample)

    print("\n" + "=" * 50 + "\n")

    predict_unimodal_runner(sample)