ailia-speech 1.3.0.4__py3-none-any.whl → 1.3.0.5__py3-none-any.whl

ailia_speech/__init__.py

@@ -475,19 +475,24 @@ class Whisper(AiliaSpeechModel):
 
         self._check(dll.ailiaSpeechPushInputData(self._instance, audio_waveform, channels, audio_waveform.shape[0] // channels, sampling_rate))
         self._check(dll.ailiaSpeechFinalizeInputData(self._instance))
-        self._check(dll.ailiaSpeechTranscribe(self._instance))
 
-        count = ctypes.c_uint(0)
-        self._check(dll.ailiaSpeechGetTextCount(self._instance, ctypes.byref(count)))
-        results = []
-        for i in range(count.value):
-            text = AILIASpeechText()
-            self._check(dll.ailiaSpeechGetText(self._instance, ctypes.byref(text), AILIA_SPEECH_TEXT_VERSION, i))
-            results.append({"text" : text.text.decode(), "time_stamp_begin" : text.time_stamp_begin, "time_stamp_end" : text.time_stamp_end, "person_id" : text.person_id, "language" : text.language.decode(), "confidence" : text.confidence})
+        while True:
+            complete = ctypes.c_uint(0)
+            self._check(dll.ailiaSpeechComplete(self._instance, ctypes.byref(complete)))
+            if complete.value == 1:
+                break
 
-        self._check(dll.ailiaSpeechResetTranscribeState(self._instance))
+            self._check(dll.ailiaSpeechTranscribe(self._instance))
+
+            count = ctypes.c_uint(0)
+            self._check(dll.ailiaSpeechGetTextCount(self._instance, ctypes.byref(count)))
+            results = []
+            for i in range(count.value):
+                text = AILIASpeechText()
+                self._check(dll.ailiaSpeechGetText(self._instance, ctypes.byref(text), AILIA_SPEECH_TEXT_VERSION, i))
+                yield {"text" : text.text.decode(), "time_stamp_begin" : text.time_stamp_begin, "time_stamp_end" : text.time_stamp_end, "person_id" : text.person_id, "language" : text.language.decode(), "confidence" : text.confidence}
 
-        return results
+        self._check(dll.ailiaSpeechResetTranscribeState(self._instance))
 
     def __del__(self):
         if self._instance:

{ailia_speech-1.3.0.4.data → ailia_speech-1.3.0.5.data}/scripts/__init__.py

@@ -475,19 +475,24 @@ class Whisper(AiliaSpeechModel):
 
         self._check(dll.ailiaSpeechPushInputData(self._instance, audio_waveform, channels, audio_waveform.shape[0] // channels, sampling_rate))
         self._check(dll.ailiaSpeechFinalizeInputData(self._instance))
-        self._check(dll.ailiaSpeechTranscribe(self._instance))
 
-        count = ctypes.c_uint(0)
-        self._check(dll.ailiaSpeechGetTextCount(self._instance, ctypes.byref(count)))
-        results = []
-        for i in range(count.value):
-            text = AILIASpeechText()
-            self._check(dll.ailiaSpeechGetText(self._instance, ctypes.byref(text), AILIA_SPEECH_TEXT_VERSION, i))
-            results.append({"text" : text.text.decode(), "time_stamp_begin" : text.time_stamp_begin, "time_stamp_end" : text.time_stamp_end, "person_id" : text.person_id, "language" : text.language.decode(), "confidence" : text.confidence})
+        while True:
+            complete = ctypes.c_uint(0)
+            self._check(dll.ailiaSpeechComplete(self._instance, ctypes.byref(complete)))
+            if complete.value == 1:
+                break
 
-        self._check(dll.ailiaSpeechResetTranscribeState(self._instance))
+            self._check(dll.ailiaSpeechTranscribe(self._instance))
+
+            count = ctypes.c_uint(0)
+            self._check(dll.ailiaSpeechGetTextCount(self._instance, ctypes.byref(count)))
+            results = []
+            for i in range(count.value):
+                text = AILIASpeechText()
+                self._check(dll.ailiaSpeechGetText(self._instance, ctypes.byref(text), AILIA_SPEECH_TEXT_VERSION, i))
+                yield {"text" : text.text.decode(), "time_stamp_begin" : text.time_stamp_begin, "time_stamp_end" : text.time_stamp_end, "person_id" : text.person_id, "language" : text.language.decode(), "confidence" : text.confidence}
 
-        return results
+        self._check(dll.ailiaSpeechResetTranscribeState(self._instance))
 
     def __del__(self):
         if self._instance:

{ailia_speech-1.3.0.4.dist-info → ailia_speech-1.3.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ailia_speech
-Version: 1.3.0.4
+Version: 1.3.0.5
 Summary: ailia AI Speech
 Home-page: https://ailia.jp/
 Author: ax Inc.
@@ -49,19 +49,20 @@ import os
 import urllib.request
 
 # Load target audio
-ref_file_path = "demo.wav"
-if not os.path.exists(ref_file_path):
+input_file_path = "demo.wav"
+if not os.path.exists(input_file_path):
 	urllib.request.urlretrieve(
 		"https://github.com/axinc-ai/ailia-models/raw/refs/heads/master/audio_processing/whisper/demo.wa",
 		"demo.wav"
 	)
-audio_waveform, sampling_rate = librosa.load(ref_file_path, mono=True)
+audio_waveform, sampling_rate = librosa.load(input_file_path, mono=True)
 
 # Infer
 speech = ailia_speech.Whisper()
 speech.initialize_model(model_path = "./models/", model_type = ailia_speech.AILIA_SPEECH_MODEL_TYPE_WHISPER_MULTILINGUAL_SMALL)
 recognized_text = speech.transcribe(audio_waveform, sampling_rate)
-print(recognized_text)
+for text in recognized_text:
+	print(text)
 ```
 
 ## API specification

{ailia_speech-1.3.0.4.dist-info → ailia_speech-1.3.0.5.dist-info}/RECORD

@@ -1,12 +1,12 @@
 ailia_speech/LICENSE_AILIA_EN.pdf,sha256=1DzVViPnw1uAS8gJ5a8uN3iZNNR5I1ItIXmezHfUpeM,70149
 ailia_speech/LICENSE_AILIA_JA.pdf,sha256=s628QN47S2bNqIfuSjm2LBf0vIluv2df6MSemn6Ksmw,174134
-ailia_speech/__init__.py,sha256=ooZ5u_Iv8SDEHnqqUR9B9FcYjJRvHt1GjaZdIqUMfRk,25430
+ailia_speech/__init__.py,sha256=owi3WkmDlaYf0P1TJF_CVyVSW-cLMj3kp9RoVH9Rd3c,25637
 ailia_speech/linux/arm64-v8a/libailia_speech.so,sha256=JAOwnBr7lbiMZmPCM99pd4vJQ08ZuXDPpq-FurrXSnE,166096
 ailia_speech/linux/x64/libailia_speech.so,sha256=WbFvA5wKTgS_Zx8ErT7WBKJbzOUexavr4nP4EkLNawQ,171360
 ailia_speech/mac/libailia_speech.dylib,sha256=-JAC40yLslAVMvfh6LhDvP3Zyt3hIT3WZc7wa9-07zU,317112
 ailia_speech/windows/x64/ailia_speech.dll,sha256=WJCOHi0Na4tdMG1RT7dA7yAoWumiGSWeW1vxUtiXDS8,126464
-ailia_speech-1.3.0.4.data/scripts/__init__.py,sha256=ooZ5u_Iv8SDEHnqqUR9B9FcYjJRvHt1GjaZdIqUMfRk,25430
-ailia_speech-1.3.0.4.dist-info/METADATA,sha256=OrP0OpJWFywZVcdzchCDDu_UdBXSABFtvHk7M4Pmtx4,1907
-ailia_speech-1.3.0.4.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-ailia_speech-1.3.0.4.dist-info/top_level.txt,sha256=Ou9XeJ9AvdK8eutw07oosCthftD1tRYzAgNY2BrYhDc,13
-ailia_speech-1.3.0.4.dist-info/RECORD,,
+ailia_speech-1.3.0.5.data/scripts/__init__.py,sha256=owi3WkmDlaYf0P1TJF_CVyVSW-cLMj3kp9RoVH9Rd3c,25637
+ailia_speech-1.3.0.5.dist-info/METADATA,sha256=webGrA8CrUKo8WSk_vIvqb_aDekuX7yzXlelkeOrP6M,1932
+ailia_speech-1.3.0.5.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+ailia_speech-1.3.0.5.dist-info/top_level.txt,sha256=Ou9XeJ9AvdK8eutw07oosCthftD1tRYzAgNY2BrYhDc,13
+ailia_speech-1.3.0.5.dist-info/RECORD,,