Update Milestone5API.py

Milestone5API.py

@@ -11,7 +11,7 @@ from pydub import AudioSegment
 from moviepy.editor import VideoFileClip, AudioFileClip, TextClip, CompositeVideoClip
 import subprocess
 
-def add_text_to_video(input_video_path, output_video_path, text):
+def add_text_to_video(input_video_path, output_video_path, text, new_video_path):
     try:
         subprocess.run([
             'ffmpeg',
@@ -25,7 +25,7 @@ def add_text_to_video(input_video_path, output_video_path, text):
     except subprocess.CalledProcessError as e:
         print("Error adding text to video:", e)
 
-def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_path, translated_text):
+def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_path, translated_text,new_video_path):
     print("Loading video...")
     video = VideoFileClip(video_file_path)
     print("Video loaded successfully.")
@@ -44,7 +44,7 @@ def replace_video_audio(video_file_path, new_audio_file_path, output_video_file_
 
     try:
         video.write_videofile(output_video_file_path, codec="libx264", audio_codec="aac", remove_temp=True)
-        add_text_to_video(output_video_file_path, output_video_file_path, translated_text)
+        add_text_to_video(output_video_file_path, output_video_file_path, translated_text, new_video_path)
         print("Video file written successfully.")
     except Exception as e:
         print("Error writing video file:", e)
@@ -66,10 +66,10 @@ def text_to_speech(translated_text_file_path, video_path):
     tts.tts_to_file(text=translated_text, file_path=output_file_path)
 
     output_video_path = f"CS370_Milestone5/videos/{file_name_only}_new.mp4"
-
+    new_video_path = f"CS370_Milestone5/videos/{file_name_only}_new2.mp4"
   
     replace_video_audio(video_path, output_file_path, output_video_path,translated_text)
-    return output_video_path
+    return new_video_path
 
 def translate_text_file(input_text_path, output_text_path, source_lang='en', target_lang='fr', model_name="Helsinki-NLP/opus-mt-en-fr", batch_size=8):
     model = MarianMTModel.from_pretrained(model_name)