糾錯(cuò)幫【python文稿AI糾錯(cuò)工具】

糾錯(cuò)幫【python文稿AI糾錯(cuò)工具】，由論壇大神制作的一個(gè)論文糾錯(cuò)工具，由Python語言編寫，可以檢查出文檔內(nèi)的語句，錯(cuò)字內(nèi)容，選擇好文件導(dǎo)入后就可以一鍵進(jìn)行檢測，能有效檢查出一些基礎(chǔ)錯(cuò)誤，提高文章的準(zhǔn)確率。本次放出大神制作的這個(gè)糾錯(cuò)幫小程序工具下載。

糾錯(cuò)幫工具使用

選擇文件并加載

點(diǎn)擊開始檢測一鍵查詢

查詢結(jié)束后會(huì)顯示錯(cuò)誤點(diǎn)并自動(dòng)糾正

糾錯(cuò)幫工具源碼

import PySimpleGUI as sg

from aip import AipNlp

from docx import Document

from docx.shared import RGBColor

import os

import pickle

import random

import time

class TextAIAnalyse(object):

    """

    :param

    """

    def __init__(self, doc_path, app_id, api_key, secret_key):

        """

        :param doc_path：文章路徑

        :param app_id: 應(yīng)用id，自己去百度ai控制臺(tái)構(gòu)建一個(gè)應(yīng)用，就會(huì)有id了

        :param api_key:

        :param secret_key:

        """

        self.client = AipNlp(app_id, api_key, secret_key)

        self.document = Document(doc_path)

        self.doc_path = doc_path

        text_list1 = self.filter_style()

        self.text_list2 = self.filter_short_text(text_list1, 12)

    def filter_style(self):

        """

        樣式過濾

        :param

        """

        delete_style = ['Title', 'Heading 1', 'Quote']  # 去除標(biāo)題，一級標(biāo)題，圖表鏈接

        list1 = [x.text for x in self.document.paragraphs if x.style.name not in delete_style]

        return list1

    @staticmethod

    def filter_short_text(list1: list, length: int):

        """

        去除短文本

        :param list1: 列表格式的文本集

        :param length:最短文本長度

        """

        list2 = [x.strip() for x in list1]  # 去除兩邊空格

        list3 = [x for x in list2 if len(x) > length]

        return list3

    def split_text(self, list1: list):

        """

        對段落進(jìn)行分句，粗糙分詞

        :param

        """

        list2 = []

        for x in list1:

            x_list1 = x.split('。')  # 以句號(hào)進(jìn)行分詞,分號(hào)暫時(shí)不考慮

            for xx in x_list1:

                if xx[-1:] not in ['。', '；', '：']:  # 如果本局不是以句號(hào)結(jié)尾，則給它加上句號(hào)

                    xx += '。'

                    list2.append(xx)

        list3 = self.filter_short_text(list2, 10)

        return list3

    def split_text2(self, list1: list):

        """

        對段落進(jìn)行分句，加上分號(hào)

        :param

        """

        list2 = []

        for x in list1:

            x_list1 = x.split('。')  # 以句號(hào)進(jìn)行分詞

            for xx in x_list1:

                x_list2 = xx.split('；')  # 以中文分號(hào)進(jìn)行分詞

                for xxx in x_list2:

                    if xxx[-1:] not in ['。', '；', '：']:  # 如果本局不是以句號(hào)結(jié)尾，則給它加上句號(hào)

                        xxx += '。'

                        list2.append(xxx)

        list3 = self.filter_short_text(list2, 10)

        return list3

    def ai_analyse(self, text1):

        """

        AI對句子進(jìn)行糾錯(cuò)

        :param

        """

        result1 = None

        try:

            result1 = self.client.ecnet(text1)

        except Exception as err:

            return False

        vet = result1['item']['vec_fragment']  # 可替換詞

        score = result1['item']['score']  # 評分

        if len(vet) == 0:

            return False  # 沒有錯(cuò)誤

        # elif score > 0.5:  # 如果可信度

        else:

            return result1  # 返回分析

    def save_analyse(self, result):

        """

        :param

        """

        text = result['text']

        text_encode = text.encode('gbk')

        vet_list = result['item']['vec_fragment']

        """ 開始寫入word """

        basename = os.path.basename(self.doc_path)

        path = './分析結(jié)果/{}.docx'.format(basename[:6])

        dir_name = './分析結(jié)果'

        if not os.path.exists(dir_name):  # 如果文件夾不存在

            os.mkdir(dir_name)

        a = vet_list[0]['begin_pos']  # 獲取第一個(gè)錯(cuò)誤標(biāo)簽的開始位置

        b = vet_list[-1]['end_pos']  # 獲取最后一個(gè)錯(cuò)誤的結(jié)束位置

        if not os.path.exists(path):  # 如果文件不存在

            doc = Document()

        else:

            doc = Document(path)

        doc.add_paragraph('錯(cuò)誤寫法', style='heading 1')  # 一級標(biāo)題

        p = doc.add_paragraph()  # 創(chuàng)建一個(gè)空段落

        p.add_run(text_encode[:a].decode('gbk'))  # 寫入沒有錯(cuò)誤的部分

        start_list = [x['begin_pos'] for x in vet_list]  # 記錄開始的位置

        end_list = [x['end_pos'] for x in vet_list]  # 記錄結(jié)束的位置

        if len(vet_list) == 1:  # 如果只有一個(gè)錯(cuò)誤

            run1 = p.add_run(text_encode[start_list[0]:end_list[0]].decode('gbk'))

            run1.font.color.rgb = RGBColor(255, 0, 0)  # 設(shè)置紅色字體

        else:  # 如果有多個(gè)錯(cuò)誤

            for i in range(len(vet_list)):

                run = p.add_run(text_encode[start_list:end_list[i]].decode('gbk'))

                run.font.color.rgb = RGBColor(255, 0, 0)  # 設(shè)置紅色字體

                if i < len(vet_list) - 1 and start_list[i + 1] - end_list[i] > 1:  # 如果后面一處錯(cuò)誤與前面一處錯(cuò)誤存在間距

                    p.add_run(text_encode[end_list:start_list[i + 1]].decode('gbk'))  # 增加一個(gè)沒有樣式的普通字體

        if len(text_encode) - b > 1:  # 如果最后一個(gè)錯(cuò)誤后面存在正確的內(nèi)容

            p.add_run(text_encode[b:].decode('gbk'))  # 寫入后面無錯(cuò)的內(nèi)容

        doc.add_paragraph('正確寫法', style='heading 1')  # 一級標(biāo)題

        correct = result['item']['correct_query']  # 正確的內(nèi)容

        doc.add_paragraph(correct)  # 寫入正確的內(nèi)容

        doc.save(path)

if __name__ == '__main__':

    my_font = 'Deja_Vu_Sans_Mono.ttf'

    my_font_style1 = (my_font, 11, "normal")

    # 菜單欄

    menu_def = [

        ['&菜單', ['使用說明', '更新記錄']],

        ['&文件', ['載入配置', '保存配置']]

    ]

    # 布局欄

    layout1 = [

        [sg.Menu(menu_def, tearoff=True)],

        [sg.Text('APP_ID:', size=(12, None)), sg.Input(key='app_id')],

        [sg.Text('API_KEY', size=(12, None)), sg.Input(key='api_key')],

        [sg.Text('SECRET_KEY', size=(12, None)), sg.Input(key='secret_key')],

        [sg.Text('文件位置'), sg.Input(key='file_name', size=(51, None))],

        [sg.FileBrowse('選擇文件', target='file_name'), sg.Button('開始檢測'),

         sg.CBox('中文分號(hào)分句', default=False, key='split_type'), sg.Button('退出')]

    ]

    # 窗口欄

    windows1 = sg.Window('糾錯(cuò)幫V1.1', layout=layout1, font=my_font_style1)

    for i in range(10):

        event1, value1 = windows1.read()

        if event1 in ('退出', None):

            break

        elif event1 == '使用說明':

            sg.popup('1.搜索百度AI開放平臺(tái)', '2.點(diǎn)擊控制臺(tái)，注冊并登錄', '3.選擇自然語音處理', '4.創(chuàng)建應(yīng)用',

                     '5.填寫appid, api_key, secret_key', '6.選擇需要糾錯(cuò)的文件', '7.點(diǎn)擊開始檢測', title='使用說明',

                     font=my_font_style1)

        elif event1 == '更新記錄':

            sg.popup(

                'V1.1更新記錄'

                '1.增加了分號(hào)分句功能',

                '2.增加了導(dǎo)出word對比功能',

                title='提示', font=my_font_style1)

        elif event1 == '保存配置':

            APP_ID = windows1['app_id'].get()

            API_KEY = windows1['api_key'].get()

            SECRET_KEY = windows1['secret_key'].get()

            file_path = windows1['file_name'].get()

            split_type = windows1['split_type'].get()

            if len(APP_ID) > 3 and len(API_KEY) > 5 and len(SECRET_KEY) > 5:

                dict1 = {

                    'app_id': APP_ID,

                    'api_key': API_KEY,

                    'secret_key': SECRET_KEY,

                    'split_type': split_type

                }

                with open('info.pkl', 'wb') as f:

                    pickle.dump(dict1, f)

                sg.popup('保存完畢', '已經(jīng)生成一個(gè)info.pkl文件到本地', title='提示', auto_close=True,

                         auto_close_duration=3, font=my_font_style1)

            else:

                sg.popup('請檢查你的api相關(guān)信息是否填寫完成', title='錯(cuò)誤提示', font=my_font_style1)

        elif event1 == '載入配置':

            if not os.path.exists('info.pkl'):

                sg.popup('沒有找到你的配置文件info.pkl', '請檢查你的文件是否在當(dāng)前路徑', title='錯(cuò)誤提示',

                         font=my_font_style1)

            else:

                with open('info.pkl', 'rb') as f:

                    dict2 = pickle.load(f)

                    windows1['app_id'].update(dict2['app_id'])

                    windows1['api_key'].update(dict2['api_key'])

                    windows1['secret_key'].update(dict2['secret_key'])

                    windows1['split_type'].update(dict2['split_type'])

                    sg.popup('配置文件載入完畢', title='提示', auto_close_duration=3, auto_close=True, font=my_font_style1)

        elif event1 == '開始檢測':

            APP_ID = windows1['app_id'].get()

            API_KEY = windows1['api_key'].get()

            SECRET_KEY = windows1['secret_key'].get()

            file_path = windows1['file_name'].get()

            split_type = windows1['split_type'].get()

            if len(APP_ID) > 3 and len(API_KEY) > 5 and len(SECRET_KEY) > 5:

                doc = TextAIAnalyse(file_path, APP_ID, API_KEY, SECRET_KEY)

                text_list2 = doc.text_list2

                if split_type:  # 如果選擇的True,也就是支持分號(hào)

                    text_list3 = doc.split_text(text_list2)

                else:

                    text_list3 = doc.split_text2(text_list2)

                sg.popup('開始檢測，共有{}句'.format(len(text_list3)),

                         '預(yù)計(jì)用時(shí){}秒'.format(len(text_list3)*2), auto_close_duration=5, auto_close=True)

                layout2 = [

                    [sg.Text('處理進(jìn)度條', font=my_font_style1),

                     sg.ProgressBar(len(text_list3), orientation='h', key='bar', size=(50, 20))],

                    [sg.Button('取消', font=my_font_style1)]

                ]

                windows2 = sg.Window(title='進(jìn)度條', layout=layout2, font=my_font_style1)

                bar = windows2['bar']

                for ii in range(len(text_list3)):

                    event2, value2 = windows2.read(timeout=10)

                    if event2 in ('取消', None):

                        break

                    result2 = doc.ai_analyse(text_list3[ii])

                    print(text_list3[ii])

                    if bool(result2):

                        print(result2)

                        doc.save_analyse(result2)

                    time.sleep(0.5 + random.random() / 10)

                    bar.UpdateBar(ii + i)

                windows2.close()

                sg.popup('已經(jīng)檢測完成', '并且生成了一個(gè)“分析結(jié)果”文件夾到本地', title='提示', font=my_font_style1)

            elif len(file_path) < 5:

                sg.popup('親！', '你還沒有選擇檢測的文件', title='提示', font=my_font_style1)

            else:

                sg.popup('請輸入你的api信息', '詳情請查看使用說明', title='提示',font=my_font_style1)

    windows1.close()

相關(guān)新聞

文本糾錯(cuò)

文本糾錯(cuò)目前可以做到糾正形近字、音近字、成語使用、量詞搭配、語法這些錯(cuò)誤。

形近字糾錯(cuò)

就比如把“時(shí)候”寫成了“時(shí)侯”，“領(lǐng)域”寫成“領(lǐng)城”。如果自己檢查的時(shí)候，很難發(fā)現(xiàn)形近字錯(cuò)誤。

有了秘塔寫作貓，直接把word文檔上傳到里面，有什么錯(cuò)誤直接就智能查找并顯示出來了。