Update app.py

app.py

@@ -5,28 +5,49 @@ from PIL import Image
 import tempfile
 import imageio
 
-# Use a public AnimateDiff-compatible model (Realistic Vision base)
+# Load the model
 model_id = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
 pipe = AnimateDiffPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
 
-def generate_video(image, prompt):
+def estimate_time(num_frames):
+    est_seconds = int(num_frames * 2.5)  # rough multiplier
+    return f"Estimated time: ~{est_seconds} seconds"
+
+def generate_video(image, prompt, num_frames):
+    status = "Generating..."
     image = image.convert("RGB").resize((512, 512))
-    result = pipe(prompt=prompt, image=image, num_frames=16, guidance_scale=7.5)
+
+    result = pipe(prompt=prompt, image=image, num_frames=num_frames, guidance_scale=7.5)
     frames = result.frames
 
     video_path = tempfile.NamedTemporaryFile(suffix=".mp4", delete=False).name
     imageio.mimsave(video_path, frames, fps=8)
-    return video_path
-
-gr.Interface(
-    fn=generate_video,
-    inputs=[
-        gr.Image(type="pil", label="Upload Image"),
-        gr.Textbox(label="Describe Motion (Prompt)")
-    ],
-    outputs=gr.Video(label="Generated Video"),
-    title="🌀 Image + Prompt to Animation",
-    description="Upload a still image and describe how you want it animated!"
-).launch()
+
+    status = f"Done! ({num_frames} frames)"
+    return video_path, status
+
+with gr.Blocks() as demo:
+    gr.Markdown("# 🌀 Image + Prompt to Video Generator")
+    with gr.Row():
+        image_input = gr.Image(type="pil", label="Upload Image")
+        prompt_input = gr.Textbox(label="Describe Motion (Prompt)")
+    with gr.Row():
+        num_frames_slider = gr.Slider(8, 32, value=16, step=8, label="Number of Frames")
+        time_output = gr.Textbox(label="⏱️ Estimated Time", interactive=False)
+    with gr.Row():
+        generate_btn = gr.Button("🎬 Generate Video")
+        status_output = gr.Textbox(label="🔄 Status", interactive=False)
+    video_output = gr.Video(label="🎥 Output Video")
+
+    # Events
+    num_frames_slider.change(fn=estimate_time, inputs=num_frames_slider, outputs=time_output)
+
+    generate_btn.click(
+        fn=generate_video,
+        inputs=[image_input, prompt_input, num_frames_slider],
+        outputs=[video_output, status_output]
+    )
+
+demo.launch()