Update app.py

app.py

@@ -1,58 +1,22 @@
 from diffusers import DiffusionPipeline
 import gradio as gr
-import moviepy.editor as mpy
-import os
 
-# Load the pipeline
+# Load the DiffusionPipeline
 pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-video-diffusion-img2vid")
 
-def generate_video(prompt, num_frames=10, duration=5, audio_path=None):
-    # Generate images based on the prompt
-    frames = [pipe(prompt).images[0] for _ in range(num_frames)]
+# Define the function for Gradio
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
 
-    # Save frames to create a video
-    frame_files = []
-    for i, img in enumerate(frames):
-        file_path = f"frame_{i}.png"
-        img.save(file_path)
-        frame_files.append(file_path)
-
-    # Create video from frames
-    clip = mpy.ImageSequenceClip(frame_files, fps=num_frames / duration)
-    
-    # Add audio if provided
-    if audio_path and os.path.exists(audio_path):
-        audio = mpy.AudioFileClip(audio_path)
-        clip = clip.set_audio(audio.set_duration(clip.duration))
-
-    # Save the video file
-    video_path = "generated_video.mp4"
-    clip.write_videofile(video_path, codec="libx264")
-    
-    # Clean up frame files
-    for file_path in frame_files:
-        os.remove(file_path)
-    
-    return video_path
-
-# Gradio interface
-def gradio_interface(prompt, num_frames, duration, audio_file):
-    video_path = generate_video(prompt, num_frames, duration, audio_file.name if audio_file else None)
-    return video_path
-
-# Gradio setup
+# Set up Gradio interface
 interface = gr.Interface(
-    fn=gradio_interface,
-    inputs=[
-        gr.inputs.Textbox(label="Prompt"),
-        gr.inputs.Slider(minimum=5, maximum=50, step=1, default=10, label="Number of Frames"),
-        gr.inputs.Slider(minimum=1, maximum=15, step=1, default=5, label="Duration (seconds)"),
-        gr.inputs.File(label="Audio File (Optional)"),
-    ],
-    outputs="video",
-    title="Generate Video from Text Prompt with Audio",
-    description="Generate a video based on a text prompt, with optional background music."
+    fn=generate_image,
+    inputs="text",
+    outputs="image",
+    title="Stable Video Diffusion Image Generator",
+    description="Generate an image based on your prompt using Stable Diffusion."
 )
 
-# Launch Gradio app
+# Launch the Gradio app
 interface.launch()