Update app.py

app.py

@@ -9,9 +9,10 @@ import os
 script_generator = pipeline("text-generation", model="gpt2")
 
 # Function to generate a comedy script using GPT-2
-def generate_comedy_script(prompt):
-    result = script_generator(prompt, max_length=150, num_return_sequences=1)
-    return result[0]["generated_text"]
+from transformers import pipeline
+
+script_generator = pipeline("text-generation", model="gpt2", truncation=True, max_length=100)  # Adjust max_length as needed
+
 
 from gtts import gTTS
 import os
@@ -23,18 +24,21 @@ def text_to_speech(script):
     return audio_file
 
 
-# Function to generate simple animated text video using MoviePy
-def generate_animation(script, output_video_file="animated_script.mp4"):
+from moviepy.editor import TextClip
+
+def generate_animation(script):
+    lines = script.split('\n')
     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
+
+    for line in lines:
+        text_clip = TextClip(line, fontsize=40, color='white', size=(800, 400), bg_color='black', method='caption', print_cmd=True)
+        text_clip = text_clip.set_duration(3)  # Set duration for each clip
         clips.append(text_clip)
 
-    video = concatenate_videoclips(clips)
-    video.write_videofile(output_video_file, fps=24)
+    # Concatenate all text clips into one video clip
+    final_video = concatenate_videoclips(clips)
+    return final_video
 
-    return output_video_file
 
 import numpy as np
 from scipy.io.wavfile import write