Update Milestone5API.py

Milestone5API.py

@@ -8,17 +8,20 @@ import torch
 from TTS.api import TTS
 from moviepy.editor import VideoFileClip
 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):
     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)
+    new_audio = AudioFileClip(new_audio_file_path)
     print("New audio loaded successfully.")
 
     print("Trimming audio...")
-    new_audio = new_audio[:int(video.duration * 1000)]
+    new_audio = new_audio.subclip(0, video.duration)
     print("Audio trimmed successfully.")
 
     print("Setting audio to video...")
@@ -32,6 +35,7 @@ def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_
     except Exception as e:
         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)