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
)