如何使用model.insertrow pyqt5添加pyqt5按钮

from PyQt5.QtWidgets import QApplication, QMainWindow, QTableView, QPushButton
from PyQt5.QtCore import Qt, QAbstractTableModel

class CustomTableModel(QAbstractTableModel):
    def __init__(self, data, headers):
        super().__init__()
        self.data = data
        self.headers = headers

    def rowCount(self, parent):
        return len(self.data)

    def columnCount(self, parent):
        return len(self.headers)

    def data(self, index, role):
        if role == Qt.DisplayRole:
            return self.data[index.row()][index.column()]

    def headerData(self, section, orientation, role):
        if role == Qt.DisplayRole:
            if orientation == Qt.Horizontal:
                return self.headers[section]

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        # 创建数据和表头
        data = [['Data 1', 'Data 2'], ['Data 3', 'Data 4']]
        headers = ['Column 1', 'Column 2']

        # 创建模型
        self.model = CustomTableModel(data, headers)

        # 创建表格视图
        self.table_view = QTableView()
        self.table_view.setModel(self.model)

        # 创建按钮
        button = QPushButton('Button')
        button.clicked.connect(self.add_button)

        # 布局
        layout = QVBoxLayout()
        layout.addWidget(self.table_view)
        layout.addWidget(button)

        # 设置主窗口布局
        central_widget = QWidget()
        central_widget.setLayout(layout)
        self.setCentralWidget(central_widget)

    def add_button(self):
        # 获取当前行数
        row_count = self.model.rowCount(None)

        # 插入新行
        self.model.insertRow(row_count)

        # 创建按钮
        button = QPushButton('New Button')

        # 将按钮添加到新行中
        index = self.model.index(row_count, 0)
        self.table_view.setIndexWidget(index, button)

if __name__ == '__main__':
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec_()