Update app.py

app.py

@@ -14,6 +14,7 @@ from ultralyticsplus import YOLO, render_result
 
 from gtts import gTTS
 import os
+import pygame
 
 from convert import convert_to_braille_unicode, parse_xywh_and_class
 
@@ -29,13 +30,6 @@ def load_image(image_path):
     image = PIL.Image.open(image_path)
     return image
 
-def generate_audio(text, filename="detected_braille.mp3"):
-    """Convert text to speech and save audio as file"""
-    tts = gTTS(text=text, lang='en')
-    audio_path = f"./output_audio/detected_braille.mp3"
-    tts.save(audio_path)
-    return audio_path
-
 # title
 st.title("Braille Pattern Detection")
 
@@ -106,9 +100,34 @@ try:
         box_classes = box_line[:, -1]
         for each_class in box_classes:
             str_left_to_right += convert_to_braille_unicode(model.names[int(each_class)])
-        st.write(str_left_to_right)
         result += str_left_to_right + "\n"
-        audio_file_path = generate_audio(result)
+        st.write(str_left_to_right)
+
+def text_to_speech_gtts(text, lang='en'):
+    # 将文本转换为语音
+    tts = gTTS(text=text, lang=lang)
+    # 保存音频文件
+    tts.save(f"./output_audio/output.mp3")
+
+audio_file_path = f"./output_audio/output.mp3"
+
+text_to_speech_gtts(result)
+
+def play_mp3(file_path):
+    # 初始化 pygame
+    pygame.mixer.init()
+    # 加载 MP3 文件
+    pygame.mixer.music.load(file_path)
+    # 播放 MP3 文件
+    pygame.mixer.music.play()
+
+    # 等待音频播放完毕
+    while pygame.mixer.music.get_busy():
+        continue
+
+# 调用函数，播放 MP3 文件
+play_mp3("f./output_audio/output.mp3")
+
         
 except Exception as ex:
     st.write("Please try again with images with types of JPG, JPEG, PNG ...")