1 Star 0 Fork 0

yuyangup/ComfyUI-IF_AI_tools

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
text_cleanup.py 1.18 KB
一键复制 编辑 原始数据 按行查看 历史
ImpactFrames 提交于 2024-05-12 22:36 . Add files via upload
import re
def process_text(generated_text, remove_weights=True, remove_author=True):
if remove_author:
generated_text = re.sub(r'\bby:.*', '', generated_text)
if remove_weights:
generated_text = re.sub(r'\(([^)]*):[\d\.]*\)', r'\1', generated_text)
generated_text = re.sub(r'(\w+):[\d\.]*(?=[ ,]|$)', r'\1', generated_text)
# Remove markup tags
generated_text = re.sub(r'<[^>]*>', '', generated_text)
# Remove lonely symbols and formatting
generated_text = re.sub(r'(?<=\s):(?=\s)', '', generated_text)
generated_text = re.sub(r'(?<=\s);(?=\s)', '', generated_text)
generated_text = re.sub(r'(?<=\s),(?=\s)', '', generated_text)
generated_text = re.sub(r'(?<=\s)#(?=\s)', '', generated_text)
# Clean up extra spaces and punctuation
generated_text = re.sub(r'\s{2,}', ' ', generated_text)
generated_text = re.sub(r'\.,', ',', generated_text)
generated_text = re.sub(r',,', ',', generated_text)
# Remove audio tags
if '<audio' in generated_text:
print(f"iF_prompt_MKR: Audio has been generated.")
generated_text = re.sub(r'<audio.*?>.*?</audio>', '', generated_text)
return generated_text.strip()
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/yuyangup/ComfyUI-IF_AI_tools.git
git@gitee.com:yuyangup/ComfyUI-IF_AI_tools.git
yuyangup
ComfyUI-IF_AI_tools
ComfyUI-IF_AI_tools
IF_AI_tools_RAG_linux_mac

搜索帮助

0d507c66 1850385 C8b1a773 1850385