draw_wordcloud.py

from nltk.corpus import stopwords
import matplotlib.pyplot as plt
from wordcloud import WordCloud 

def draw_wordcloud(msgs: list):
	"""
	Draw wordcloud for visualization of the most used words 
	during conversation
	args: 
	-----
	msgs: list: chat content
	
	Returns:
	-------
	a wordcloud with the most used words having the 
		highest font value 
	"""
	allWords = ' '.join([twts for twts in msgs])
	wordCloud_img = WordCloud(width=500, height=250, random_state=21,
				  max_words=200, mode='RGBA',max_font_size=170,
				  stopwords=wordcloud.STOPWORDS, scale=9,
                        	  min_word_length=4).generate(allWords)
	
	plt.figure(figsize=(20, 12))
	plt.imshow(wordCloud_img, interpolation="bilinear")
	plt.axis('off')
	plt.tight_layout(); 
	plt.title('most used words', size=20)
	plt.savefig('wordcloud.jpg')
	plt.show()