videos_to_gif.py

#!/usr/bin/python

import sys, os, re, subprocess, shutil, yaml, pysrt
from PIL import Image, ImageFont, ImageDraw
from slugify import slugify

directory = "screenshots"
gif_dir = "gifs"
font = ImageFont.truetype("font/DejaVuSansCondensed-BoldOblique.ttf", 14)

def striptags(data):
  # I'm a bad person, don't ever do this.
  # Only okay, because of how basic the tags are.
  p = re.compile(r'<.*?>')
  return p.sub('', data)

def drawText(draw, x, y, text, font):
  # black outline
  draw.text((x-1, y),text,(0,0,0),font=font)
  draw.text((x+1, y),text,(0,0,0),font=font)
  draw.text((x, y-1),text,(0,0,0),font=font)
  draw.text((x, y+1),text,(0,0,0),font=font)

  # white text
  draw.text((x, y),text,(255,255,255),font=font)

def makeGif(video, starts, ends, strings, output):
  if not os.path.exists(gif_dir):
    os.makedirs(gif_dir)

  for index in range(0, len(starts)):
    if not os.path.exists(directory):
      os.makedirs(directory)

    string = strings[index]
    start = starts[index]
    end = ends[index]
    text = striptags(string).split("\n")

    subprocess.call(['avconv', '-i', video, '-vf', 'scale=w=400:h=-1', '-r', '15', '-ss', start, '-t', end, os.path.join(directory, 'image-%05d.png')])

    file_names = sorted((fn for fn in os.listdir(directory)))
    images = []

    for f in file_names:
      image = Image.open(os.path.join(directory,f))
      draw = ImageDraw.Draw(image)

      # reddit tells me this patten sucks, but I like it
      try:
        image_size
      except NameError:
        image_size = image.size

      # multiple lines in text
      if len(text) == 2:
        # at most 2?
        text_size = font.getsize(text[0])
        x = (image_size[0]/2) - (text_size[0]/2)
        y = image_size[1] - (2*text_size[1]) - 5 # padding
        drawText(draw, x, y, text[0], font)

        text_size = font.getsize(text[1])
        x = (image_size[0]/2) - (text_size[0]/2)
        y += text_size[1]
        drawText(draw, x, y, text[1], font)
      else:
        text_size = font.getsize(text[0])
        x = (image_size[0]/2) - (text_size[0]/2)
        y = image_size[1] - text_size[1] - 5 # padding
        drawText(draw, x, y, text[0], font)

      image.save(os.path.join(directory,f))

    subprocess.call(["convert", '-loop', '0', os.path.join(directory, '*.png'), os.path.join(gif_dir, "temp"+str(index)+".gif")])
    shutil.rmtree(directory)
  
  subprocess.call(["convert", '-loop', '0', os.path.join(gif_dir, '*.gif'), output])
  shutil.rmtree(gif_dir)
  

def generateAllGifs():
  stream = file('files.yml', 'r')
  data = yaml.load(stream)
  
  if "outpath" in data:
    outpath = data["outpath"]
  else:
    outpath = ""

  for file_data in data["files"]:
    video_file_path = file_data["video"]
    sub_file_path = file_data["subs"]

    if "encoding" in file_data:
      sub_encoding = file_data["encoding"]
    else:
      sub_encoding = "utf-8"

    subs = pysrt.open(sub_file_path, encoding=sub_encoding)

    # generate a gif for every line of dialogue
    for sub in subs:
      # 00:00:00,000 => 00:00:00.000
      start = str(sub.start).replace(',', '.')
      end = str(sub.end - sub.start).replace(',', '.')

      gif_filename = os.path.join(outpath, slugify(sub.text) + ".gif")
      
      if os.path.isfile(gif_filename):
        next
      else:
        print "generating " + gif_filename + "..."
        makeGif(video_file_path, start, end, sub.text, gif_filename)

if __name__ == '__main__':
  generateAllGifs()