Creating a Word Cloud Using Python

import numpy as np
import pandas as pd
from os import path
from PIL import Image
from wordcloud import WordCloud, STOPWORDS, ImageColorGenerator
import matplotlib.pyplot as plt
import string
from scipy.ndimage import gaussian_gradient_magnitude
%matplotlib inline

script_file = 'data/JurassicPark-Final.txt'

with open(script_file,'r') as f:
    lines = f.readlines()
    for line in lines:
	words = line.split()
	for word in words:
	    exclude = set(string.punctuation)
	    stripped_word = ''.join(ch for ch in word if ch not in exclude) # Exclude punctuation
	    if stripped_word != "":
	        word_list.append(stripped_word)

text = " ".join(word for word in word_list) #this is now a huge string containing the entire script!

logo_mask = np.array(Image.open("img/logo_large.png"))
mask = logo_mask # left over from some other testing


edges = np.mean([gaussian_gradient_magnitude(mask[:, :, i] / 255., 2) for i in range(3)], axis=0)
mask[edges > 0.08] = 255

plt.imshow(mask)
plt.show()

# Create stopword list:
stopwords = set(STOPWORDS)
stopwords.update(["youre", "Theyre", "Ill", "well","Im","Thats", "Id","Dont", "Theres","Ive",
                 "Didnt","Hes"]) #ignore some boring words

# Create a word cloud image
wc = WordCloud(stopwords = stopwords,max_words=5000, mask=mask, max_font_size=100, random_state=20,
               relative_scaling=0.5)

# Generate a wordcloud
wc.generate(text)

#create coloring from image
image_colors = ImageColorGenerator(mask)
wc.recolor(color_func=image_colors)
plt.figure(figsize=(10, 10))
plt.imshow(wc, interpolation="bilinear")


image = Image.open('img/logo_large.png')# add the logo back in so you can see it a little better
plt.imshow(image, alpha=0.2)

plt.axis("off")

wc.to_file("img/word_cloud_jurassic.png") #only saves word cloud
plt.savefig("img/word_cloud_jurassic2.png", bbox_inches = 'tight') # saves everything