Update Milestone5API.py

Milestone5API.py

@@ -9,16 +9,29 @@ from TTS.api import TTS
 from moviepy.editor import VideoFileClip
 from pydub import AudioSegment
 def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_path):
-
+    print("Loading video...")
     video = VideoFileClip(video_file_path)
+    print("Video loaded successfully.")
 
+    print("Loading new audio...")
     new_audio = AudioSegment.from_wav(new_audio_file_path)
+    print("New audio loaded successfully.")
 
+    print("Trimming audio...")
     new_audio = new_audio[:int(video.duration * 1000)]
+    print("Audio trimmed successfully.")
 
+    print("Setting audio to video...")
     video = video.set_audio(new_audio)
+    print("Audio set to video successfully.")
+
+    print("Writing video file...")
+    try:
+        video.write_videofile(output_video_file_path, codec="libx264", audio_codec="aac", remove_temp=True)
+        print("Video file written successfully.")
+    except Exception as e:
+        print("Error writing video file:", e)
 
-    video.write_videofile(output_video_file_path, codec="libx264", audio_codec="aac", temp_audiofile='temp_audio.mp3', remove_temp=True)
 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)
@@ -35,7 +48,7 @@ def text_to_speech(translated_text_file_path, video_path):
 
     output_video_path = f"CS370_Milestone5/videos/{file_name_only}_new.mp4"
 
-    # Replace the video's audio with the generated audio
+  
     replace_video_audio(video_path, output_file_path, output_video_path)
     return output_video_path