Update app.py

app.py

@@ -1,11 +1,13 @@
 import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
-from moviepy.editor import TextClip, CompositeVideoClip, concatenate_videoclips
+from PIL import Image, ImageDraw, ImageFont
+from moviepy.editor import CompositeVideoClip, concatenate_videoclips, ImageClip
 import numpy as np
 from scipy.io.wavfile import write
 from pydub import AudioSegment
 
+
 script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)
 
 
@@ -29,14 +31,19 @@ def generate_animation(script):
         texts = [item['generated_text'] for item in script if isinstance(item, dict) and 'generated_text' in item]
         script = " ".join(texts)
 
-    lines = script.split('. ')  # Changed to splitting on periods for better line segmentation
+    lines = script.split('. ')
     clips = []
     for line in lines:
-        text_clip = TextClip(line, fontsize=40, color='white', size=(800, 400), method='label')  # Changed method to 'label'
-        text_clip = text_clip.set_duration(3)
-        clips.append(text_clip)
+        img = Image.new('RGB', (800, 400), color=(0, 0, 0))
+        d = ImageDraw.Draw(img)
+        fnt = ImageFont.load_default()
+        d.text((10, 180), line, font=fnt, fill=(255, 255, 255))
+        img.save('/tmp/temp.png')
+        clips.append(ImageClip('/tmp/temp.png').set_duration(3))
     final_video = concatenate_videoclips(clips, method="compose")
-    return final_video
+    final_video.write_videofile("/tmp/final_video.mp4", fps=24)
+    return "/tmp/final_video.mp4"
+
 
 
 def generate_sine_wave(frequency, duration, sample_rate=44100, amplitude=0.5):