Update app.py

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 from transformers import pipeline
 from gtts import gTTS
-from moviepy.editor import ImageSequenceClip, CompositeVideoClip, ImageClip, AudioFileClip, VideoFileClip, TextClip
+from moviepy.editor import ImageSequenceClip, CompositeVideoClip, ImageClip, AudioFileClip, VideoFileClip
 from PIL import Image, ImageDraw, ImageFont
 import numpy as np
 from scipy.io.wavfile import write
 
+# Load and Initialize Models
 script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)
 
 def generate_comedy_script(prompt):
@@ -31,14 +32,12 @@ def create_images_from_script(script):
         image_paths.append(image_path)
     return image_paths
 
-
 def generate_animation_with_images(script):
     image_paths = create_images_from_script(script)
     clips = []
     for i, img_path in enumerate(image_paths):
         image_clip = ImageClip(img_path).set_duration(3).set_position(('center', 'center'))
         clips.append(image_clip)
-    
     final_video = concatenate_videoclips(clips, method="compose")
     final_video.write_videofile("/tmp/final_video.mp4", fps=24)
     return "/tmp/final_video.mp4"
@@ -50,7 +49,6 @@ def combine_audio_video(video_path, audio_path):
     final_video.write_videofile("/tmp/final_comedy_video.mp4", fps=24)
     return "/tmp/final_comedy_video.mp4"
 
-
 def generate_comedy_and_animation(prompt):
     script = generate_comedy_script(prompt)
     audio_file = text_to_speech(script)
@@ -85,7 +83,6 @@ def generate_kids_animation_with_music(theme, output_video_file="kids_animation.
         frame_path = f'/tmp/kids_temp_{i}.png'
         img.save(frame_path)
         clips.append(ImageClip(frame_path).set_duration(1).set_position(('center', 'center')))
-    
     final_video = CompositeVideoClip(clips, size=(800, 400))
     final_video = final_video.set_audio(AudioFileClip(music_file))
     final_video.write_videofile(output_video_file, fps=24)
@@ -123,5 +120,5 @@ with gr.Blocks() as app:
             outputs=[kids_music_audio, kids_music_video]
         )
 
-
 app.launch()
+