Update audio_processing/T2A.py

audio_processing/T2A.py

@@ -12,6 +12,12 @@ class T2A:
     def __init__(self, input_text: str):
         self.output_model = pipe_tts(input_text)
 
+    def __get_duration(self, raw: bytes):
+        chunk = io.BytesIO(raw)
+        audio, sample_rate = librosa.load(chunk, sr=SAMPLING_RATE)
+        duration = librosa.get_duration(y=audio, sr=sample_rate)
+        return duration
+
     def get_audio(self):
         try:
             synth = self.output_model["audio"][0]
@@ -22,8 +28,12 @@ class T2A:
                 sf.write(buffer, synth, SAMPLING_RATE, format='wav')
                 output = buffer.getvalue() # bytes
 
-            print(f"output : {output}, type : {type(output)}")
+            print(f"type : {type(output)}")
+
+            duration = __get_duration(output)
+
+            print(f"duration : {duration}")
 
-            return output
+            return output, SAMPLING_RATE, duration
         except Exception as e:
             logging.error(f"Raised an error : {e}")