宋词自动生成

article/2025/9/9 8:02:18

利用宋词语料库，通过单双词的组合与模板的匹配，实现指定词牌宋词的生成

import random
import tkinter as tk
import re
from tkinter import messageboxlist = []class Window:# 界面设计def __init__(self, root):label1 = tk.Label(root, text='输入词牌名')label1.place(x=5, y=5)label2 = tk.Label(root, text='宋词生成结果')label2.place(x=5, y=70)self.entryCp = tk.Entry(root)self.entryCp.place(x=120, y=5)self.text = tk.Text(root)self.text.place(y=100)self.get = tk.Button(root, text='生成宋词', command=self.songci)self.get.place(x=120, y=40)exit = tk.Button(root, text="退出程序", command=root.destroy)exit.place(x=200, y=40)# 生成宋词def songci(self):# 获取词牌名name = self.entryCp.get()# 读取源文件file = open('Ci.txt', 'r', encoding='utf-8')for line in file.readlines():line = line.strip()list.append(line)# 匹配索引位置，遇空则继续s = list.index(name) + 1while list[s] == '':s += 1# 去掉分隔符，生成新列表cut = re.split('[，。、]', list[s])newcut = ['' for s in range(len(cut))]for i in range(len(cut)):for j in range(len(cut[i])):if '\u4e00' <= cut[i][j] <= '\u9fa5':newcut[i] += cut[i][j]temp = ''for i in range(len(newcut)):sum = 0n = 1sum = sum + ntemp = temp + str(n)while len(newcut[i]) - sum > 1:# n = random.randint(1, 2)sum = sum + ntemp = temp + str(n)if len(newcut[i]) - sum == 1:temp = temp + str(1) + '/'if len(newcut[i]) - sum == 0:temp = temp + '/'temp = temp[:-1]# 读入单双词语料file1 = open('Ciout1.txt', 'r', encoding='utf-8')file2 = open('Ciout2.txt', 'r', encoding='utf-8')list1, list2 = [], []# 去掉逗号，向新列表中加入元素# 单词line1 = re.split('[（，）]', file1.read())for i in range(int((len(line1) - 1) / 3)):for j in range(int(line1[i * 3 + 2])):list1.append(line1[i * 3 + 1])# 双词line2 = re.split('[（，）]', file2.read())for i in range(int((len(line2) - 1) / 3)):for j in range(int(line2[i * 3 + 2])):list2.append(line2[i * 3 + 1])# 输出内容content = ''for i in temp:if i == '1':content = content + list1[random.randint(0, len(list1) - 1)]elif i == '2':content = content + list2[random.randint(0, len(list2) - 1)]else:content = content + '/'print(content)self.text.insert(tk.END, content)file.close()file1.close()file2.close()root = tk.Tk()
root.title("宋词自动生成器")
root.geometry("350x300+500+250")window = Window(root)
root.minsize(550, 250)
root.mainloop()