Update app.py

app.py

@@ -179,35 +179,26 @@ def generate_video(image):
     video_clip = ImageSequenceClip(video_frames, fps=24)
     video_duration = video_clip.duration
 
-    # 生成音频数据
+    # 生成音频
     audio_text = text2text_A(text)
-    audio_data, audio_rate = text2audio(audio_text, video_duration)
+    audio_data, audio_rate = text2audio(audio_text, video_clip.duration)
 
     # 将音频数据写入临时文件
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".wav") as tmpfile:
+    with tempfile.NamedTemporaryFile(delete=True, suffix=".wav") as tmpfile:
         scipy.io.wavfile.write(tmpfile, audio_rate, audio_data)
-        tmpfile_path = tmpfile.name
-
-    # 创建AudioFileClip对象
-    audio_clip = AudioFileClip(tmpfile_path)
+        tmpfile.flush()  # 确保数据已写入磁盘
+        audio_clip = AudioFileClip(tmpfile.name)
 
     # 将音频添加到视频中
     video_clip = video_clip.set_audio(audio_clip)
-    print("audio_done")
 
     # 将视频写入临时文件
     with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmpfile:
         video_clip.write_videofile(tmpfile.name, codec="libx264", audio_codec="aac")
         video_file_path = tmpfile.name
 
-    # 读取临时文件数据并删除
-    with open(video_file_path, 'rb') as f:
-        video_data = f.read()
-    video_io = BytesIO(video_data)
-    os.remove(video_file_path)
-    os.remove(tmpfile_path)
-    print("video_done")
-    return video_io
+    # 函数现在返回视频文件的路径，不再需要读取数据并删除
+    return video_file_path
     
 import traceback