import os from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip from faster_whisper import WhisperModel from googletrans import Translator import pysubs2 import subprocess import time # =================== 步骤1：提取音频 =================== def extract_audio(video_path, audio_path): try: video = VideoFileClip(video_path) audio = video.audio audio.write_audiofile(audio_path) return audio_path except Exception as e: print(f"❌ 音频提取失败: {e}") return None # =================== 步骤2：日语语音转文字 =================== def speech_to_japanese_text(audio_path): try: model = WhisperModel("large-v2", device="cpu", compute_type="int8") # 可改用 "cuda" 加速 segments, _ = model.transcribe(audio_path, language="ja") # 语言设为日语 text = " ".join([segment.text for segment in segments]) return text except Exception as e: print(f"❌ 语音转文字失败: {e}") return None # =================== 步骤3：日语翻译为中文 =================== def translate_japanese_to_chinese(text): try: translator = Translator() translated = translator.translate(text, src='ja', dest='zh-cn') # 指定源语言和目标语言 return translated.text except Exception as e: print(f"❌ 翻译失败: {e}") return None # =================== 步骤4：生成SRT字幕文件 =================== def generate_srt(text, srt_path): try: with open(srt_path, "w", encoding="utf-8") as f: lines = text.split("。 ") # 按日语句号分割 for i, line in enumerate(lines): start_time = i * 5 # 假设每句持续5秒 end_time = start_time + 5 f.write(f"{i+1}\n{start_time} --> {end_time}\n{line}\n\n") return srt_path except Exception as e: print(f"❌ SRT生成失败: {e}") return None # =================== 步骤5：字幕时间轴同步 =================== def sync_subtitles(video_path, srt_path, output_srt_path): try: command = [ "subsync", video_path, "-s", srt_path, "-o", output_srt_path ] subprocess.run(command, check=True) return output_srt_path except Exception as e: print(f"❌ 字幕同步失败: {e}") return None # =================== 步骤6：字幕嵌入视频 =================== def add_subtitles(video_path, subtitle_path, output_path): try: video = VideoFileClip(video_path) subs = pysubs2.load(subtitle_path) # 将字幕转换为TextClip subtitle_clips = [ TextClip(sub.text, fontsize=24, color='white', bg_color='black') .set_position(('center', 'bottom')) .set_duration((sub.end - sub.start) / 1000) # 转换为秒 .set_start(sub.start / 1000) # SRT时间单位是毫秒 for sub in subs ] final_video = CompositeVideoClip([video] + subtitle_clips) final_video.write_videofile(output_path, codec='libx264') print(f"✅ 成功生成带中文字幕的视频: {output_path}") except Exception as e: print(f"❌ 字幕嵌入失败: {e}") # =================== 主流程 =================== def process_video(video_path): temp_files = { "audio": "temp_audio.wav", "japanese_text": "japanese_text.txt", "chinese_text": "chinese_text.txt", "srt": "chinese_subtitles.srt", "synced_srt": "synced_subtitles.srt", "output_video": "output_video.mp4" } print("【步骤1】提取音频...") audio_path = extract_audio(video_path, temp_files["audio"]) if not audio_path: return print("【步骤2】日语语音转文字...") japanese_text = speech_to_japanese_text(audio_path) if not japanese_text: return print("【步骤3】日语翻译为中文...") chinese_text = translate_japanese_to_chinese(japanese_text) if not chinese_text: return print("【步骤4】生成SRT字幕...") srt_path = generate_srt(chinese_text, temp_files["srt"]) if not srt_path: return print("【步骤5】字幕时间轴同步...") synced_srt_path = sync_subtitles(video_path, srt_path, temp_files["synced_srt"]) if not synced_srt_path: return print("【步骤6】字幕嵌入视频...") add_subtitles(video_path, synced_srt_path, temp_files["output_video"]) # =================== 运行主流程 =================== if __name__ == "__main__": input_video_path = "input_video.mp4" # 替换为你的视频文件路径 process_video(input_video_path)