Update audio.py

audio.py

@@ -8,7 +8,7 @@ os.environ["IMAGEIO_FFMPEG_EXE"] = "/usr/local/bin/ffmpeg"
 # 如果需要，设置 ImageMagick 的路径
 # os.environ["IMAGEMAGICK_BINARY"] = "/usr/local/bin/convert"
 
-def gen_audio(subtitles):
+def gen_audio(subtitles, tmpdir):
 # 字幕列表，表示在指定时间段显示对应的字幕
 # subtitles = [
 #    [[0, 3], "這裡請注意右臂"],
@@ -19,7 +19,7 @@ def gen_audio(subtitles):
     print("===> Subtitles:")
     for subtitle in subtitles:
         print(subtitle)
-    combined_clip = VideoFileClip("tmp/output.mp4")
+    combined_clip = VideoFileClip(tmpdir + "/output.mp4")
     # 获取视频总时长
     total_duration = combined_clip.duration
 
@@ -77,8 +77,8 @@ def gen_audio(subtitles):
     video_with_audio = video_with_subtitles.set_audio(audio_clip)
 
     # 导出带语音和字幕的视频
-    video_with_audio.write_videofile('tmp/output_with_subtitles.mp4', fps=24)
-    return 'tmp/output_with_subtitles.mp4'
+    video_with_audio.write_videofile(tmpdir + '/output_with_subtitles.mp4', fps=24)
+    return tmpdir + '/output_with_subtitles.mp4'
 
 
 def merge_subtitles(subtitles):
@@ -116,7 +116,7 @@ def merge_subtitles(subtitles):
                 break  # 退出合并循环
 
         # 添加合并后的字幕到新列表
-        current_content = '這裡請注意' + '、'.join(list(set(current_content.split('、'))))
+        current_content = '這裡請注意' + '、'.join(list(set(current_content.replace("細節", "").split('、'))))
         merged_subtitles.append([[current_start, current_end], current_content])
     
     return merged_subtitles