Spaces:
Build error
Build error
File size: 1,630 Bytes
9ba4ba9 39a17cb 9ba4ba9 32410f8 9ba4ba9 32410f8 9ba4ba9 e46e599 9ba4ba9 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 | import os
import random
from PIL import Image
from moviepy import ImageSequenceClip
from dalle import text2im
from typing import List
def generate_image(prompt: str, size: tuple = (512, 512)) -> Image:
"""
Generates an image from the given prompt.
"""
response = text2im({
"prompt": prompt,
"size": "1024x1024"
})
image_data = response["images"][0]
image = Image.open(image_data)
image = image.resize(size)
return image
def save_images(image_list: List[Image], folder="generated_images") -> List[str]:
"""
Saves generated images to the specified folder.
"""
if not os.path.exists(folder):
os.makedirs(folder)
paths = []
for i, image in enumerate(image_list):
path = os.path.join(folder, f"frame_{i}.png")
image.save(path)
paths.append(path)
return paths
def create_video(image_paths: List[str], output="output_video.mp4", fps=12):
"""
Creates a video from the list of image paths.
"""
clip = ImageSequenceClip(image_paths, fps=fps)
clip.write_videofile(output, codec="libx264")
def generate_video(prompt: str, num_frames: int = 20):
"""
Generates a video from a prompt.
"""
images = []
for i in range(num_frames):
modified_prompt = f"{prompt}, transitioning scene, frame {i+1}/{num_frames}"
images.append(generate_image(modified_prompt))
image_paths = save_images(images)
create_video(image_paths)
if __name__ == "__main__":
prompt = "A beautiful futuristic cityscape at night with neon lights and flying cars."
generate_video(prompt)
|