import numpy as np

from PIL import Image     # converting images into arrays
import matplotlib as mpl
import matplotlib.pyplot as plt

# import package and its set of stopwords
from wordcloud import WordCloud, STOPWORDS

# remove stopwords
stopwords = set(STOPWORDS)

# we can add any word that we don't want visualize
stopwords.add('word')

# open the file and read it into a variable
words = open('words.txt', 'r').read()

# instantiate a word cloud object
new_wc = WordCloud(
    background_color='white',
    max_words=2000,
    stopwords=stopwords
)

# generate the word cloud
new_wc.generate(words)

# display the word cloud
plt.imshow(new_wc, interpolation='bilinear')
plt.axis('off')
plt.show()

# we can resize it
fig = plt.figure()
fig.set_figwidth(14)     # set width
fig.set_figheight(18)     # set height

# display the cloud
plt.imshow(new_wc, interpolation='bilinear')
plt.axis('off')
plt.show()

# We can use an image as background instead the rectangle when instantiate the object
new_img = np.array(Image.open('image.png'))

new_wc = WordCloud(
    background_color='white',
    max_words=2000,
    mask=new_img,
    stopwords=stopwords
)