Update app.py

app.py

@@ -82,7 +82,53 @@ def generate_kids_animation_with_music(theme, output_video_file="kids_animation.
         img = Image.new('RGB', (800, 400), color=(0, 0, 255))
         d = ImageDraw.Draw(img)
         fnt = ImageFont.load_default()
-        d.text((10, 180
+        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 music_file, output_video_file
+
+# Main Function to Generate Kids Content
+def generate_kids_content(theme):
+    music_file, video_file = generate_kids_animation_with_music(theme)
+    return music_file, video_file
+
+# Gradio Interface
+with gr.Blocks() as app:
+    gr.Markdown("## AI Comedy and Kids Content Generator")
+
+    # Comedy Animation Tab
+    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]
+        )
+
+    # Kids Music Animation Tab
+    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()
+