Update app.py

app.py

@@ -1,5 +1,7 @@
 import streamlit as st
 import importlib.util
+import langdetect
+
 try:
     # 检查 accelerate 库是否安装
     spec = importlib.util.find_spec("accelerate")
@@ -29,7 +31,6 @@ def img2text(url):
 #  text2story 
 def text2story(text):
     try:
-        
         story_generator = pipeline("text-generation", model="perplexity-ai/r1-1776", trust_remote_code=True)
         story = story_generator(text, max_length=200, num_return_sequences=1)[0]['generated_text']
         return story
@@ -40,6 +41,15 @@ def text2story(text):
 # text2audio
 def text2audio(story_text):
     try:
+        # 检测故事的语言
+        detected_lang = langdetect.detect(story_text)
+        tts = gTTS(text=story_text, lang=detected_lang)
+        audio_file = io.BytesIO()
+        tts.write_to_fp(audio_file)
+        audio_file.seek(0)
+        return audio_file
+    except langdetect.LangDetectException:
+        st.error("无法检测故事的语言，默认使用英语进行语音合成。")
         tts = gTTS(text=story_text, lang='en')
         audio_file = io.BytesIO()
         tts.write_to_fp(audio_file)
@@ -86,5 +96,8 @@ if uploaded_file is not None:
                              format="audio/mpeg",
                              start_time=0)
 
-    # 删除临时文件
-    os.remove(temp_file_path)
+    # 删除临时文件并进行异常处理
+    try:
+        os.remove(temp_file_path)
+    except Exception as e:
+        st.error(f"删除临时文件时出错: {e}")