当前位置：首页 > news >正文

Pixel Couplet Gen详细步骤：从ModelScope拉取模型到Streamlit界面渲染

news 2026/4/19 9:44:16

Pixel Couplet Gen详细步骤：从ModelScope拉取模型到Streamlit界面渲染

1. 项目概述

Pixel Couplet Gen是一款融合传统春节文化与现代像素艺术风格的AI春联生成器。该项目基于ModelScope平台的大语言模型，通过Streamlit框架构建了独特的8-bit复古游戏界面，为用户提供沉浸式的春联创作体验。

核心特点：

模型能力：利用ModelScope的文本生成模型，自动创作符合传统格律的春联
界面风格：完全自定义的像素游戏UI，包含动态卷轴、高饱和配色等视觉元素
交互设计：模拟红白机操作反馈，增强生成过程的趣味性

2. 环境准备

2.1 基础环境配置

确保您的开发环境满足以下要求：

# Python版本要求 python --version # 需要3.8+ pip --version # 需要最新版pip # 安装核心依赖 pip install streamlit==1.30.0 pip install modelscope pip install pillow # 用于图像处理

2.2 ModelScope账号设置

访问ModelScope官网注册账号
获取API访问凭证
在本地配置访问密钥：

from modelscope import snapshot_download model_dir = snapshot_download('模型名称', cache_dir='本地缓存路径')

3. 模型集成步骤

3.1 从ModelScope拉取模型

使用以下代码加载春联生成模型：

from modelscope.pipelines import pipeline from modelscope.utils.constant import Tasks # 初始化文本生成管道 couplet_pipeline = pipeline( task=Tasks.text_generation, model='damo/nlp_gpt3_text-generation_chinese-base' )

3.2 模型调用封装

创建生成春联的核心函数：

def generate_couplet(prompt): """ 生成春联的核心函数 :param prompt: 用户输入的主题词 :return: 包含上联、下联、横批的字典 """ full_prompt = f"请创作一副关于{prompt}的春节对联，要求：" full_prompt += "1. 上联和下联各7个字 2. 横批4个字 3. 符合传统对仗要求" result = couplet_pipeline(full_prompt) return parse_couplet_result(result['text']) def parse_couplet_result(text): """解析模型输出的对联结果""" # 使用正则表达式提取对联各部分 import re pattern = r"上联：(.*?)\n下联：(.*?)\n横批：(.*)" match = re.search(pattern, text) return { 'upper': match.group(1), 'lower': match.group(2), 'horizontal': match.group(3) }

4. Streamlit界面开发

4.1 基础界面搭建

创建主界面文件app.py：

import streamlit as st # 设置页面配置 st.set_page_config( page_title="Pixel Couplet Gen", layout="wide", initial_sidebar_state="collapsed" ) # 加载自定义CSS def load_css(): with open("assets/pixel_style.css") as f: st.markdown(f"<style>{f.read()}</style>", unsafe_allow_html=True) # 主界面布局 def main(): load_css() st.title("🎮 像素皇城 · 灵蛇贺岁") # 用户输入区域 with st.container(): col1, col2 = st.columns([3,1]) with col1: user_input = st.text_input("", placeholder="输入你的新年愿望...") with col2: if st.button("🧧 生成春联"): if user_input: generate_and_display(user_input) else: st.warning("请输入主题词") if __name__ == "__main__": main()

4.2 像素风格CSS实现

创建assets/pixel_style.css文件：

/* 基础像素风格设置 */ body { background-color: #222034; font-family: 'ZCOOL QingKe HuangYou', cursive; } /* 按钮像素效果 */ .stButton>button { border: 4px solid #ff0044; background-color: #29adff; color: #fff1e8; font-weight: bold; text-shadow: 2px 2px #000; box-shadow: 4px 4px 0px #000; } /* 输入框像素效果 */ .stTextInput>div>div>input { background-color: #fff1e8; border: 4px solid #ff77a8; padding: 8px; }

5. 功能整合与部署

5.1 完整功能实现

更新app.py中的生成函数：

def generate_and_display(theme): """生成并展示像素春联""" with st.spinner('像素精灵正在创作中...'): result = generate_couplet(theme) # 对联展示容器 with st.container(): st.markdown(f""" <div class="couplet-container"> <div class="horizontal">{result['horizontal']}</div> <div class="scroll upper">{result['upper']}</div> <div class="scroll lower">{result['lower']}</div> </div> """, unsafe_allow_html=True) # 添加下载按钮 st.download_button( label="📥 保存像素春联", data=generate_image(result), file_name="pixel_couplet.png", mime="image/png" )