Create app.py

app.py

@@ -0,0 +1,198 @@
+import gradio as gr
+from PIL import Image, ImageDraw
+import numpy as np
+import imageio
+import tempfile
+from gtts import gTTS
+import torch
+from diffusers import DiffusionPipeline, DPMSolverMultistepScheduler
+
+# ========== AI MODEL SETUP (Done once on startup) ==========
+# This part loads the AI model that creates the videos.
+try:
+    device = "cuda" if torch.cuda.is_available() else "cpu"
+    video_pipe = DiffusionPipeline.from_pretrained(
+        "cerspense/zeroscope_v2_576w",
+        torch_dtype=torch.float16
+    )
+    video_pipe.scheduler = DPMSolverMultistepScheduler.from_config(video_pipe.scheduler.config)
+    video_pipe.to(device)
+    model_loaded = True
+except Exception as e:
+    print(f"Error loading model: {e}")
+    model_loaded = False
+
+# ========== UI & STYLING ==========
+# This is where we design the beautiful, child-friendly interface!
+custom_css = """
+/* --- Import a fun, rounded font from Google Fonts --- */
+@import url('https://fonts.googleapis.com/css2?family=Mali:wght@400;700&display=swap');
+
+/* --- Overall page style --- */
+.gradio-container {
+    background: linear-gradient(135deg, #a0c4ff 0%, #c4f5ff 100%); /* Sky blue gradient */
+    font-family: 'Mali', cursive; /* Use our fun, new font */
+}
+
+/* --- Main Title: Big, colorful, and magical! --- */
+#main-title {
+    font-size: 48px;
+    font-weight: 700;
+    color: #ff6b6b; /* Playful coral color */
+    text-align: center;
+    text-shadow: 2px 2px 4px rgba(0,0,0,0.2);
+    padding-top: 20px;
+}
+
+#subtitle {
+    font-size: 24px;
+    color: #4a4a4a;
+    text-align: center;
+    margin-top: -10px;
+}
+
+/* --- The main button: Looks like a magic wand button! --- */
+#animate-btn {
+    background: linear-gradient(45deg, #ffd700, #ff8c00); /* Golden-orange gradient */
+    color: white;
+    border: none;
+    border-radius: 50px; /* Fully rounded */
+    padding: 15px 30px;
+    font-size: 22px;
+    font-weight: bold;
+    box-shadow: 0 5px 15px rgba(255, 165, 0, 0.4);
+    transition: transform 0.2s, box-shadow 0.2s;
+}
+#animate-btn:hover {
+    transform: scale(1.05); /* Grow on hover */
+    box-shadow: 0 8px 20px rgba(255, 165, 0, 0.6);
+}
+
+/* --- Text input box: Looks like a story scroll --- */
+#story-input textarea {
+    background-color: #ffffff;
+    border: 3px solid #ffca7a; /* Warm, friendly border */
+    border-radius: 15px;
+    font-size: 18px;
+    padding: 15px;
+    line-height: 1.5;
+}
+
+/* --- Output containers: Framed like a picture book --- */
+.output-box {
+    border: 8px solid #8B4513; /* Wooden frame border */
+    border-image: url('https://www.publicdomainpictures.net/pictures/120000/velka/wooden-frame-texture.jpg') 20 stretch;
+    border-radius: 15px;
+    padding: 10px;
+    background: #fdf6e3; /* Parchment paper background */
+    box-shadow: 5px 5px 15px rgba(0,0,0,0.3);
+}
+
+/* --- Explainer character image: A friendly round guide --- */
+.explainer-img img {
+    border-radius: 50% !important;
+    border: 5px solid #ff6b6b;
+    object-fit: cover;
+}
+"""
+
+# ========== BACKEND LOGIC (The "How it Works" part) ==========
+
+def create_animation(story):
+    """Takes the story and creates the video."""
+    if not model_loaded:
+        # Create a dummy error output if the model failed to load
+        blank_image = np.zeros((320, 576, 3), dtype=np.uint8)
+        with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_file:
+            imageio.mimsave(temp_file.name, [blank_image for _ in range(3)], fps=3)
+            return "AI model not loaded.", "Error: AI model failed to load.", blank_image, temp_file.name
+    
+    if not story:
+        return "Please write a story first!", "", np.zeros((320, 576, 3), dtype=np.uint8), None
+    
+    prompt = f"{story.strip()}, beautiful animation, storybook style, vibrant colors, for children"
+    
+    # Generate the video
+    video_frames = video_pipe(prompt, num_inference_steps=25, height=320, width=576, num_frames=24).frames
+    
+    with tempfile.NamedTemporaryFile(suffix=".mp4", delete=False) as temp_file:
+        video_path = temp_file.name
+        imageio.mimsave(video_path, video_frames, fps=12)
+
+    final_frame_image = video_frames[-1]
+    
+    code_explanation = f"""
+    ### 📜 The Magic Spell (Prompt) We Used:
+    The AI was told to create a video based on this magic spell:
+    **"{prompt}"**
+    """
+    
+    story_explanation = f"Hooray! ✨ Our AI artist read your story about '{story}' and drew this wonderful movie for you!"
+
+    # Generate audio
+    with tempfile.NamedTemporaryFile(suffix=".mp3", delete=False) as temp_audio:
+        tts = gTTS(story_explanation, lang='en')
+        tts.save(temp_audio.name)
+        audio_path = temp_audio.name
+    
+    return story_explanation, audio_path, code_explanation, final_frame_image, video_path
+
+# ========== THE GRADIO APP INTERFACE ==========
+
+with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as app:
+    gr.Markdown("# 🏰 Storybook Animator 🏰", elem_id="main-title")
+    gr.Markdown("Let's bring your imagination to life!", elem_id="subtitle")
+
+    with gr.Row():
+        # --- LEFT SIDE: INPUT ---
+        with gr.Column(scale=2):
+            gr.Markdown("### ✍️ Write Your Magical Story Here:")
+            story_input = gr.Textbox(
+                label="Story",
+                placeholder="A happy little fox making friends with a butterfly...",
+                lines=5,
+                elem_id="story-input",
+                show_label=False
+            )
+            animate_btn = gr.Button("✨ Create My Animation! ✨", elem_id="animate-btn")
+            
+            gr.Examples(
+                examples=[
+                    ["A friendly dragon flying over a candy castle."],
+                    ["A little robot planting a flower on the moon."],
+                    ["A curious squirrel having a tea party with a gnome."],
+                    ["A mermaid discovering a sunken treasure chest filled with glowing pearls."]
+                ],
+                inputs=story_input,
+                label="🌟 Or Try These Fun Ideas:"
+            )
+
+        # --- RIGHT SIDE: OUTPUT ---
+        with gr.Column(scale=3):
+            with gr.Tabs():
+                with gr.TabItem("🎬 Your Movie!"):
+                    output_video = gr.Video(label="Your Story Comes Alive", elem_classes="output-box")
+                
+                with gr.TabItem("🖼️ Picture Preview"):
+                    output_image = gr.Image(label="A Snapshot from Your Story", elem_classes="output-box")
+                    
+                with gr.TabItem("💡 How It Works"):
+                    story_explanation_text = gr.Textbox(label="What Happened", interactive=False, elem_classes="output-box")
+                    explainer_audio = gr.Audio(label="Listen to the Story", type="filepath")
+                    code_explanation_output = gr.Markdown(elem_classes="output-box")
+    
+    # --- Connect the button to the function ---
+    animate_btn.click(
+        fn=create_animation,
+        inputs=story_input,
+        outputs=[
+            story_explanation_text,
+            explainer_audio,
+            code_explanation_output,
+            output_image,
+            output_video
+        ]
+    )
+
+if __name__ == "__main__":
+    app.launch()