Update app.py

app.py

@@ -105,27 +105,29 @@ try:
 except Exception as ex:
     st.write("Please try again with images with types of JPG, JPEG, PNG ...")
 
+# 创建保存音频的文件夹
+AUDIO_OUTPUT_FOLDER = "./output_audio"
+os.makedirs(AUDIO_OUTPUT_FOLDER, exist_ok=True)
+
 def text_to_speech_gtts(text, lang='en', filename="output.mp3"):
     """将文本转换为语音并保存音频文件"""
+    audio_path = os.path.join(AUDIO_OUTPUT_FOLDER, filename)
     tts = gTTS(text=text, lang=lang)
-    audio_path = "./output_audio/output.mp3"
     tts.save(audio_path)
     return audio_path
 
 try:
     # 生成语音文件
     audio_file_path = text_to_speech_gtts(result)
-
     # 在 Streamlit 中播放音频
     st.audio(audio_file_path, format="audio/mp3")
-except Exception as ex:
-    st.write("An error occurred while processing the audio.")
-    
     # 提供下载按钮
-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",
-    )
+    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("An error occurred while processing the audio.")