from PIL import Image, ImageFont
from handright import Template, handwrite
import docx
def generate(text_file,ttf_file,template):
    # 自动缩进排版，如果已在word里设置缩进可以注释本段
    # indent_size控制缩进,file_path文档路径

    word_num=34
    def get_text(file_path, indent_size=4):
        doc = docx.Document(file_path)
        texts = []
        indent = ' '*indent_size

        for paragraph in doc.paragraphs:
            text=indent+paragraph.text+'\n'
            for i in range(len(text)//word_num+1):
                # print(text[i*word_num:(i+1)*word_num])
                texts.append(text[i*word_num:(i+1)*word_num])
        return ''.join(texts)

    # 根目录下的word文档
    text = get_text(text_file)
    print(text)
    ttf_file = ttf_file

    temp=template(ttf_file)
    images = handwrite(text, temp)
    for i, im in enumerate(images):
        assert isinstance(im, Image.Image)
        # im.show()

        print('文件{}.jpg 已生成！'.format(i))
        im.save('output/{}.png'.format(i))  # 生成的图片生成在当前目录下
        # break