Update app.py

app.py

@@ -1,27 +1,42 @@
+import gradio as gr
 from diffusers import StableDiffusionPipeline
-import os
 import ffmpeg
+import os
 
-model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
+# Load the Stable Diffusion model
+model = StableDiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-lite")
 
 def generate_video(prompt):
+    # Generate frames
     frames = []
-    for i in range(10):  # Generate 10 frames
+    for i in range(5):  # Generate 5 frames
         image = model(prompt).images[0]
         frame_path = f"frame_{i}.png"
         image.save(frame_path)
         frames.append(frame_path)
 
+    # Combine frames into a video
     output_video = "output.mp4"
     (
         ffmpeg
-        .input("frame_%d.png", framerate=1)
+        .input("frame_%d.png", framerate=1)  # Adjust framerate for longer/shorter videos
         .output(output_video)
         .run(overwrite_output=True)
     )
 
-    # Clean up frames
+    # Clean up temporary frame files
     for frame in frames:
         os.remove(frame)
 
-    return output_video
+    return output_video  # Return the path to the video file
+
+# Gradio interface
+with gr.Blocks() as demo:
+    gr.Markdown("# AI Video Generator")
+    prompt_input = gr.Textbox(label="Enter your video prompt", placeholder="Type something creative...")
+    video_output = gr.File(label="Download Your Video")
+    generate_button = gr.Button("Generate Video")
+
+    generate_button.click(fn=generate_video, inputs=prompt_input, outputs=video_output)
+
+demo.launch()