词云生成工具制作教程：从零开始构建你的专属词云工具

import os
from flask import Flask, render_template, request, send_file, redirect, url_for
from wordcloud import WordCloud
import numpy as np
from PIL import Image
import io
app = Flask(
__name__
)
UPLOAD_FOLDER = 'uploads'
os.makedirs(UPLOAD_FOLDER, exist_ok=True)
def get_mask(shape, custom_img_path=None):
    if custom_img_path:
        mask = np.array(Image.open(custom_img_path))
    else:
        # 根据不同的形状生成简单的遮罩
        if shape == 'circle':
            # 生成圆形遮罩
            x, y = np.ogrid[:500, :500]
            center = (250, 250)
            mask = (x - center[0]) ** 2 + (y - center[1]) ** 2 > 240**2
            mask = 255 * mask.astype(int)
        elif shape == 'square':
            mask = None  # 无需遮罩，默认方形效果
        elif shape == 'heart':
            # 简单的心形遮罩示例（实际中可使用预先设计好的图片）
            x, y = np.ogrid[:500, :500]
            mask = ((x - 250)**2 + (y - 250)**2 - 10000)  # 仅作示例，请根据需要设计更合理的遮罩
            mask = np.where(mask < 0, 0, 255).astype(np.uint8)
        else:
            mask = None
    return mask
def get_color_func(color_choice):
    # 返回一个简单的颜色函数，根据选择返回固定颜色
    def color_func(word, font_size, position, orientation, random_state=None, **kwargs):
        return color_choice
    return color_func
@app.route('/', methods=['GET', 'POST'])
def index():
    preview_url = None
    if request.method == 'POST':
        text = request.form.get('text', '')
        color_choice = request.form.get('color', 'black')
        shape = request.form.get('shape', 'square')
        custom_file = request.files.get('custom_image')
        custom_img_path = None
        # 保存用户上传的图片
        if custom_file and custom_file.filename:
            custom_img_path = os.path.join(UPLOAD_FOLDER, custom_file.filename)
            custom_file.save(custom_img_path)

        mask = get_mask(shape, custom_img_path)
        wc = WordCloud(width=500, height=500, mask=mask, background_color='white',
                       color_func=get_color_func(color_choice))
        wc.generate(text)

        # 将生成的词云保存到内存中
        img_io = io.BytesIO()
        wc.to_image().save(img_io, 'PNG')
        img_io.seek(0)
        # 将图片存储到 session 或者临时路径（此处为了简单起见直接返回图片）
        return send_file(img_io, mimetype='image/png')

    return render_template('index.html', preview_url=preview_url)
if 
name
 == '
__main__
':
    app.run(debug=True)

2.前端页面（templates/index.html）

<!DOCTYPE html>
<html lang="zh-CN">
<head>
    <meta charset="UTF-8">
    <title>图生云工具</title>
</head>
<body>
    <h1>图生云工具</h1>
    <form method="POST" enctype="multipart/form-data">
        <div>
            <label for="text">输入文本：</label><br>
            <textarea name="text" id="text" rows="10" cols="50" placeholder="请输入要生成词云的文本"></textarea>
        </div>
        <div>
            <label for="color">选择颜色：</label>
            <select name="color" id="color">
                <option value="black">黑色</option>
                <option value="red">红色</option>
                <option value="blue">蓝色</option>
                <option value="green">绿色</option>
            </select>
        </div>
        <div>
            <label for="shape">选择形状：</label>
            <select name="shape" id="shape">
                <option value="circle">圆形</option>
                <option value="square">方形</option>
                <option value="heart">心形</option>
                <option value="custom">自定义</option>
            </select>
        </div>
        <div id="customImageDiv" style="display: none;">
            <label for="custom_image">上传自定义图片：</label>
            <input type="file" name="custom_image" id="custom_image" accept="image/*">
        </div>
        <div>
            <button type="submit">生成词云</button>
        </div>
    </form>
    <div>
        <!-- 此处可以用于实时预览生成的词云图片 -->
        {% if preview_url %}
        <h2>预览：</h2>
        <img src="{{ preview_url }}" alt="词云预览">
        {% endif %}
    </div>
    <script>
        // 根据形状选择显示或隐藏自定义图片上传区域
        document.getElementById('shape').addEventListener('change', function() {
            if (this.value === 'custom') {
                document.getElementById('customImageDiv').style.display = 'block';
            } else {
                document.getElementById('customImageDiv').style.display = 'none';
            }
        });
    </script>
</body>
</html>

3.说明

文本输入与参数选择： 用户在页面中输入文本，并选择对应的颜色和形状。如果选择“自定义”，则会显示上传图片的选项。
词云生成： 后端接收到表单后，根据选项生成词云。
- 若用户上传了自定义图片，则使用该图片作为遮罩；否则根据选定形状调用 get_mask 方法生成对应遮罩。
- 使用 color_func 设置词云中所有词汇的颜色为选定颜色。
实时预览： 为简化示例，这里提交表单后直接返回生成的词云图片。实际开发中可使用 AJAX 或 WebSocket 实现无刷新预览。

这个示例可以作为基础，在此基础上你还可以扩展更多功能，例如动态调整字体、词频过滤、更多配色方案等。

扩展阅读：

顶级10款在线词云生成器推荐：打造独具匠心的文字云艺术	https://blog.csdn.net/moton2017/article/details/146420233
词云生成工具制作教程：从零开始构建你的专属词云工具	https://blog.csdn.net/moton2017/article/details/146419170