中秋节老家要贴对联之python无线对联生成器

class CoupletGenerator(QWidget):
    def __init__(self, parent=None, **kwargs):
        super(CoupletGenerator, self).__init__(parent)
       
        self.setWindowTitle('对联生成器')
        self.setWindowIcon(QIcon('data/icon.jpg'))
        self.setFixedSize(600, 400)

 self.label_title = QLabel('对联主题:')
        self.line_title = QLineEdit('中秋佳节')
        self.generate_btn = QPushButton('生成')
        self.label_result = QLabel('生成结果:')
        self.text_couple = QTextEdit()
 
        grid = QGridLayout()
        grid.addWidget(self.label_title, 1, 0)
        grid.addWidget(self.line_title, 1, 1, 1, 39)
        grid.addWidget(self.generate_btn, 1, 40)
        grid.addWidget(self.label_result, 2, 0)
        grid.addWidget(self.text_couple, 2, 1, 1, 40)
        self.setLayout(grid)

  def generate(self):
        if not self.line_title.text().strip():
            return
        url = f'https://aip.baidubce.com/rpc/2.0/nlp/v1/couplets?access_token={self.access_token}'
        headers = {
            'Content-Type': 'application/json'
        }
        all_couplets = []
        index_ranges = [(0, 2), (3, 5), (6, 8), (9, 11), (12, 14)]
        for idx in range(5):
            params = {
                'text': self.line_title.text(),
                'index': random.randint(*index_ranges[idx]),
            }
            response = requests.post(url, headers=headers, json=params)
            response_json = response.json()
            center = response_json['couplets']['center']
            first = response_json['couplets']['first']
            second = response_json['couplets']['second']
            text = f'横批: {center}\n上联: {first}\n下联: {second}'
            all_couplets.append(text)
        self.text_couple.setText('\n\n\n'.join(all_couplets))