// Copyright (c) 2026, CIIL Anuvani-LID. Apache-2.0.
// Binding test (see tests/bindings/python_ref.py): the Java API, on a desktop JVM against the Linux build of
// libsherpa-onnx-jni.so.
import com.k2fsa.sherpa.onnx.*;
import java.io.File;

public class Main {
    public static void main(String[] args) {
        String model = args[0], vad = args[1], longWav = args[2];

        AudioTaggingModelConfig modelConfig =
                AudioTaggingModelConfig.builder().setLid(model).setNumThreads(2).setDebug(false).build();
        AudioTagging tagger = new AudioTagging(
                AudioTaggingConfig.builder().setModel(modelConfig).setTopK(23).build());
        for (int i = 3; i < args.length; ++i) {
            WaveReader reader = new WaveReader(args[i]);
            OfflineStream stream = tagger.createStream();
            stream.acceptWaveform(reader.getSamples(), reader.getSampleRate());
            AudioEvent[] events = tagger.compute(stream);
            stream.release();
            StringBuilder sb = new StringBuilder();
            for (int k = 0; k < events.length; ++k) {
                if (k > 0) sb.append(",");
                sb.append("[\"").append(events[k].getName()).append("\",").append(events[k].getIndex())
                  .append(",").append(events[k].getProb()).append("]");
            }
            System.out.println("{\"file\":\"" + new File(args[i]).getName() + "\",\"events\":[" + sb + "]}");
        }
        tagger.release();

        LidPipeline pipeline = new LidPipeline(
                LidPipelineConfig.builder().setModel(modelConfig).setVadModel(vad).build());
        WaveReader reader = new WaveReader(longWav);
        String json = pipeline.process(reader.getSamples(), reader.getSampleRate());
        System.out.println("{\"pipeline_file\":\"" + new File(longWav).getName() + "\",\"pipeline\":" + json + "}");
        pipeline.release();
    }
}