Spaces:
Runtime error
Runtime error
| import gradio as gr | |
| from transformers import pipeline | |
| import pyttsx3 | |
| from moviepy.editor import TextClip, CompositeVideoClip, concatenate_videoclips | |
| from pydub import AudioSegment | |
| import os | |
| # Load pre-trained Hugging Face text generation model for comedy scripts | |
| 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"] | |
| # Text-to-Speech (TTS) using pyttsx3 for comedy | |
| def text_to_speech(script, output_audio_file="script_audio.mp3"): | |
| engine = pyttsx3.init() | |
| engine.save_to_file(script, output_audio_file) | |
| engine.runAndWait() | |
| return output_audio_file | |
| # Function to generate simple animated text video using MoviePy | |
| def generate_animation(script, output_video_file="animated_script.mp4"): | |
| 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"): | |
| # Simple sine wave generation using pydub for kids' music | |
| music = AudioSegment.sine(frequency=261.63) # C4 note | |
| music += AudioSegment.sine(frequency=293.66) # D4 note | |
| music += AudioSegment.sine(frequency=329.63) # E4 note | |
| # Extend for more complexity, then export | |
| music.export(output_music_file, format="wav") | |
| return output_music_file | |
| # 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) | |
| 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 | |
| # Gradio interface to bring everything together | |
| def generate_comedy_and_animation(prompt): | |
| # Comedy Script Generation | |
| script = generate_comedy_script(prompt) | |
| audio_file = text_to_speech(script) | |
| video_file = generate_animation(script) | |
| return script, audio_file, video_file | |
| def generate_kids_content(theme): | |
| video_file, music_file = generate_kids_animation_with_music(theme) | |
| return music_file, video_file | |
| # Gradio app for Comedy and Kids Content | |
| with gr.Blocks() as app: | |
| gr.Markdown("## AI Comedy and Kids Content Generator") | |
| with gr.Tab("Generate Comedy Animation"): | |
| prompt_input = gr.Textbox(label="Comedy Prompt") | |
| generate_btn = gr.Button("Generate Comedy Script and Animation") | |
| comedy_script = gr.Textbox(label="Generated Script") | |
| comedy_audio = gr.Audio(label="Generated Audio") | |
| comedy_video = gr.Video(label="Generated Animation") | |
| generate_btn.click( | |
| generate_comedy_and_animation, | |
| inputs=prompt_input, | |
| outputs=[comedy_script, comedy_audio, comedy_video] | |
| ) | |
| with gr.Tab("Generate Kids Music Animation"): | |
| theme_input = gr.Textbox(label="Kids Music Theme") | |
| generate_music_btn = gr.Button("Generate Kids Music and Animation") | |
| kids_music_audio = gr.Audio(label="Generated Music") | |
| kids_music_video = gr.Video(label="Generated Kids Animation") | |
| generate_music_btn.click( | |
| generate_kids_content, | |
| inputs=theme_input, | |
| outputs=[kids_music_audio, kids_music_video] | |
| ) | |
| app.launch() | |