Update app.py

app.py

@@ -103,39 +103,34 @@ try:
         result += str_left_to_right + "\n"
         st.write(str_left_to_right)
 
-def text_to_speech_gtts(text, lang='en'):
-    # 将文本转换为语音
-    tts = gTTS(text=text, lang=lang)
-    # 保存音频文件
-    tts.save("./output_audio/output.mp3")
-
-audio_file_path = "./output_audio/output.mp3"
-
-text_to_speech_gtts(result)
+AUDIO_OUTPUT_FOLDER = "./output_audio"
+os.makedirs(AUDIO_OUTPUT_FOLDER, exist_ok=True)
 
-def play_mp3(file_path):
-    # 初始化 pygame
-    pygame.mixer.init()
-    # 加载 MP3 文件
-    pygame.mixer.music.load(file_path)
-    # 播放 MP3 文件
-    pygame.mixer.music.play()
+def text_to_speech_gtts(text, lang='en', filename="output.mp3"):
+    """将文本转换为语音并保存音频文件"""
+    tts = gTTS(text=text, lang=lang)
+    audio_path = os.path.join(AUDIO_OUTPUT_FOLDER, filename)
+    tts.save(audio_path)
+    return audio_path
 
-    # 等待音频播放完毕
-    while pygame.mixer.music.get_busy():
-        continue
+# 示例检测结果文本
+detected_text = "This is a sample Braille detection result."
 
-# 调用函数，播放 MP3 文件
-play_mp3("./output_audio/output.mp3")
+try:
+    # 生成语音文件
+    audio_file_path = text_to_speech_gtts(detected_text)
+
+    # 在 Streamlit 中播放音频
+    st.audio(audio_file_path, format="audio/mp3")
+
+    # 提供下载按钮
+    with open(audio_file_path, "rb") as audio_file:
+        st.download_button(
+            label="Download Braille Audio",
+            data=audio_file,
+            file_name="detected_braille.mp3",
+            mime="audio/mp3",
+        )
 
-        
 except Exception as ex:
-    st.write("Please try again with images with types of JPG, JPEG, PNG ...")
-
-with open(audio_file_path, "rb") as fl:
-    st.download_button(
-        "Download Braille Audio",
-        data=fl,
-        file_name="detected_braille.mp3",
-        mime="audio/mp3",
-    )
+    st.write("An error occurred while processing the audio.")