春联生成模型-中文-base实操手册批量生成100副春联并导出为PDF教程1. 引言春节贴春联是咱们的传统习俗但每年想新词儿是不是挺头疼的自己编吧文采不够网上找吧又觉得千篇一律。要是能有个工具输入“发财”、“平安”这样的祝福词就能自动生成一副对仗工整、寓意吉祥的春联那该多省事。今天要介绍的这个“春联生成模型-中文-base”就是这样一个帮你解决烦恼的AI工具。它基于达摩院AliceMind团队的大模型技术专门针对春联场景做了优化。你只需要给它两个字的祝福词比如“吉祥”、“如意”它就能在几秒钟内为你创作出与之相关的上下联和横批。更棒的是这个模型已经封装成了开箱即用的Web应用通过Gradio界面点点鼠标就能用。但如果你觉得一次生成一副不过瘾想批量生成几十上百副或者想把生成的春联整理成漂亮的PDF文档保存或打印那这篇教程就是为你准备的。在接下来的内容里我会手把手带你完成三件事第一快速把这个春联生成服务跑起来第二写一个简单的Python脚本实现批量生成100副不同主题的春联第三把这些春联自动排版并导出成一个美观的PDF文件。整个过程不需要高深的AI知识会用基本的Python代码就行。2. 环境准备与快速启动在开始批量生成之前我们得先把服务跑起来。根据提供的资料项目结构很清晰启动也非常简单。2.1 确认环境与模型首先确保你的环境里有Python 3.10或更高版本。你可以通过命令行检查python3 --version接下来是最关键的一步模型文件。根据说明模型需要预先放在这个路径下/root/ai-models/iic/spring_couplet_generation。请务必确认这个目录存在并且里面包含了模型所需的文件。如果模型路径不对服务是无法正常加载的。2.2 一键启动服务项目提供了两种启动方式推荐使用启动脚本更省心。方式一使用启动脚本这是最简单的方法。在项目根目录spring_couplet_generation/下直接运行./start.sh这个脚本会自动处理依赖安装和环境准备然后启动应用。方式二直接运行Python脚本如果你喜欢更直接的方式也可以运行python3 /root/spring_couplet_generation/app.py无论用哪种方式当你看到终端输出类似“Running on local URL: http://0.0.0.0:7860”的信息时就说明服务启动成功了。2.3 访问与测试打开你的浏览器在地址栏输入http://localhost:7860。你会看到一个简洁的网页界面中间有一个输入框。试着输入一个两字祝福词比如“安康”然后点击“提交”按钮。稍等片刻下方就会显示出生成的上联、下联和横批。界面上还有一个“复制”按钮可以一键复制生成的春联文本。手动测试成功说明我们的服务基础功能是完好的接下来就可以用代码来批量调用它了。3. 批量生成春联的Python脚本手动在网页上点100次显然不现实。我们需要写一个脚本自动向这个服务发送请求并保存结果。这里会用到Python的requests库来调用接口。3.1 理解服务接口虽然我们是通过网页访问但Gradio应用背后其实提供了一个API接口。我们可以直接向这个接口发送HTTP POST请求来生成春联。通过浏览器的开发者工具按F12切换到“网络”标签我们观察一次生成请求可以发现请求地址http://localhost:7860/api/predict/请求方法POST请求数据一个JSON格式的数据里面包含了我们输入的祝福词。3.2 编写批量生成脚本下面是一个完整的Python脚本示例它能够循环生成春联并保存到文件里。import requests import json import time import random # 服务地址 API_URL http://localhost:7860/api/predict/ # 准备一批祝福词你可以随意扩充这个列表 blessing_words [ 吉祥, 如意, 平安, 健康, 发财, 富贵, 幸福, 快乐, 团圆, 和睦, 兴旺, 昌盛, 顺心, 好运, 福气, 长寿, 美满, 甜蜜, 成功, 辉煌, 和谐, 文明, 民主, 富强, # 注意此处“民主”、“富强”为常见祝福词汇无特殊含义。 敬业, 诚信, 友善, 勤俭, 勇敢, 坚强, 智慧, 聪明, 美丽, 帅气, 青春, 活力, 阳光, 开朗, 幽默, 风趣, # ... 可以继续添加凑够或超过100个词 ] def generate_couplet(word): 生成一副春联 payload { data: [word] # Gradio API期望的格式 } headers { Content-Type: application/json } try: response requests.post(API_URL, datajson.dumps(payload), headersheaders, timeout30) response.raise_for_status() # 检查请求是否成功 result response.json() # Gradio返回的数据结构可能在‘data’字段里 if data in result: couplet_data result[data][0] # 通常是一个列表取第一个元素 # 假设返回格式是 [上联, 下联, 横批] if len(couplet_data) 3: upper_line couplet_data[0] lower_line couplet_data[1] horizontal couplet_data[2] return upper_line, lower_line, horizontal except requests.exceptions.RequestException as e: print(f请求失败祝福词‘{word}’: {e}) except (KeyError, IndexError, json.JSONDecodeError) as e: print(f解析结果失败祝福词‘{word}’: {e}) print(f原始返回: {result}) return None, None, None def batch_generate_couplets(num100): 批量生成春联 all_couplets [] # 如果祝福词不够就循环使用 words_to_use blessing_words * (num // len(blessing_words) 1) words_to_use words_to_use[:num] print(f开始批量生成 {num} 副春联...) for i, word in enumerate(words_to_use, 1): print(f正在生成第 {i} 副祝福词: {word}, end... ) upper, lower, horizontal generate_couplet(word) if upper and lower and horizontal: couplet_info { id: i, blessing_word: word, upper_line: upper, lower_line: lower, horizontal: horizontal } all_couplets.append(couplet_info) print(成功) else: print(失败) # 稍微延迟一下避免对服务造成太大压力 time.sleep(0.5) print(f生成完成成功 {len(all_couplets)} 副失败 {num - len(all_couplets)} 副。) return all_couplets def save_to_txt(couplets, filenamespring_couplets.txt): 保存春联到文本文件 with open(filename, w, encodingutf-8) as f: for c in couplets: f.write(f【第{c[id]:03d}副】 主题{c[blessing_word]}\n) f.write(f上联{c[upper_line]}\n) f.write(f下联{c[lower_line]}\n) f.write(f横批{c[horizontal]}\n) f.write(- * 40 \n) print(f春联已保存到文件: {filename}) if __name__ __main__: # 生成100副春联 my_couplets batch_generate_couplets(100) # 保存为文本文件 save_to_txt(my_couplets)脚本使用说明将上面的代码保存为一个.py文件比如batch_generate.py。确保你的春联生成服务正在运行http://localhost:7860可以访问。在命令行运行这个脚本python3 batch_generate.py。脚本会开始工作在终端打印生成进度最终将所有春联保存到spring_couplets.txt文件中。这个文本文件已经是一个不错的成果了但如果我们想要更正式、更适合打印或分享的格式就需要把它变成PDF。4. 将春联导出为精美PDF文本文件虽然方便但排版简单不够美观。我们可以使用reportlab这个Python库来创建排版精美的PDF文档。4.1 安装PDF生成库首先需要安装reportlab库pip install reportlab4.2 编写PDF导出脚本下面的脚本会读取上一步生成的文本文件或者直接使用内存中的春联数据创建一个每页显示一副春联的PDF。from reportlab.lib.pagesizes import A4 from reportlab.pdfgen import canvas from reportlab.pdfbase import pdfmetrics from reportlab.pdfbase.ttfonts import TTFont from reportlab.lib.utils import simpleSplit import os # 注册中文字体非常重要否则PDF无法显示中文 # 你需要指定一个系统中存在的TrueType中文字体文件路径 # 例如使用SimSun宋体或SimHei黑体 FONT_PATH /usr/share/fonts/truetype/wqy/wqy-microhei.ttc # 文泉驿微米黑一种常见开源字体 # 或者使用系统自带的例如 # FONT_PATH /System/Library/Fonts/PingFang.ttc # macOS # FONT_PATH C:/Windows/Fonts/simhei.ttf # Windows # 检查字体文件是否存在如果不存在尝试寻找其他常见字体 if not os.path.exists(FONT_PATH): # 尝试其他可能路径 common_paths [ /usr/share/fonts/truetype/wqy/wqy-microhei.ttc, /usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc, ./simhei.ttf # 可以将字体文件放在脚本同级目录 ] for path in common_paths: if os.path.exists(path): FONT_PATH path break else: print(警告未找到中文字体文件PDF可能无法正常显示中文。) FONT_PATH None if FONT_PATH: try: pdfmetrics.registerFont(TTFont(ChineseFont, FONT_PATH)) CHINESE_FONT ChineseFont except: print(注册中文字体失败将使用默认字体可能不支持中文。) CHINESE_FONT Helvetica else: CHINESE_FONT Helvetica def create_couplet_pdf(couplets, filenamespring_couplets.pdf): 创建春联PDF文档 c canvas.Canvas(filename, pagesizeA4) width, height A4 for idx, couplet in enumerate(couplets): # 开始新的一页 c.setFont(CHINESE_FONT, 24) c.setTitle(fAI生成春联集 - 第{idx1}副) # 1. 绘制页面顶部的主题和编号 theme_text f第{couplet[id]:03d}副 · 主题{couplet[blessing_word]} c.setFont(CHINESE_FONT, 16) c.setFillColorRGB(0.8, 0.2, 0.2) # 红色 c.drawCentredString(width / 2.0, height - 80, theme_text) # 2. 绘制装饰性横线 c.setStrokeColorRGB(0.9, 0.3, 0.3) c.setLineWidth(1) c.line(50, height - 100, width - 50, height - 100) # 3. 绘制横批在顶部 c.setFont(CHINESE_FONT, 28) c.setFillColorRGB(0, 0, 0) # 黑色 horizontal couplet[horizontal] # 横批通常从右向左写但我们按常规从左向右居中显示 c.drawCentredString(width / 2.0, height - 180, horizontal) # 4. 绘制上联右侧 c.setFont(CHINESE_FONT, 22) upper_line couplet[upper_line] # 将长文本拆分为多行春联一般每句7个字但模型可能生成更长的 upper_lines simpleSplit(upper_line, CHINESE_FONT, 22, width/2 - 100) upper_y_start height - 300 for i, line in enumerate(upper_lines): c.drawRightString(width / 2.0 - 60, upper_y_start - i*40, line) # 5. 绘制下联左侧 lower_line couplet[lower_line] lower_lines simpleSplit(lower_line, CHINESE_FONT, 22, width/2 - 100) lower_y_start height - 300 for i, line in enumerate(lower_lines): c.drawString(width / 2.0 60, lower_y_start - i*40, line) # 6. 绘制春联中间的装饰竖线模拟门框 c.setStrokeColorRGB(0.7, 0.7, 0.7) c.setLineWidth(0.5) c.line(width / 2.0, height - 250, width / 2.0, height - 450) # 7. 页面底部信息 c.setFont(CHINESE_FONT, 10) c.setFillColorRGB(0.5, 0.5, 0.5) page_info f由 春联生成模型-中文-base 生成 · 第 {idx1} 页 / 共 {len(couplets)} 页 c.drawCentredString(width / 2.0, 40, page_info) # 完成当前页开始下一页 c.showPage() # 保存PDF c.save() print(fPDF文档已生成: {filename}) print(f共计 {len(couplets)} 副春联{len(couplets)} 页。) def read_couplets_from_txt(filenamespring_couplets.txt): 从文本文件读取春联数据 couplets [] with open(filename, r, encodingutf-8) as f: lines f.readlines() current_couplet {} for line in lines: line line.strip() if line.startswith(【第): # 解析新的一副春联 if current_couplet: couplets.append(current_couplet) parts line.strip(【】).split(】 主题) current_couplet { id: int(parts[0].replace(第, ).replace(副, )), blessing_word: parts[1], upper_line: , lower_line: , horizontal: } elif line.startswith(上联): current_couplet[upper_line] line.replace(上联, ) elif line.startswith(下联): current_couplet[lower_line] line.replace(下联, ) elif line.startswith(横批): current_couplet[horizontal] line.replace(横批, ) # 忽略分隔线 if current_couplet: couplets.append(current_couplet) return couplets if __name__ __main__: # 方式一如果你有之前保存的txt文件可以读取它 # couplets_list read_couplets_from_txt(spring_couplets.txt) # 方式二直接使用批量生成脚本得到的数据需要将两个脚本合并或导入 # 这里假设我们直接调用批量生成函数需要将前面的脚本整合进来 from batch_generate import batch_generate_couplets, save_to_txt # 假设批量生成脚本叫 batch_generate.py print(开始批量生成并创建PDF...) # 生成春联数据 couplets_data batch_generate_couplets(20) # 先生成20副作为示例生成100副时间较长 # 保存文本备份 save_to_txt(couplets_data, couplets_for_pdf.txt) # 创建PDF create_couplet_pdf(couplets_data, 精美春联集.pdf)脚本使用说明确保已安装reportlab库。关键步骤指定正确的中文字体路径FONT_PATH变量。否则生成的PDF可能显示为乱码或方框。你可以根据自己系统的字体位置进行修改。这个脚本可以独立运行它会先调用批量生成函数生成20副作为演示然后创建PDF。你也可以先运行批量生成脚本得到spring_couplets.txt然后修改主函数部分使用read_couplets_from_txt函数来读取数据并生成PDF。运行脚本python3 create_pdf.py。生成的PDF文档每页显示一副春联包含主题、横批、上联和下联并进行了简单的排版美化更适合打印或电子阅读。5. 进阶技巧与问题排查掌握了基本操作后我们再来看看如何用得更好以及遇到问题怎么办。5.1 提升生成效果与效率祝福词的选择模型的输出质量与输入的两个字祝福词密切相关。选择寓意明确、常用的词汇如“福寿”、“康宁”往往比生僻词效果更好。你可以建立自己的“优质祝福词库”。批量生成的优化调整延迟脚本中time.sleep(0.5)是为了避免请求过快。如果服务稳定可以适当减小这个值以提高速度。使用多线程/异步如果需要生成上千副春联可以考虑使用concurrent.futures或asyncio来并发请求大幅缩短总时间。错误重试在网络不稳定或服务短暂异常时可以为失败的请求添加重试机制。PDF排版定制reportlab功能强大你可以轻松修改脚本中的坐标、颜色、字体大小来实现不同的排版风格比如一页显示多副春联或者添加更复杂的背景图案。5.2 常见问题与解决方法问题服务启动失败提示模型找不到或加载错误。解决确认模型文件是否严格按照要求放置在/root/ai-models/iic/spring_couplet_generation目录下。检查目录权限。问题访问http://localhost:7860无响应。解决首先检查服务是否真的启动成功查看终端有无报错。如果是远程服务器确保你是在服务器本地访问或者使用了正确的IP地址和端口映射。问题批量生成脚本报错Connection refused。解决确保春联生成服务app.py在脚本运行前已经启动并在7860端口监听。问题生成的PDF中文显示为方框。解决这是最常见的问题。确保FONT_PATH变量指向了一个真实存在的、包含中文的TrueType.ttf或.ttc字体文件。可以将字体文件拷贝到项目目录下然后修改路径为./your_font.ttf。问题生成的春联内容不太对仗或不通顺。解决这是模型本身能力的局限。可以尝试更换不同的祝福词。对于不满意的结果可以手动筛选和编辑。理解这是一个辅助创作工具其输出可以作为灵感来源不必追求完美。6. 总结回顾一下我们完成了从零开始使用“春联生成模型-中文-base”的完整旅程快速部署我们通过简单的命令启动了基于Gradio的Web服务并通过浏览器进行了手动测试。批量生成我们编写了Python脚本利用requests库自动化调用服务API成功生成了大量春联并将结果保存到了文本文件中。美化输出我们进一步使用reportlab库将文本格式的春联转换成了排版精美、适合打印和分享的PDF文档。这个过程不仅展示了如何操作一个具体的AI模型应用更提供了一个通用的思路将交互式AI服务与自动化脚本、结果后处理工具相结合可以极大地拓展其应用场景和实用性。你可以把这个方法用到其他地方。比如用AI文案生成模型批量创作节日祝福语再用脚本整理成贺卡或者用AI绘图模型生成一系列图片然后自动打包。关键在于把AI看作一个强大的“内容生成器”而我们用编程脚本作为“流水线控制器”和“包装工”。希望这份实操手册能帮你高效地创作出属于自己的AI春联集。无论是用于家庭装饰、朋友分享还是作为一份独特的数字年货都承载了一份科技带来的新年趣味。获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。
春联生成模型-中文-base实操手册：批量生成100副春联并导出为PDF教程
发布时间：2026/5/19 19:17:25
春联生成模型-中文-base实操手册批量生成100副春联并导出为PDF教程1. 引言春节贴春联是咱们的传统习俗但每年想新词儿是不是挺头疼的自己编吧文采不够网上找吧又觉得千篇一律。要是能有个工具输入“发财”、“平安”这样的祝福词就能自动生成一副对仗工整、寓意吉祥的春联那该多省事。今天要介绍的这个“春联生成模型-中文-base”就是这样一个帮你解决烦恼的AI工具。它基于达摩院AliceMind团队的大模型技术专门针对春联场景做了优化。你只需要给它两个字的祝福词比如“吉祥”、“如意”它就能在几秒钟内为你创作出与之相关的上下联和横批。更棒的是这个模型已经封装成了开箱即用的Web应用通过Gradio界面点点鼠标就能用。但如果你觉得一次生成一副不过瘾想批量生成几十上百副或者想把生成的春联整理成漂亮的PDF文档保存或打印那这篇教程就是为你准备的。在接下来的内容里我会手把手带你完成三件事第一快速把这个春联生成服务跑起来第二写一个简单的Python脚本实现批量生成100副不同主题的春联第三把这些春联自动排版并导出成一个美观的PDF文件。整个过程不需要高深的AI知识会用基本的Python代码就行。2. 环境准备与快速启动在开始批量生成之前我们得先把服务跑起来。根据提供的资料项目结构很清晰启动也非常简单。2.1 确认环境与模型首先确保你的环境里有Python 3.10或更高版本。你可以通过命令行检查python3 --version接下来是最关键的一步模型文件。根据说明模型需要预先放在这个路径下/root/ai-models/iic/spring_couplet_generation。请务必确认这个目录存在并且里面包含了模型所需的文件。如果模型路径不对服务是无法正常加载的。2.2 一键启动服务项目提供了两种启动方式推荐使用启动脚本更省心。方式一使用启动脚本这是最简单的方法。在项目根目录spring_couplet_generation/下直接运行./start.sh这个脚本会自动处理依赖安装和环境准备然后启动应用。方式二直接运行Python脚本如果你喜欢更直接的方式也可以运行python3 /root/spring_couplet_generation/app.py无论用哪种方式当你看到终端输出类似“Running on local URL: http://0.0.0.0:7860”的信息时就说明服务启动成功了。2.3 访问与测试打开你的浏览器在地址栏输入http://localhost:7860。你会看到一个简洁的网页界面中间有一个输入框。试着输入一个两字祝福词比如“安康”然后点击“提交”按钮。稍等片刻下方就会显示出生成的上联、下联和横批。界面上还有一个“复制”按钮可以一键复制生成的春联文本。手动测试成功说明我们的服务基础功能是完好的接下来就可以用代码来批量调用它了。3. 批量生成春联的Python脚本手动在网页上点100次显然不现实。我们需要写一个脚本自动向这个服务发送请求并保存结果。这里会用到Python的requests库来调用接口。3.1 理解服务接口虽然我们是通过网页访问但Gradio应用背后其实提供了一个API接口。我们可以直接向这个接口发送HTTP POST请求来生成春联。通过浏览器的开发者工具按F12切换到“网络”标签我们观察一次生成请求可以发现请求地址http://localhost:7860/api/predict/请求方法POST请求数据一个JSON格式的数据里面包含了我们输入的祝福词。3.2 编写批量生成脚本下面是一个完整的Python脚本示例它能够循环生成春联并保存到文件里。import requests import json import time import random # 服务地址 API_URL http://localhost:7860/api/predict/ # 准备一批祝福词你可以随意扩充这个列表 blessing_words [ 吉祥, 如意, 平安, 健康, 发财, 富贵, 幸福, 快乐, 团圆, 和睦, 兴旺, 昌盛, 顺心, 好运, 福气, 长寿, 美满, 甜蜜, 成功, 辉煌, 和谐, 文明, 民主, 富强, # 注意此处“民主”、“富强”为常见祝福词汇无特殊含义。 敬业, 诚信, 友善, 勤俭, 勇敢, 坚强, 智慧, 聪明, 美丽, 帅气, 青春, 活力, 阳光, 开朗, 幽默, 风趣, # ... 可以继续添加凑够或超过100个词 ] def generate_couplet(word): 生成一副春联 payload { data: [word] # Gradio API期望的格式 } headers { Content-Type: application/json } try: response requests.post(API_URL, datajson.dumps(payload), headersheaders, timeout30) response.raise_for_status() # 检查请求是否成功 result response.json() # Gradio返回的数据结构可能在‘data’字段里 if data in result: couplet_data result[data][0] # 通常是一个列表取第一个元素 # 假设返回格式是 [上联, 下联, 横批] if len(couplet_data) 3: upper_line couplet_data[0] lower_line couplet_data[1] horizontal couplet_data[2] return upper_line, lower_line, horizontal except requests.exceptions.RequestException as e: print(f请求失败祝福词‘{word}’: {e}) except (KeyError, IndexError, json.JSONDecodeError) as e: print(f解析结果失败祝福词‘{word}’: {e}) print(f原始返回: {result}) return None, None, None def batch_generate_couplets(num100): 批量生成春联 all_couplets [] # 如果祝福词不够就循环使用 words_to_use blessing_words * (num // len(blessing_words) 1) words_to_use words_to_use[:num] print(f开始批量生成 {num} 副春联...) for i, word in enumerate(words_to_use, 1): print(f正在生成第 {i} 副祝福词: {word}, end... ) upper, lower, horizontal generate_couplet(word) if upper and lower and horizontal: couplet_info { id: i, blessing_word: word, upper_line: upper, lower_line: lower, horizontal: horizontal } all_couplets.append(couplet_info) print(成功) else: print(失败) # 稍微延迟一下避免对服务造成太大压力 time.sleep(0.5) print(f生成完成成功 {len(all_couplets)} 副失败 {num - len(all_couplets)} 副。) return all_couplets def save_to_txt(couplets, filenamespring_couplets.txt): 保存春联到文本文件 with open(filename, w, encodingutf-8) as f: for c in couplets: f.write(f【第{c[id]:03d}副】 主题{c[blessing_word]}\n) f.write(f上联{c[upper_line]}\n) f.write(f下联{c[lower_line]}\n) f.write(f横批{c[horizontal]}\n) f.write(- * 40 \n) print(f春联已保存到文件: {filename}) if __name__ __main__: # 生成100副春联 my_couplets batch_generate_couplets(100) # 保存为文本文件 save_to_txt(my_couplets)脚本使用说明将上面的代码保存为一个.py文件比如batch_generate.py。确保你的春联生成服务正在运行http://localhost:7860可以访问。在命令行运行这个脚本python3 batch_generate.py。脚本会开始工作在终端打印生成进度最终将所有春联保存到spring_couplets.txt文件中。这个文本文件已经是一个不错的成果了但如果我们想要更正式、更适合打印或分享的格式就需要把它变成PDF。4. 将春联导出为精美PDF文本文件虽然方便但排版简单不够美观。我们可以使用reportlab这个Python库来创建排版精美的PDF文档。4.1 安装PDF生成库首先需要安装reportlab库pip install reportlab4.2 编写PDF导出脚本下面的脚本会读取上一步生成的文本文件或者直接使用内存中的春联数据创建一个每页显示一副春联的PDF。from reportlab.lib.pagesizes import A4 from reportlab.pdfgen import canvas from reportlab.pdfbase import pdfmetrics from reportlab.pdfbase.ttfonts import TTFont from reportlab.lib.utils import simpleSplit import os # 注册中文字体非常重要否则PDF无法显示中文 # 你需要指定一个系统中存在的TrueType中文字体文件路径 # 例如使用SimSun宋体或SimHei黑体 FONT_PATH /usr/share/fonts/truetype/wqy/wqy-microhei.ttc # 文泉驿微米黑一种常见开源字体 # 或者使用系统自带的例如 # FONT_PATH /System/Library/Fonts/PingFang.ttc # macOS # FONT_PATH C:/Windows/Fonts/simhei.ttf # Windows # 检查字体文件是否存在如果不存在尝试寻找其他常见字体 if not os.path.exists(FONT_PATH): # 尝试其他可能路径 common_paths [ /usr/share/fonts/truetype/wqy/wqy-microhei.ttc, /usr/share/fonts/opentype/noto/NotoSansCJK-Regular.ttc, ./simhei.ttf # 可以将字体文件放在脚本同级目录 ] for path in common_paths: if os.path.exists(path): FONT_PATH path break else: print(警告未找到中文字体文件PDF可能无法正常显示中文。) FONT_PATH None if FONT_PATH: try: pdfmetrics.registerFont(TTFont(ChineseFont, FONT_PATH)) CHINESE_FONT ChineseFont except: print(注册中文字体失败将使用默认字体可能不支持中文。) CHINESE_FONT Helvetica else: CHINESE_FONT Helvetica def create_couplet_pdf(couplets, filenamespring_couplets.pdf): 创建春联PDF文档 c canvas.Canvas(filename, pagesizeA4) width, height A4 for idx, couplet in enumerate(couplets): # 开始新的一页 c.setFont(CHINESE_FONT, 24) c.setTitle(fAI生成春联集 - 第{idx1}副) # 1. 绘制页面顶部的主题和编号 theme_text f第{couplet[id]:03d}副 · 主题{couplet[blessing_word]} c.setFont(CHINESE_FONT, 16) c.setFillColorRGB(0.8, 0.2, 0.2) # 红色 c.drawCentredString(width / 2.0, height - 80, theme_text) # 2. 绘制装饰性横线 c.setStrokeColorRGB(0.9, 0.3, 0.3) c.setLineWidth(1) c.line(50, height - 100, width - 50, height - 100) # 3. 绘制横批在顶部 c.setFont(CHINESE_FONT, 28) c.setFillColorRGB(0, 0, 0) # 黑色 horizontal couplet[horizontal] # 横批通常从右向左写但我们按常规从左向右居中显示 c.drawCentredString(width / 2.0, height - 180, horizontal) # 4. 绘制上联右侧 c.setFont(CHINESE_FONT, 22) upper_line couplet[upper_line] # 将长文本拆分为多行春联一般每句7个字但模型可能生成更长的 upper_lines simpleSplit(upper_line, CHINESE_FONT, 22, width/2 - 100) upper_y_start height - 300 for i, line in enumerate(upper_lines): c.drawRightString(width / 2.0 - 60, upper_y_start - i*40, line) # 5. 绘制下联左侧 lower_line couplet[lower_line] lower_lines simpleSplit(lower_line, CHINESE_FONT, 22, width/2 - 100) lower_y_start height - 300 for i, line in enumerate(lower_lines): c.drawString(width / 2.0 60, lower_y_start - i*40, line) # 6. 绘制春联中间的装饰竖线模拟门框 c.setStrokeColorRGB(0.7, 0.7, 0.7) c.setLineWidth(0.5) c.line(width / 2.0, height - 250, width / 2.0, height - 450) # 7. 页面底部信息 c.setFont(CHINESE_FONT, 10) c.setFillColorRGB(0.5, 0.5, 0.5) page_info f由 春联生成模型-中文-base 生成 · 第 {idx1} 页 / 共 {len(couplets)} 页 c.drawCentredString(width / 2.0, 40, page_info) # 完成当前页开始下一页 c.showPage() # 保存PDF c.save() print(fPDF文档已生成: {filename}) print(f共计 {len(couplets)} 副春联{len(couplets)} 页。) def read_couplets_from_txt(filenamespring_couplets.txt): 从文本文件读取春联数据 couplets [] with open(filename, r, encodingutf-8) as f: lines f.readlines() current_couplet {} for line in lines: line line.strip() if line.startswith(【第): # 解析新的一副春联 if current_couplet: couplets.append(current_couplet) parts line.strip(【】).split(】 主题) current_couplet { id: int(parts[0].replace(第, ).replace(副, )), blessing_word: parts[1], upper_line: , lower_line: , horizontal: } elif line.startswith(上联): current_couplet[upper_line] line.replace(上联, ) elif line.startswith(下联): current_couplet[lower_line] line.replace(下联, ) elif line.startswith(横批): current_couplet[horizontal] line.replace(横批, ) # 忽略分隔线 if current_couplet: couplets.append(current_couplet) return couplets if __name__ __main__: # 方式一如果你有之前保存的txt文件可以读取它 # couplets_list read_couplets_from_txt(spring_couplets.txt) # 方式二直接使用批量生成脚本得到的数据需要将两个脚本合并或导入 # 这里假设我们直接调用批量生成函数需要将前面的脚本整合进来 from batch_generate import batch_generate_couplets, save_to_txt # 假设批量生成脚本叫 batch_generate.py print(开始批量生成并创建PDF...) # 生成春联数据 couplets_data batch_generate_couplets(20) # 先生成20副作为示例生成100副时间较长 # 保存文本备份 save_to_txt(couplets_data, couplets_for_pdf.txt) # 创建PDF create_couplet_pdf(couplets_data, 精美春联集.pdf)脚本使用说明确保已安装reportlab库。关键步骤指定正确的中文字体路径FONT_PATH变量。否则生成的PDF可能显示为乱码或方框。你可以根据自己系统的字体位置进行修改。这个脚本可以独立运行它会先调用批量生成函数生成20副作为演示然后创建PDF。你也可以先运行批量生成脚本得到spring_couplets.txt然后修改主函数部分使用read_couplets_from_txt函数来读取数据并生成PDF。运行脚本python3 create_pdf.py。生成的PDF文档每页显示一副春联包含主题、横批、上联和下联并进行了简单的排版美化更适合打印或电子阅读。5. 进阶技巧与问题排查掌握了基本操作后我们再来看看如何用得更好以及遇到问题怎么办。5.1 提升生成效果与效率祝福词的选择模型的输出质量与输入的两个字祝福词密切相关。选择寓意明确、常用的词汇如“福寿”、“康宁”往往比生僻词效果更好。你可以建立自己的“优质祝福词库”。批量生成的优化调整延迟脚本中time.sleep(0.5)是为了避免请求过快。如果服务稳定可以适当减小这个值以提高速度。使用多线程/异步如果需要生成上千副春联可以考虑使用concurrent.futures或asyncio来并发请求大幅缩短总时间。错误重试在网络不稳定或服务短暂异常时可以为失败的请求添加重试机制。PDF排版定制reportlab功能强大你可以轻松修改脚本中的坐标、颜色、字体大小来实现不同的排版风格比如一页显示多副春联或者添加更复杂的背景图案。5.2 常见问题与解决方法问题服务启动失败提示模型找不到或加载错误。解决确认模型文件是否严格按照要求放置在/root/ai-models/iic/spring_couplet_generation目录下。检查目录权限。问题访问http://localhost:7860无响应。解决首先检查服务是否真的启动成功查看终端有无报错。如果是远程服务器确保你是在服务器本地访问或者使用了正确的IP地址和端口映射。问题批量生成脚本报错Connection refused。解决确保春联生成服务app.py在脚本运行前已经启动并在7860端口监听。问题生成的PDF中文显示为方框。解决这是最常见的问题。确保FONT_PATH变量指向了一个真实存在的、包含中文的TrueType.ttf或.ttc字体文件。可以将字体文件拷贝到项目目录下然后修改路径为./your_font.ttf。问题生成的春联内容不太对仗或不通顺。解决这是模型本身能力的局限。可以尝试更换不同的祝福词。对于不满意的结果可以手动筛选和编辑。理解这是一个辅助创作工具其输出可以作为灵感来源不必追求完美。6. 总结回顾一下我们完成了从零开始使用“春联生成模型-中文-base”的完整旅程快速部署我们通过简单的命令启动了基于Gradio的Web服务并通过浏览器进行了手动测试。批量生成我们编写了Python脚本利用requests库自动化调用服务API成功生成了大量春联并将结果保存到了文本文件中。美化输出我们进一步使用reportlab库将文本格式的春联转换成了排版精美、适合打印和分享的PDF文档。这个过程不仅展示了如何操作一个具体的AI模型应用更提供了一个通用的思路将交互式AI服务与自动化脚本、结果后处理工具相结合可以极大地拓展其应用场景和实用性。你可以把这个方法用到其他地方。比如用AI文案生成模型批量创作节日祝福语再用脚本整理成贺卡或者用AI绘图模型生成一系列图片然后自动打包。关键在于把AI看作一个强大的“内容生成器”而我们用编程脚本作为“流水线控制器”和“包装工”。希望这份实操手册能帮你高效地创作出属于自己的AI春联集。无论是用于家庭装饰、朋友分享还是作为一份独特的数字年货都承载了一份科技带来的新年趣味。获取更多AI镜像想探索更多AI镜像和应用场景访问 CSDN星图镜像广场提供丰富的预置镜像覆盖大模型推理、图像生成、视频生成、模型微调等多个领域支持一键部署。
