write text on images

gistfile1.py

	import os
	from PIL import Image
	from PIL import ImageFont
	from PIL import ImageDraw 

	def list_files(path):
	    # returns a list of names (with extension, without full path) of all files 
	    # in folder path
	    files = []
	    for name in os.listdir(path):
	        if os.path.isfile(os.path.join(path, name)):
	            files.append(name)
	    return files
	
	tagmap = {}
	# load labeled data
	f = open('../nfl_commercials/jia_result')
	for line in f:
		array = line.split("\t")
		tagContentStr = array[2].strip()[1:-1]
		tagContent = tagContentStr.split(",")
		tags = {};
		for singletagstr in tagContent:
			tag_score = singletagstr.strip()[1:]
			if len(tag_score.split(":")) != 2:
				# print singletagstr, tag_score, len(tag_score.split(":"))
				continue
			tag   = tag_score.split(":")[0]
			score = tag_score.split(":")[1].strip()[:6]
			# print tag, score
			tags[tag] = score
		tagmap[ array[0] ] = tags
		# print array[0]
		# print array[0], tagContent[0].strip()[1:], tagContent[1].strip()[1:], tagContent[2].strip()[1:]
	f.close()

	# images = list_files("../nfl_commercials")
	# for singleimage in images:
	# 	if singleimage.endswith(".jpg"):
	# 		print singleimage
	for key, tagSet in tagmap.items():
		img = Image.open("../nfl_commercials/" + key)
		draw = ImageDraw.Draw(img)
		font = ImageFont.truetype("Roboto-Regular.ttf", 20)
		yheight = 10
		counter = 0;
		# print tagSet, type(tagSet)
		for tag, score in tagSet.items():
			width, height = font.getsize(tag)
			draw.text((10, yheight), tag + ' : ' + score, (255,0,0),font=font)
			yheight += height * 1.1
			if counter > 3:
				break
			counter += 1
		img.save('output/' + key)
		print key, tagSet