Update Milestone5API.py

Milestone5API.py

@@ -8,22 +8,17 @@ import torch
 from TTS.api import TTS
 from moviepy.editor import VideoFileClip
 from pydub import AudioSegment
-from moviepy.editor import VideoFileClip, AudioFileClip, TextClip, CompositeVideoClip
 import subprocess
+from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
+
+def add_text_to_video(input_video_path, output_video_path, text):
+    video = VideoFileClip(input_video_path)
+    txt_clip = TextClip(text, fontsize=70, color='white')
+    txt_clip = txt_clip.set_duration(video.duration)
+    txt_clip = txt_clip.set_position(('center', 'bottom'))
+    video_with_text = CompositeVideoClip([video, txt_clip])
+    video_with_text.write_videofile(output_video_path, codec="libx264", audio_codec="aac", remove_temp=True)
 
-def add_text_to_video(input_video_path, output_video_path, text, new_video_path):
-    try:
-        subprocess.run([
-            'ffmpeg',
-            '-y',
-            '-i', input_video_path,
-            '-vf', f"drawtext=text='{text}':fontsize=24:fontcolor=white:x=(w-text_w)/2:y=(h-text_h)/2",
-            '-c:a', 'copy',
-            new_video_path
-        ],check=True, stdout=subprocess.PIPE)
-        print("Text added to video successfully.")
-    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,new_video_path):
     print("Loading video...")
@@ -44,7 +39,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, new_video_path)
+        add_text_to_video(output_video_file_path, new_video_path, translated_text)
         print("Video file written successfully.")
     except Exception as e:
         print("Error writing video file:", e)