Upload tryVosk.py

tryVosk.py

@@ -0,0 +1,79 @@
+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()