Add Description

Videobook/Videobook.py

@@ -14,50 +14,14 @@ class Videobook:
   def get_sentences(self, story):
     return self.preprocessor(story)
 
-  # def generate_voice(self, story, sentences , path = 'tts.mp3'):
-  #   for i,n in enumerate(sentences):
-  #       tts=gTTS(n,lang='en')
-  #       tts.save('tts'+str(i)+'.mp3')
-  #   lgth=[]
-  #   for i in range(len(sentences)):
-  #       lgth.append(MP3('tts'+str(i)+'.mp3').info.length)
-  #       os.remove(os.path.join(os.getcwd(),'tts'+str(i)+'.mp3'))
-  #   tts=gTTS(story,lang='en')
-  #   tts.save(path)
-  #   return lgth
-
   def generate_imgs(self, sentences, steps):
     imgs = []
     for sentence in sentences:
       sentence['pos'] = self.style + ' of ' + sentence['pos'] + ', ' + self.tags
       imgs.append(self.pipe.generate(prompt = sentence['pos'], negative_prompt = sentence['neg'], num_inference_steps = steps))
-    return imgs
-
-  # def addBuffer(self, imgs, lgth):
-  #   imgs_buff = []
-  #   for i,img in enumerate(imgs):
-  #     for j in range(ceil(lgth[i] * self.fps)):
-  #       imgs_buff.append(img)
-  #   return imgs_buff
-
-  # def imgs_to_video(self, imgs, video_name='video.mp4'):
-  #   video_dims = (imgs[0].width, imgs[0].height)
-  #   fourcc = cv2.VideoWriter_fourcc(*'DIVX')    
-  #   video = cv2.VideoWriter(video_name, fourcc, self.fps, video_dims)
-  #   for img in imgs:
-  #     tmp_img = img.copy()
-  #     video.write(cv2.cvtColor(np.array(tmp_img), cv2.COLOR_RGB2BGR))
-  #   video.release()
-
-  # def make_video(self, imgs, lengths, video_name = "finished_video.mp4"):
-  #   self.imgs_to_video(self.addBuffer(imgs, lengths), 'test_video.mp4')
-  #   input_audio = ffmpeg.input(os.path.join(os.getcwd(),'tts.mp3'))
-  #   input_video = ffmpeg.input(os.path.join(os.getcwd(),'test_video.mp4'))
-  #   ffmpeg.concat(input_video, input_audio, v=1, a=1).output(video_name).run(overwrite_output=True)
-    
+    return imgs    
 
-  def generate(self, story, api_key, fps, style, tags, model, steps):
-    self.fps = fps
+  def generate(self, story, api_key, style, tags, model, steps):
     self.style = style
     self.tags = tags
     if model == "Stable Diffusion v2.1":

app.py

@@ -3,6 +3,7 @@ import gradio as gr
 
 gen = Videobook()
 with gr.Blocks() as demo:
+  gr.Description("Videobook", "A tool to generate Narrated Videos from Story")
   with gr.Row():
     with gr.Column():
       story = gr.inputs.Textbox(lines = 5, label = "Story")
@@ -10,9 +11,7 @@ with gr.Blocks() as demo:
       tags = gr.inputs.Textbox(default = "high quality, 3d render", label = "Tags")
       style = gr.Dropdown(["Cartoon", "Anime Style", "Realistic Image"], value = "Cartoon", label = "Style")
       model = gr.Dropdown(["Stable Diffusion v2.1", "Kadinsky"], value = "Stable Diffusion v2.1", label = "Model")
-      with gr.Row():
-        steps = gr.Radio([25, 50], value = 50, label = "Steps")
-        fps = gr.Radio([10, 24, 60], value = 10, label = "FPS")
+      steps = gr.Radio([25, 50], value = 50, label = "Steps")
     output = gr.outputs.Video()
   run = gr.Button(label = "Generate Video")
   run.click(gen.generate, inputs = [story, api_key, fps, style, tags, model, steps], outputs = output)