Update app.py

app.py

@@ -14,44 +14,39 @@ def install_imagemagick():
 
 install_imagemagick()
 
-def create_text_clip(text, fontsize, color, size):
-    img = Image.new('RGB', size, color='black')
+from moviepy.editor import VideoFileClip, CompositeVideoClip, ImageClip
+from PIL import Image, ImageDraw, ImageFont
+
+def create_text_clip(text, fontsize=70, color='white', size=(1080, 1920)):
+    font = ImageFont.truetype("arial.ttf", fontsize)
+    img = Image.new('RGBA', size, (0, 0, 0, 0))
     draw = ImageDraw.Draw(img)
-    font_path = "arial.ttf"  # Make sure this file is in the root directory of your space
-    font = ImageFont.truetype(font_path, fontsize)
     w, h = draw.textbbox((0, 0), text, font=font)[2:]
     draw.text(((size[0] - w) / 2, (size[1] - h) / 2), text, font=font, fill=color)
-    return np.array(img)
-
-def process_video(text):
-    video_folder = "videos"
-    video_files = [os.path.join(video_folder, f) for f in os.listdir(video_folder) if f.endswith(('mp4', 'mov', 'avi', 'mkv'))]
-    if not video_files:
-        raise FileNotFoundError("No video files found in the specified directory.")
-
-    selected_video = random.choice(video_files)
-    video = VideoFileClip(selected_video)
-    start_time = random.uniform(0, max(0, video.duration - 60))
-    video = video.subclip(start_time, min(start_time + 60, video.duration))
-
-    # Resize the video to have a width of 1080 while keeping the aspect ratio
-    new_width = 1080
-    new_height = int(video.h * (new_width / video.w))
-    video = video.resize(width=new_width, resample=Image.LANCZOS)
-
-    # Crop the video to a 9:16 aspect ratio
-    target_height = 1920
-    video = video.crop(y_center=video.h / 2, height=target_height)
-
-    text_lines = text.split()
-    text = "\n".join([" ".join(text_lines[i:i+8]) for i in range(0, len(text_lines), 8)])
-
-    text_img = create_text_clip(text, fontsize=70, color='white', size=video.size)
-    text_clip = ImageClip(text_img).set_duration(video.duration).set_position(('center', 'center')).set_opacity(1)
+    return img
 
+def process_video(video_path, text):
+    video = VideoFileClip(video_path)
+    
+    # Calculate new width to maintain the 9:16 aspect ratio
+    new_width = int(video.h * 9 / 16)
+    
+    # Resize video to maintain 9:16 aspect ratio
+    video = video.resize(width=new_width)
+    
+    # Create a text image
+    text_img = create_text_clip(text, fontsize=70, color='white', size=(new_width, video.h))
+    
+    # Convert the text image to an ImageClip
+    text_clip = ImageClip(text_img).set_duration(video.duration)
+    
+    # Composite the video with the text clip
     final_clip = CompositeVideoClip([video, text_clip])
-
-    output_path = "output.mp4"
+    
+    # Save the final video
+    final_clip.write_videofile("output_video.mp4", codec="libx264", fps=24)
+    
+    return "output_video.mp4"
     final_clip.write_videofile(output_path, codec="libx264")
 
     return output_path