前言
时间过得好快,一眨眼2024
年就来了。这两年对互联网冲击最大的想必就是大语言模型的AIGC
能力了。从国外的GPT
到国内的文心一言
、混元
、盘古
等等,AIGC
的出现大大的提升了我们的生产力,现在GPT
逐渐也变成了我每天都会使用到的工具。
那趁着新年即将来临,我们今天就使用AIGC
去生成春节必不可少的元素——春联。 链接
注册
对于大模型接口来说,我这边选择的是国产的讯飞星火大模型。选择它的原因绝对不是它现在可以领取免费的token
,而是我觉得我们需要支持国产软件自主化(手动狗头),如果你不喜欢用这个也可以使用别的,思路都是一样的。
首先打开它的官网 xinghuo.xfyun.cn/sparkapi 注册一个账号,然后点击领取token
就好了
领取到了之后就可以对着它的文档来使用API
了。
构造鉴权API
我们是通过websocket
去使用星火的大模型api
,在连接之前则需要构造一个鉴权参数。这个参数主要是为了安全以及用户身份的鉴权吧,这里主要还是要根据它的文档里面的规则去构造,实现代码如下:
const getWebsocketUrl = () => { var apiKey = APIKey; var apiSecret = APISecret; var url = "wss://spark-api.xf-yun.com/v3.1/chat"; var host = location.host; var date = new Date().toGMTString(); var algorithm = "hmac-sha256"; var headers = "host date request-line"; var signatureOrigin = `host: ${host}\ndate: ${date}\nGET /v3.1/chat HTTP/1.1`; var signatureSha = CryptoJS.HmacSHA256(signatureOrigin, apiSecret); var signature = CryptoJS.enc.Base64.stringify(signatureSha); var authorizationOrigin = `api_key="${apiKey}", algorithm="${algorithm}", headers="${headers}", signature="${signature}"`; var authorization = btoa(authorizationOrigin); url = `${url}?authorization=${authorization}&date=${date}&host=${host}`; return url; };
生成春联
在拿到上面的鉴权URL
之后,我们就可以使用大模型的接口来生成春联了。
这段时间其实也用大模型接口去做了一些其他的事情,对于大模型来说,prompt
可以算是比较核心的东西。一个精准的prompt
可以让大模型更好的理解我们的意图,并给出更符合我们要求的答案。
我们先用antd
来搭建一个简单的表单,对于主题或者上下联字数这种可以从表单中获取值然后动态拼接到prompt
中
下面就是我用来生成春联的一个prompt
,仅供参考:
const prompt = `你是一个编写春联的专家,下面我将会给你一些要求,你帮我编写一幅春联,需要有美好寓意,要求如下: 主题:${theme} 上下联字数:${count} 以json格式返回,返回示例如下: { "top":"知廉标五德", "bottom":"报午必三鸣", "banner":"新春", } `;
对于星火的API
来说,使用的时候需要注意以下几点:
- 回复是分段的,我们需要把结果拼起来
status
为2
时表示最后一段- 回复完之后会断开连接
拿到了连接的URL
,又有了prompt
,那么接下来的代码实现逻辑就很清晰了,具体代码如下:
const resultStr = useRef(""); const [result, setResult] = useState({}); const [loading, setLoading] = useState(false); const send = async () => { setLoading(true); const url = getWebsocketUrl(); const instance = new WebSocket(url); instance.onopen = async () => { const { theme, count } = await form.validateFields(); resultStr.current = ""; const prompt = `你是一个编写春联的专家,下面我将会给你一些要求,你帮我编写一幅春联,需要有美好寓意,要求如下: 主题:${theme} 上下联字数:${count} 以json格式返回,返回示例如下: { "top":"知廉标五德", "bottom":"报午必三鸣", "banner":"新春", } `; const params = { header: { app_id: APPID, uid: "jayliang", }, parameter: { chat: { domain: "generalv3", temperature: 0.5, max_tokens: 1024, }, }, payload: { message: { text: [{ role: "user", content: prompt }], }, }, }; instance.send(JSON.stringify(params)); }; instance.onmessage = (msg) => { const data = JSON.parse(msg.data); const status = data.payload.choices.status; const content = data.payload.choices.text[0].content; resultStr.current += content; if (status === 2) { setResult(JSON.parse(resultStr.current)); setLoading(false); } }; };
最后拿到的结果就是一个json
,拿到这个json
我们就可以很方便的渲染了。渲染的时候可以注意一些细节对于春联来说,文字的排版应该是从上往下的,所以可以加上这一段css
:
.contentItem { writing-mode: vertical-rl; text-orientation: mixed; white-space: nowrap; padding: 40px 0; }
而且横批应该是从右到左读的,所以可以加上一个反转字符串:
result.banner.split("").reverse().join("")
那现在就来看一下整体的效果吧: 链接
优化字体
对于春联来说,一般会选择具有书法特色的字体,使整体春联更具艺术感和节庆氛围,比如行书、隶书、楷书、草书等等。那么我们就可以引入一些个性化的字体,让我们的春联更加的写实。
我是在这个字体网站下载了一个行书的字体,这里我只是用来个人使用演示,如有侵权,我会马上删除。
下载好之后在css中加入如下代码,就可以使用自定义字体:
@font-face { font-family: "xingshu"; src: url("../../assets/ziti.ttf") format("truetype"); } .container { font-family: "xingshu", sans-serif; }
下载了行书字体之后,我们的春联的样式就更加好看了~
最后
本文介绍了一种AIGC
生成春联的方式,其实大模型的能力远不止这些,你可以调试好prompt
让他帮你实现非常多的功能。如果你有别的想法,欢迎在评论区进行交流~看都看到这里了,点点关注点点赞吧~
最后给大家拜个早年,祝大家龙年行大运,身体健康,合家幸福~