source: gs3-extensions/atea-nlp-tools/trunk/src/koreromaori-proxy/src/main/java/org/atea/nlptools/koreromaoriinterface/services/ReoTuhituhiApiService.java@ 35241

package org.atea.nlptools.koreromaoriinterface.services;

import java.io.InputStream;
import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;
import java.util.concurrent.Callable;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;

import com.google.gson.Gson;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.atea.nlptools.koreromaoriinterface.models.AudioFilePart;
import org.atea.nlptools.koreromaoriinterface.models.TranscriptionResult;

/**
 * Functions to interact with the Reo Tuhituhi API.
 */
public class ReoTuhituhiApiService
{
    private static final Logger logger = LogManager.getLogger(ReoTuhituhiApiService.class);

    private final ExecutorService threadPool;
    private final Gson jsonSerialiser;
    private final String apiEndpoint;
    private final String apiKey;

    public ReoTuhituhiApiService(Gson jsonSerialiser, String apiEndpoint, String apiKey)
    {
        this.jsonSerialiser = jsonSerialiser;
        this.apiEndpoint = apiEndpoint;
        this.apiKey = apiKey;

        threadPool = java.util.concurrent.Executors.newFixedThreadPool(3);
    }

    /**
     * Queries the Reo Tuhituhi API to transcribe the given audio files.
     * 
     * @param audioFileParts The audio files to retrieve a transcription for.
     * @return A list of {@link TranscriptionResult} objects.
     * @throws Exception
     */
    public List<AudioFilePart> GetTranscriptions(Iterable<AudioFilePart> audioFileParts)
        throws Exception
    {
        Queue<Future<AudioFilePart>> apiCalls = new LinkedList<Future<AudioFilePart>>();

        // Queue each transcription request up asynchronously
        for (AudioFilePart part : audioFileParts)
        {
            //Callable<TranscriptionResult> transcriptionTask = getTranscriptionCallable(audioStream);
            //apiCalls.add(threadPool.submit(transcriptionTask));
            //logger.debug("Adding transcription task to thread pool.");
        }

        List<AudioFilePart> apiResults = new ArrayList<AudioFilePart>(apiCalls.size());
        
        for (AudioFilePart part : audioFileParts)
        {
            TranscriptionResult res = getTranscription(part.dataStream);
            part.setTranscriptionResult(res);
            apiResults.add(part);
        }

        // Wait on the result of each call
        // TODO: Implement proper timeout here
        // while (!apiCalls.isEmpty())
        // {
        //     apiResults.add(apiCalls.remove().get());
        //     logger.debug("API call has completed.");
        // }

        return apiResults;
    }

    private Callable<AudioFilePart> getTranscriptionCallable(final AudioFilePart part)
    {
        return new Callable<AudioFilePart>()
        {
            @Override
            public AudioFilePart call() throws Exception
            {
                TranscriptionResult res = getTranscription(part.dataStream);
                part.setTranscriptionResult(res);
                return part;
            }
        };
    }

    private TranscriptionResult getTranscription(InputStream audioStream)
        throws Exception
    {
        String jsonResponse = HttpRequestService
            .post(apiEndpoint)
            .authorization("Basic " + apiKey)
            .send(audioStream)
            .body();

        logger.trace("Reo Tuhituhi API responded with a success status.");
        TranscriptionResult res = jsonSerialiser.fromJson(jsonResponse, TranscriptionResult.class);
        logger.trace("Reo Tuhituhi API response was succesfully deserialised.");
        return res;
    }
}