from vosk import Model, KaldiRecognizer
import wave
import json
import os

class vosk():
    def __init__(self, file_folder, outputResult, outputText, lang):
        self.file_folder = file_folder
        self.outputResult = outputResult
        self.outputText = outputText
        self.lang = lang

    def control(self):
        if os.path.exists(self.outputResult):
            os.remove(self.outputResult)
        if os.path.exists(self.outputText):
            os.remove(self.outputText)

    def useVosk(self, file_name):
        wf = wave.open(file_name, "rb")
        # initialize a str to hold results
        results = ""
        textResults = ""

        # build the model and recognizer objects.
        if lang == "Turkish" or lang == "Türkçe" or lang == "türkçe" or lang == "turkish":
            model = Model("vosk-modelss/vosk-model-tr")

        elif lang == "English" or lang == "İngilizce" or lang == "ingilizce" or lang == "english":
            model = Model("vosk-modelss/vosk-model-en")
        else:
            print("Dil bulunamadı!------The language has not been found!")

        recognizer = KaldiRecognizer(model, wf.getframerate())
        recognizer.SetWords(True)

        while True:
            data = wf.readframes(4000)
            if len(data) == 0:
                break
            if recognizer.AcceptWaveform(data):
                recognizerResult = recognizer.Result()
                results = results + recognizerResult
                results_y = (json.dumps(results, ensure_ascii=False).encode('utf-8')).decode()
                resultDict = json.loads(recognizerResult)

        # process "final" result
        results = results + recognizer.FinalResult()
        results_y = (json.dumps(results, ensure_ascii=False).encode('utf-8')).decode()
        for i in range(results_y.count('text')):
            txtt = results_y.split("text")[i + 1].split("\\n")[0].split('"')[2].split("\\")[0]
            textResults += txtt
            textResults += " "

        resultDict = json.loads(recognizer.FinalResult())
        if len(textResults) == 0:
            textResults += txtt
        return results, textResults

    def playWav(self):
        self.control()
        outputResultt = open(self.outputResult, 'w', encoding="utf-8")
        outputTextt = open(self.outputText, 'w', encoding="utf-8")
        for i in os.listdir(self.file_folder):
            if i.endswith("wav"):
                results, textResults = self.useVosk(self.file_folder + i)
                # write results to a file
                outputResultt.write("%s\n" % results)

                # write text portion of results to a file
                outputTextt.write("%s\n" % textResults)
        print("Metin alımı tamamlandı!----------Text retrieval has been completed!")
        outputTextt.close()
        outputResultt.close()


file_fold, outputR, outputT = "Sounds/tr/", "texts/resulttr.json", "texts/texttr.txt"
lang = "turkish"
vosk(file_fold, outputR, outputT, lang).playWav()