Update 3mergeDialogToVideo.py

3mergeDialogToVideo.py

@@ -21,6 +21,9 @@ from io import BytesIO
 import pandas as pd
 from PIL import Image
 import os
+from mutagen.mp3 import MP3 #读取音频获取时长
+
+
 azure_speech_key = os.getenv('azure_speech_key')
 azure_service_region = os.getenv('azure_service_region')
 my_openai_key = os.getenv('my_openai_key')
@@ -149,9 +152,12 @@ def get_audio_data(text:str)-> "返回audio data io句柄, duration":
         with tempfile.NamedTemporaryFile(dir='/',delete=False) as temp_file:
             temp_file.write(response.content)
             temp_file.close()
+            audio = MP3(temp_file.name)
+            # 获取音频时长（单位为秒）
+            audio_duration_seconds = audio.info.length #int即可
             # 在这里完成您对文件的操作，比如返回文件名
             file_name = temp_file.name
-        return file_name, str(response.audio_duration)
+        return file_name, audio_duration_seconds
 
     else:
         print("Error: Failed to synthesize audio. Status code:", response.status_code)