Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
 import pyttsx3
-from manim import Scene, Text, Write, config
+from moviepy.editor import TextClip, CompositeVideoClip, concatenate_videoclips
 from pydub import AudioSegment
 import os
 
@@ -20,20 +20,18 @@ def text_to_speech(script, output_audio_file="script_audio.mp3"):
     engine.runAndWait()
     return output_audio_file
 
-# Function to generate animated video using Manim (for comedy script)
+# Function to generate simple animated text video using MoviePy
 def generate_animation(script, output_video_file="animated_script.mp4"):
-    class ScriptAnimation(Scene):
-        def construct(self):
-            # Create a text object and animate it
-            script_text = Text(script[:200])  # Limited text for simplicity
-            self.play(Write(script_text))
-    
-    config.media_dir = "./media"
-    config.output_file = output_video_file
-    animation = ScriptAnimation()
-    animation.render()
-    
-    return os.path.join(config.media_dir, "videos", output_video_file)
+    clips = []
+    for i, line in enumerate(script.split('. ')):
+        text_clip = TextClip(line, fontsize=40, color='white', size=(800, 400), bg_color='black', method='caption')
+        text_clip = text_clip.set_duration(3)  # Each line lasts 3 seconds
+        clips.append(text_clip)
+
+    video = concatenate_videoclips(clips)
+    video.write_videofile(output_video_file, fps=24)
+
+    return output_video_file
 
 # Function to generate simple music (for kids content)
 def generate_kids_music(theme, output_music_file="kids_music.wav"):
@@ -46,21 +44,15 @@ def generate_kids_music(theme, output_music_file="kids_music.wav"):
     music.export(output_music_file, format="wav")
     return output_music_file
 
-# Function to combine kids music with an animation
+# Function to combine kids music with a simple animation
 def generate_kids_animation_with_music(theme, output_video_file="kids_animation.mp4"):
     # Step 1: Generate kids music
     music_file = generate_kids_music(theme)
     
     # Step 2: Generate simple animation for the music (using text animation as a placeholder)
-    class KidsMusicAnimation(Scene):
-        def construct(self):
-            music_text = Text("Kids Music: " + theme)
-            self.play(Write(music_text))
-    
-    config.media_dir = "./media"
-    config.output_file = output_video_file
-    animation = KidsMusicAnimation()
-    animation.render()
+    clips = [TextClip("Kids Music: " + theme, fontsize=70, color='yellow', size=(800, 400), bg_color='blue', method='caption').set_duration(5)]
+    video = CompositeVideoClip(clips)
+    video.write_videofile(output_video_file, fps=24)
     
     return output_video_file, music_file
 
@@ -109,3 +101,4 @@ with gr.Blocks() as app:
 app.launch()
 
 
+