Spaces:
Build error
Build error
File size: 2,514 Bytes
d5ba4e3 6249bc9 beab381 6249bc9 5f22376 6249bc9 beab381 6249bc9 cc7fff5 82c7c76 cc7fff5 82c7c76 cc7fff5 |
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 gradio as gr
from typing import Dict
from moviepy.editor import VideoFileClip, concatenate_videoclips
class VideoCreator:
def __init__(self,
tts_pipeline,
image_pipeline) -> None:
self.tts_pipeline = tts_pipeline
self.image_pipeline = image_pipeline
def create_video(self, scenes: Dict) -> str:
videos_dict = {}
for index, scene in enumerate(scenes):
video_scene = self._create_video_from_scene(scenes[scene])
videos_dict[index] = video_scene
merged_video = self._merge_videos(videos_dict)
return merged_video
def _create_video_from_scene(self, scene: Dict) -> str:
audio_file = self._get_audio_from_text(scene["Summary"])
bg_image = self._get_bg_image_from_description(scene["Illustration"])
video = gr.make_waveform(audio=audio_file,
bg_image=bg_image)
return video
def _get_audio_from_text(self, voice_over: str) -> str:
self.tts_pipeline.tts_to_file(text=voice_over,
file_path="output.wav")
return "output.wav"
def _get_bg_image_from_description(self, img_desc: str):
images = self.image_pipeline(img_desc)
print("Image generated!")
image_output = images.images[0]
image_output.save("img.png")
return "img.png"
def _merge_videos(self, videos_dict: Dict) -> str:
videos_to_concatenate = []
for video in range(len(videos_dict)):
video_clip = VideoFileClip(videos_dict[video])
videos_to_concatenate.append(video_clip)
final_video = concatenate_videoclips(videos_to_concatenate)
try:
final_video.write_videofile("final_video.mp4",
threads=4)
print("Saved .mp4 without Exception at final_video.mp4")
return "final_video.mp4"
except IndexError:
# Short by one frame, so get rid on the last frame:
final_video = final_video.subclip(t_end=(video_clip.duration - 1.0/final_video.fps))
final_video.write_videofile("final_video.mp4",
threads=4)
print("Saved .mp4 after Exception at final_video.mp4")
return "final_video.mp4"
except Exception as e:
print("Exception {} was raised!!".format(e))
return "final_video.mp4" |