Update Milestone5API.py

Milestone5API.py

@@ -11,7 +11,7 @@ from pydub import AudioSegment
 from moviepy.audio.io.AudioFileClip import AudioFileClip
 
 
-def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_path):
+def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_path, translated_text):
     print("Loading video...")
     video = VideoFileClip(video_file_path)
     print("Video loaded successfully.")
@@ -28,6 +28,11 @@ def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_
     video = video.set_audio(new_audio)
     print("Audio set to video successfully.")
 
+    # Add text to video
+    txt_clip = TextClip(translated_text, fontsize=70, color='white')
+    txt_clip = txt_clip.set_pos(('center', 'bottom')).set_duration(video.duration)
+    video = CompositeVideoClip([video, txt_clip])
+
     print("Writing video file...")
     try:
         video.write_videofile(output_video_file_path, codec="libx264", audio_codec="aac", remove_temp=True)
@@ -36,6 +41,7 @@ def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_
         print("Error writing video file:", e)
 
 
+
 def text_to_speech(translated_text_file_path, video_path):
     device = "cuda" if torch.cuda.is_available() else "cpu"
     tts = TTS("tts_models/fr/mai/tacotron2-DDC").to(device)