#!/usr/bin/env python
# encoding: utf-8

import jieba
import numpy as np
import PIL.Image as Image
from wordcloud import WordCloud

class wordCloud:
'''
This is a main Class, the file contains all documents.
One document contains paragraphs that have several sentences
It loads the original file and converts the original file to new content
Then the new content will be saved by this class
'''
def __init__(self):
self.bg_img = 'assets/picture.jpeg'
self.word_path = 'assets/word.txt'

def hello(self):
'''
This is a welcome speech
:return: self
'''
print('*' * 50)
print(' ' * 20 + '詞云制作')
print(' ' * 5 + 'Author: autofelix Date: 2022-01-17 13:14')
print('*' * 50)
return self

def run(self):
'''
The program entry
'''
with open(self.word_path, 'r') as f:
word = f.read()

cut_word = ' '.join(jieba.cut(word))
color_mask = np.array(Image.open(self.bg_img))

word_cloud = WordCloud(
# 設置字體，不指定就會出現(xiàn)亂碼
font_path='/System/Library/Fonts/PingFang.ttc',
# 設置背景色
background_color='white',
# 詞云形狀
mask=color_mask,
# 允許最大詞匯
max_words=120,
# 最大號字體
max_font_size=2000
).generate(cut_word)

word_cloud.to_file('word_cloud.jpg')
im = word_cloud.to_image()
im.show()


if __name__ == '__main__':
wordCloud().hello().run()