Gradio 案例——将文本文件转为词云图

2024-06-02 12:28
文章标签 案例 云图 gradio 转为 文本文件

本文主要是介绍Gradio 案例——将文本文件转为词云图，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

文章目录

  • Gradio 案例——将文本文件转为词云图
    • 界面截图
    • 依赖安装
    • 项目目录结构
    • 代码

Gradio 案例——将文本文件转为词云图

  • 利用 word_cloud 库，将文本文件转为词云图
  • 更完整、丰富的示例项目见 GitHub - AlionSSS/wordcloud-webui: The web UI for word_cloud(text to word cloud picture converter)

界面截图

image.png

依赖安装

  • 新建一个虚拟环境 Python 3.9.16
  • 依赖
    • $ pip install gradio==4.29 -i "https://pypi.doubanio.com/simple/"
    • $ pip install wordcloud==1.9.3 -i "https://pypi.doubanio.com/simple/"
    • $ pip install jieba==0.42.1 -i "https://pypi.doubanio.com/simple/"

项目目录结构

wordcloud-webui         # 目录
--/resources             # 资源目录
--/consts.py             # py文件，常量
--/gradio_interfaces.py  # py文件，Gradio视图
--/jieba_util.py         # py文件，工具库文件
--/lib_word_cloud.py     # py文件，工具库文件
--/main.py               # py文件，入口

代码

  • main.py
from gradio_interfaces import ifaceif __name__ == "__main__":iface.launch()
  • lib_word_cloud.py
from wordcloud import WordCloud, ImageColorGenerator
import numpy as np
from PIL import Imagefrom consts import *def text2wordcount_normal(text: str,background_color: str = "white",margin = 2,min_font_size = 4,max_font_size = 200,font_path = None,width: int = 400,height: int = 200,
):if not background_color or "" == str(background_color).strip():background_color = "white"if not min_font_size or  min_font_size < 1:min_font_size = 4if not max_font_size or max_font_size < 4:max_font_size = 200    if not font_path or "" == str(font_path).strip():font_path = DEFAULT_FONT_PATHif not width or width < 1:width = 400if not height or height < 1:height = 200 # Generate a word cloud imagewordcloud = WordCloud(font_path=font_path,width=width, height=height, background_color=background_color, max_words=2000, margin=margin, min_font_size=min_font_size, max_font_size=max_font_size, random_state=42).generate(text)return wordcloud.to_image()def text2wordcount_mask(text: str,background_color: str = "white",margin = 2,min_font_size = 4,max_font_size = 200,font_path = None,mask_image = None,mask_color = None,contour_width=3,contour_color="steelblue",
):if not background_color or "" == str(background_color).strip():background_color = "white"if not min_font_size or  min_font_size < 1:min_font_size = 4if not max_font_size or max_font_size < 4:max_font_size = 200   if not font_path or "" == str(font_path).strip():font_path = DEFAULT_FONT_PATHif not contour_width or contour_width < 0:contour_width = 3      if not contour_color or "" == str(contour_color).strip():contour_color = "steelblue"# mask_colorif mask_color is not None:image_colors = ImageColorGenerator(mask_color, True)else:image_colors = ImageColorGenerator(mask_image, True)# Generate a word cloud imagewordcloud = WordCloud(font_path=font_path,mask=mask_image,background_color=background_color,color_func=image_colors,contour_width=contour_width,contour_color=contour_color,max_words=2000, margin=margin, min_font_size=min_font_size, max_font_size=max_font_size, random_state=42).generate(text)return wordcloud.to_image()
  • jieba_util.py
import jieba
# jieba.enable_parallel(4)from consts import *# The function for processing text with Jieba
def jieba_processing_txt(text, userdict_list=['阿Ｑ', '孔乙己', '单四嫂子']):if userdict_list is not None:for word in userdict_list:jieba.add_word(word)mywordlist = []seg_list = jieba.cut(text, cut_all=False)liststr = "/ ".join(seg_list)with open(STOPWORDS_PATH, encoding='utf-8') as f_stop:f_stop_text = f_stop.read()f_stop_seg_list = f_stop_text.splitlines()for myword in liststr.split('/'):if not (myword.strip() in f_stop_seg_list) and len(myword.strip()) > 1:mywordlist.append(myword)return ' '.join(mywordlist)
  • gradio_interfaces.py
import gradio as grimport lib_word_cloud
import jieba_utilfrom consts import *def service_text2wc(text_file,text_lang,text_dict: str,background_color,margin,max_font_size,min_font_size,font_file,width,height,mask_image,mask_color,contour_width,contour_color,
):if not text_file:gr.Warning(f"请传入正确的文本文件！")returnif margin < 0 :gr.Warning(f"字体间隔配置不合法！")returnif min_font_size < 0 or max_font_size < 0 or min_font_size > max_font_size:gr.Warning(f"字体大小配置不合法！")returntry:with open(file=text_file.name, encoding="utf-8") as file:text = file.read()if text_lang == '中文':gr.Info(f"选择了中文，将使用Jieba库解析文本！")userdict_list = []if text_dict is not None:# userdict_list = map(lambda w: w.strip(), text_dict.split(", "))userdict_list = [w.strip() for w in text_dict.split(",")]text = jieba_util.jieba_processing_txt(text, userdict_list)font_path = font_file.name if font_file else Noneif mask_image is not None:return lib_word_cloud.text2wordcount_mask(text,background_color,margin,min_font_size,max_font_size,font_path,mask_image,mask_color,contour_width,contour_color,)else:return lib_word_cloud.text2wordcount_normal(text, background_color, margin,min_font_size,max_font_size,font_path, width, height)except Exception as e:print(e)raise gr.Error("文本转词云图时，发生异常：" + str(e))js = """
function createGradioAnimation() {var container = document.createElement('div');container.id = 'gradio-animation';container.style.fontSize = '2em';container.style.fontWeight = 'bold';container.style.textAlign = 'center';container.style.marginBottom = '20px';var text = '欢迎使用“词云转换器”!';for (var i = 0; i < text.length; i++) {(function(i){setTimeout(function(){var letter = document.createElement('span');letter.style.opacity = '0';letter.style.transition = 'opacity 0.5s';letter.innerText = text[i];container.appendChild(letter);setTimeout(function() {letter.style.opacity = '1';}, 50);}, i * 200);})(i);}var gradioContainer = document.querySelector('.gradio-container');gradioContainer.insertBefore(container, gradioContainer.firstChild);return 'Animation created';
}
"""with gr.Blocks(title="词云转换器", js=js) as iface:with gr.Row():with gr.Column():with gr.Group():with gr.Row():input_text_file = gr.File(label="待处理的文本文件（必填）")with gr.Column():gr.Label(label="Tips", value="请传入正常可读的文本文件，如以.txt结尾的文档", color="#fee2e2")gr.File(value=EXAMPLE_TEXT_FILE, label="文本文件的样例")input_text_lang = gr.Radio(label="文本语言模式", choices=["中文", "英文"], value="中文")input_text_dict = gr.Textbox(label="自定义分词词典（可选）", info="中文模式使用，多个词之间用英文逗号分隔，例如'阿Ｑ, 孔乙己, 单四嫂子'")with gr.Tab("普通模式"):with gr.Row():input_width = gr.Number(value=400, label="生成图像的宽", minimum=1)input_height = gr.Number(value=200, label="生成图像的高", minimum=1)gr.Label(label="Tips", value="使用该模式时，记得清理掉“Mask模式”下的“Mask图像”", color="#fee2e2")with gr.Tab("Mask模式"):with gr.Row():input_contour_width = gr.Number(value=3, label="轮廓线的粗细", minimum=0)input_contour_color = gr.Textbox(value="steelblue", label="轮廓线的颜色")with gr.Row():input_mask_image = gr.Image(label="Mask图像（决定词云的形状、颜色、宽高）")input_mask_color = gr.Image(label="若传入该图，则词云的颜色由该图决定")# gr.Image(value=EXAMPLE_MASK_IMAGE_PATH, label="Mask图像的样例", interactive=False)gr.Gallery(value=[EXAMPLE_MASK_IMAGE_PATH, EXAMPLE_MASK_IMAGE_PATH, EXAMPLE_MASK_IMAGE_PATH], label="Mask图像的样例", interactive=False)with gr.Column():with gr.Group():with gr.Row():with gr.Group():input_bg_color = gr.Textbox(value="white", label="词云图的背景色（默认为'white'）")input_margin = gr.Number(value=2, label="字体间隔（默认为'2'）", minimum=0)with gr.Row():input_min_font_size = gr.Number(value=4, label="字体大小-最小值", minimum=1)input_max_font_size = gr.Number(value=200, label="字体大小-最大值", minimum=4)    input_font_file = gr.File(label="词云图的字体文件（可选，如otf文件）")format_radio = gr.Radio(choices=["png", "jpeg", "webp", "bmp", "tiff"], label="词云图像格式", value="png")submit_button = gr.Button("开始处理", variant="primary")output_image = gr.Image(label="词云图", format="png")def fix_format(x):output_image.format = x return Noneformat_radio.change(fn=fix_format, inputs=format_radio)submit_button.click(fn=service_text2wc,inputs=[input_text_file,input_text_lang,input_text_dict,input_bg_color,input_margin,input_max_font_size,input_min_font_size,input_font_file,input_width,input_height,input_mask_image,input_mask_color,input_contour_width,input_contour_color,],outputs=output_image,)
  • consts.py，记得修改下下面文件的地址，和resource目录对应
# 样例文本
EXAMPLE_TEXT_FILE = r".\wordcloud-webui\resources\CalltoArms.txt"
# MASK图像样例
EXAMPLE_MASK_IMAGE_PATH = r".\wordcloud-webui\resources\parrot_mask.png "
# 分词器的 stop word 库
STOPWORDS_PATH = r".\wordcloud-webui\resources\stopwords_cn_en.txt"
# 词云图的默认字体
DEFAULT_FONT_PATH = r".\wordcloud-webui\resources\SourceHanSerifK-Light.otf"
  • resources 目录
    • parrot_mask.png parrot_mask.png
    • CalltoArms.txt https://github.com/amueller/word_cloud/blob/main/examples/wc_cn/CalltoArms.txt
    • SourceHanSerifK-Light.otf https://github.com/amueller/word_cloud/blob/main/examples/fonts/SourceHanSerif/SourceHanSerifK-Light.otf
    • stopwords_cn_en.txt https://github.com/amueller/word_cloud/blob/main/examples/wc_cn/stopwords_cn_en.txt

这篇关于Gradio 案例——将文本文件转为词云图的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/1023964。

相关文章

Hadoop企业开发案例调优场景

Hadoop企业开发案例调优场景

需求 （1）需求：从1G数据中，统计每个单词出现次数。服务器3台，每台配置4G内存，4核CPU，4线程。 （2）需求分析： 1G / 128m = 8个MapTask；1个ReduceTask；1个mrAppMaster 平均每个节点运行10个 / 3台 ≈ 3个任务（4    3    3） HDFS参数调优 （1）修改：hadoop-env.sh export HDFS_NAMENOD
阅读更多...
性能分析之MySQL索引实战案例

性能分析之MySQL索引实战案例

文章目录 一、前言二、准备三、MySQL索引优化四、MySQL 索引知识回顾五、总结 一、前言 在上一讲性能工具之 JProfiler 简单登录案例分析实战中已经发现SQL没有建立索引问题，本文将一起从代码层去分析为什么没有建立索引？ 开源ERP项目地址：https://gitee.com/jishenghua/JSH_ERP 二、准备 打开IDEA找到登录请求资源路径位置
阅读更多...
深入探索协同过滤：从原理到推荐模块案例

深入探索协同过滤：从原理到推荐模块案例

文章目录 前言一、协同过滤1. 基于用户的协同过滤（UserCF）2. 基于物品的协同过滤（ItemCF）3. 相似度计算方法 二、相似度计算方法1. 欧氏距离2. 皮尔逊相关系数3. 杰卡德相似系数4. 余弦相似度 三、推荐模块案例1.基于文章的协同过滤推荐功能2.基于用户的协同过滤推荐功能 前言     在信息过载的时代，推荐系统成为连接用户与内容的桥梁。本文聚焦于
阅读更多...
【区块链 + 人才服务】可信教育区块链治理系统 | FISCO BCOS应用案例

【区块链 + 人才服务】可信教育区块链治理系统 | FISCO BCOS应用案例

伴随着区块链技术的不断完善，其在教育信息化中的应用也在持续发展。利用区块链数据共识、不可篡改的特性， 将与教育相关的数据要素在区块链上进行存证确权，在确保数据可信的前提下，促进教育的公平、透明、开放，为教育教学质量提升赋能，实现教育数据的安全共享、高等教育体系的智慧治理。 可信教育区块链治理系统的顶层治理架构由教育部、高校、企业、学生等多方角色共同参与建设、维护，支撑教育资源共享、教学质量评估、
阅读更多...
客户案例：安全海外中继助力知名家电企业化解海外通邮困境

客户案例：安全海外中继助力知名家电企业化解海外通邮困境

1、客户背景 广东格兰仕集团有限公司（以下简称“格兰仕”），成立于1978年，是中国家电行业的领军企业之一。作为全球最大的微波炉生产基地，格兰仕拥有多项国际领先的家电制造技术，连续多年位列中国家电出口前列。格兰仕不仅注重业务的全球拓展，更重视业务流程的高效与顺畅，以确保在国际舞台上的竞争力。 2、需求痛点 随着格兰仕全球化战略的深入实施，其海外业务快速增长，电子邮件成为了关键的沟通工具。
阅读更多...
【区块链 + 人才服务】区块链集成开发平台 | FISCO BCOS应用案例

【区块链 + 人才服务】区块链集成开发平台 | FISCO BCOS应用案例

随着区块链技术的快速发展，越来越多的企业开始将其应用于实际业务中。然而，区块链技术的专业性使得其集成开发成为一项挑战。针对此，广东中创智慧科技有限公司基于国产开源联盟链 FISCO BCOS 推出了区块链集成开发平台。该平台基于区块链技术，提供一套全面的区块链开发工具和开发环境，支持开发者快速开发和部署区块链应用。此外，该平台还可以提供一套全面的区块链开发教程和文档，帮助开发者快速上手区块链开发。
阅读更多...
STL经典案例（四）——实验室预约综合管理系统（项目涉及知识点很全面，内容有点多，耐心看完会有收获的！）

STL经典案例（四）——实验室预约综合管理系统（项目涉及知识点很全面，内容有点多，耐心看完会有收获的！）

项目干货满满，内容有点过多，看起来可能会有点卡。系统提示读完超过俩小时，建议分多篇发布，我觉得分篇就不完整了，失去了这个项目的灵魂 一、需求分析 高校实验室预约管理系统包括三种不同身份：管理员、实验室教师、学生 管理员：给学生和实验室教师创建账号并分发 实验室教师：审核学生的预约申请 学生：申请使用实验室 高校实验室包括：超景深实验室（可容纳10人）、大数据实验室（可容纳20人）、物联网实验
阅读更多...
（入门篇）JavaScript 网页设计案例浅析-简单的交互式图片轮播

（入门篇）JavaScript 网页设计案例浅析-简单的交互式图片轮播

网页设计已经成为了每个前端开发者的必备技能，而 JavaScript 作为前端三大基础之一，更是为网页赋予了互动性和动态效果。本篇文章将通过一个简单的 JavaScript 案例，带你了解网页设计中的一些常见技巧和技术原理。今天就说一说一个常见的图片轮播效果。相信大家在各类电商网站、个人博客或者展示页面中，都看到过这种轮播图。它的核心功能是展示多张图片，并且用户可以通过点击按钮，左右切换图片。
阅读更多...
单精度浮点数按存储格式转为整数的程序

单精度浮点数按存储格式转为整数的程序

///#include<cstdio>//-----------------union int_char{unsigned char ch[4];float i;};void out_put(union int_char x)//x86是小端对其模式，即最数据的最低位存储在地址的最低位上。{printf("单精度浮点数值为：%f\n",x.i,x.i);printf("存储位置从左到右
阅读更多...
SpringMVC的第一个案例 Helloword 步骤

SpringMVC的第一个案例 Helloword 步骤

第一步：web.xml配置 <?xml version="1.0" encoding="UTF-8"?> <web-app version="2.5" xmlns="http://java.sun.com/xml/ns/javaee" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocati
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2