Update app.py

app.py

@@ -12,6 +12,9 @@ import streamlit as st
 import torch
 from ultralyticsplus import YOLO, render_result
 
+from gtts import gTTS
+import os
+
 from convert import convert_to_braille_unicode, parse_xywh_and_class
 
 
@@ -26,6 +29,15 @@ def load_image(image_path):
     image = PIL.Image.open(image_path)
     return image
 
+def play_audio(text):
+    """Convert text to speech and play audio in Streamlit"""
+    tts = gTTS(text=text, lang='en')
+    tts.save("detected_braille.mp3")
+    audio_file = open("detected_braille.mp3", "rb")
+    audio_bytes = audio_file.read()
+    st.audio(audio_bytes, format="audio/mp3")
+    audio_file.close()
+    os.remove("detected_braille.mp3")  # 删除文件，节省空间
 
 # title
 st.title("Braille Pattern Detection")
@@ -101,6 +113,7 @@ try:
             )
         st.write(str_left_to_right)
         
+# Convert detected Braille text to audio and play
     if detected_text:
         play_audio(detected_text)