Update app.py

app.py

@@ -76,5 +76,51 @@ def generate_kids_animation_with_music(theme, output_video_file="kids_animation.
     music_file = generate_kids_music(theme)
     clips = []
     for i in range(5):
+        img = Image.new('RGB', (800, 400), color=(0, 0, 255))
+        d = ImageDraw.Draw(img)
+        fnt = ImageFont.load_default()
+        d.text((10, 180), f"Kids Music: {theme}", font=fnt, fill=(255, 255, 0))
+        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)
+    return output_video_file, music_file
+
+def generate_kids_content(theme):
+    video_file, music_file = generate_kids_animation_with_music(theme)
+    return music_file, video_file
+
+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()
+