com.simiacryptus.util.audio.TranscriptionProcessor.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of joe-penai Show documentation

A Java client for OpenAI's API

There is a newer version: 1.0.33

@file:Suppress("MemberVisibilityCanBePrivate")

package com.simiacryptus.util.audio

import com.simiacryptus.openai.OpenAIClient
import java.util.*

open class TranscriptionProcessor(
    var client: OpenAIClient,
    var audioBuffer: Deque,
    var continueFn: () -> Boolean,
    var prompt: String = "",
    var onText: (String) -> Unit,
) {
    fun run() {
        while (this.continueFn() || audioBuffer.isNotEmpty()) {
            val recordAudio = audioBuffer.poll()
            if (null == recordAudio) {
                Thread.sleep(1)
            } else {
                var text = client.transcription(recordAudio, prompt)
                if (prompt.isNotEmpty()) text = " $text"
                val newPrompt = (prompt + text).split(" ").takeLast(32).joinToString(" ")
                prompt = newPrompt
                onText(text)
            }
        }
    }
}