Update app.py

app.py

@@ -2,39 +2,43 @@ import gradio as gr
 from modelscope.pipelines import pipeline
 from modelscope.utils.constant import Tasks
 
-# Load ModelScope's Text-to-Video Model
-video_model = pipeline(Tasks.text_to-video-synthesis, model="damo-vilab/modelscope-damo-text-to-video")
-
-def generate_pokemon_video(prompt, style, duration, image):
-    """ Generate Pokémon-themed video based on user input. """
-    prompt = f"{prompt}, Pokémon style: {style}"
-    
-    # Prepare input parameters
-    inputs = {'text': prompt, 'duration': duration}
+# Load the Pokémon-specific text-to-video model.
+# NOTE: Replace 'pokemon/pokemon-anime-text-to-video' with your actual model ID.
+video_model = pipeline(Tasks.text_to_video_synthesis, model='pokemon/pokemon-anime-text-to-video')
+
+def generate_pokemon_anime_video(prompt, style, duration, image):
+    """
+    Generate a Pokémon anime–themed video.
+    The prompt is enriched with style and context, ensuring the model understands
+    references to Ash, Pikachu, and other trainers.
+    """
+    full_prompt = (f"{prompt}, in {style} style. "
+                   "Ensure the scene reflects the world of Pokémon, featuring Ash, "
+                   "Pikachu, and recognizable elements from the Pokémon series.")
     
+    # Prepare inputs for the model
+    inputs = {'text': full_prompt, 'duration': duration}
     if image:
-        inputs['image'] = image  # Use uploaded image if provided
+        inputs['image'] = image
 
-    # Generate video
     result = video_model(inputs)
     return result["output_video"]
 
-# Gradio UI
+# Build the Gradio UI
 with gr.Blocks() as iface:
-    gr.Markdown("# 🎥 PokeVidGen - AI Pokémon Shorts Generator")
-    gr.Markdown("Enter a prompt, choose a Pokémon style, and generate a video!")
+    gr.Markdown("# 🎥 PokeVidGen AI")
+    gr.Markdown("Generate Pokémon anime shorts! Enter a scene prompt, select an animation style, set the video duration, and optionally upload an image.")
 
     with gr.Row():
-        prompt = gr.Textbox(label="Enter Pokémon Scene", placeholder="Pikachu using Thunderbolt in a forest")
-        style = gr.Dropdown(["Anime", "Classic Pokémon", "Modern 3D"], label="Pokémon Style", value="Anime")
-
+        prompt = gr.Textbox(label="Enter Pokémon Scene", placeholder="Ash battles with Team Rocket with Pikachu's Thunderbolt")
+        style = gr.Dropdown(["Anime Classic", "Modern 3D", "Cartoon"], label="Animation Style", value="Anime Classic")
+    
     duration = gr.Slider(1, 10, step=1, label="Video Duration (Seconds)", value=5)
     image = gr.Image(label="Upload an Image (Optional)", type="filepath")
 
-    generate_btn = gr.Button("Generate Pokémon Video")
-
-    output_video = gr.Video(label="Generated Pokémon Video")
+    generate_btn = gr.Button("Generate Pokémon Anime Video")
+    output_video = gr.Video(label="Generated Video")
 
-    generate_btn.click(generate_pokemon_video, inputs=[prompt, style, duration, image], outputs=output_video)
+    generate_btn.click(generate_pokemon_anime_video, inputs=[prompt, style, duration, image], outputs=output_video)
 
 iface.launch()