記一次智慧搜題軟體的開發

Al發表於2020-12-18

很早之前曾經做過一個圖片識別的專案,當時有一項功能是整題識別,即傳入數學題的截圖,可通過ocr技術識別出圖片內容,但當時只限於識別文字,並未作更深一步的處理,現在想來實用性並不強,畢竟大家更需要的是解題思路,而不是讓AI讀出題幹(題乾的文字,我都認識,連起來我就不知道怎麼下手去做了 = = ),最近剛好有時間,於是嘗試來為有娃的朋友做一個搜題神器。

鑑於之前整題識別的開發使用有道智雲的良好體驗,我再次開啟其官方文件,果然找到了拍照搜題服務的開放API,輕車熟路地做了一個簡單的批量搜題demo, 下面分享一下開發過程。

呼叫API介面的準備工作

首先,是需要在有道智雲的個人頁面上建立例項、建立應用、繫結應用和例項,獲取到應用的id和金鑰。具體個人註冊的過程和應用建立過程詳見文章

建立應用例項

開發過程詳細介紹

下面介紹具體的程式碼開發過程。

API接收的引數較為簡單:

欄位名 型別 含義 必填 備註
q text 要識別的圖片,需要Base64編碼 True 必須是Base64編碼(baes64前邊不要加上data:image/png;base64)
appKey text 應用ID True 可在應用管理檢視
salt text UUID True uuid
curtime text 當前UTC時間戳(秒) true TimeStamp
sign text 簽名 sha256(應用ID+input+salt+curtime+應用金鑰);input的生成規則見表下的備註 True sha256(應用ID+input+salt+curtime+應用金鑰)
signType text 簽名型別 true v2
type text 上傳型別, 僅支援base64上傳,請填寫固定值1 True 1
searchType text 搜尋型別,img為圖片搜題,text為文字搜題 false img

簽名sign生成方法如下:
signType=v2;
sign=sha256(應用ID+input+salt+curtime+應用金鑰)。
其中,input的計算方式為:input=q前10個字元 + q長度 + q後10個字元(當q長度大於20)或 input=q字串(當q長度小於等於20)。

需要注意的是,API對題目圖片有如下要求:

規則 描述
傳輸方式 HTTPS
請求方式 POST
字元編碼 統一使用UTF-8編碼
請求格式 表單
響應格式 JSON
圖片格式 jpg/png/bmp
圖片大小 1MB以下
文字長度 50個字元以下

Demo開發:

這個demo使用python3開發,包括maindow.py,QuestionClass.py,OcrQuestion.py 三個檔案,分別為demo的介面、介面邏輯處理和ocr搜題方法的封裝。

  1. 介面部分:

    UI 部分較簡單,主要功能為選擇待題目圖片、選擇批改結果儲存路徑。其佈局程式碼如下:

    root=tk.Tk()
    root.title(" youdao ocr question test")
    frm = tk.Frame(root)
    frm.grid(padx='50', pady='50')
    
    # 選題和結果儲存按鈕
    btn_get_file = tk.Button(frm, text='選擇題目圖片', command=get_files)
    btn_get_file.grid(row=0, column=0, ipadx='3', ipady='3', padx='10', pady='20')
    text1 = tk.Text(frm, width='40', height='10')
    text1.grid(row=0, column=1)
    btn_get_result_path=tk.Button(frm,text='選擇搜尋結果路徑',command=set_result_path)
    btn_get_result_path.grid(row=1,column=0)
    text2=tk.Text(frm,width='40', height='2')
    text2.grid(row=1,column=1)
    
    
    # 搜題按鈕
    btn_sure=tk.Button(frm,text="搜題",command=search_question_files)
    btn_sure.grid(row=4,column=1)
    
    root.mainloop()
    

    其中啟動按鈕btn_sure的繫結事件search_question_files()來根據題目照片搜題,並在完成後開啟結果儲存路徑:

    def search_question_files():
        question.start_ocr()
        os.system('start '+question.result_path)
    
  2. QuestionClass.py

    這裡主要配合UI的邏輯,呼叫搜題方法。

    首先定義一個類Question:

    class Question():
        def __init__(self,file_paths,result_path):	
            self.file_paths=file_paths		# 題目照片儲存路徑
            self.result_path=result_path	# 結果路徑
    

    start_ocr()方法呼叫connect()方法依次搜題並儲存結果。

    def start_ocr(self):
        for file_path in self.file_paths:
            result=connect(file_path)
            print(file_path)
           self.save_result_format(file_path,result)
    

    從OcrQuestion.py的connect方法獲取的結果是json格式,save_result_format()方法,解析從介面取得的介面,格式整理,儲存結果到html:

        def save_result_format(self,file_path,result):
            result_file_name=os.path.basename(file_path).split('.')[0]+'_result.html'
            f=open(self.result_path+'/'+result_file_name,'w',encoding='utf-8')
            result_json= json.loads(result)
            if result_json['errorCode'] == '0':
                data=result_json['data']
                questions=data["questions"]
                text=data["text"]
                f.write("題目識別:<br/>"+text)
                i=0
                for answers in questions:
                    i=i+1
                    subject="科目:"+answers["subject"]+"<br>"
                    answer="答案:" +answers["answer"]+"<br>"
                    analysis="分析:"+answers["analysis"]+"<br>"
                    knowledge="知識點:"+answers["knowledge"]+"<br>"
                    print(subject+answer+analysis+knowledge)
                    result_each="<h3>搜題結果"+str(i)+"<br></h3>"
                    result_each=result_each+subject+answer+analysis+knowledge+"<br>=================這是一條分隔符============<br>"
                    f.write(result_each)
            else:
                f.write("result error code:"+result_json['errorCode'])
    
  3. OcrQuestion.py

    OcrQuestion.py 中封裝請求ocr搜題API的方法,其中最主要的方法是connect():

    def connect(pic_path):
        f = open(pic_path, 'rb')  # 二進位制方式開啟圖檔案
        q = base64.b64encode(f.read()).decode('utf-8')  # 讀取檔案內容,轉換為base64編碼
        f.close()
    
        data = {}
        data['q'] = q
        data['signType'] = 'v2'
        curtime = str(int(time.time()))
        data['curtime'] = curtime
        salt = str(uuid.uuid1())
        signStr = APP_KEY + truncate(q) + salt + curtime + APP_SECRET
        sign = encrypt(signStr)
        data['appKey'] = APP_KEY
        data['salt'] = salt
        data['sign'] = sign
    
        response = do_request(data)
        result=response.content.decode('utf-8')
        print(result)
        return result
    
    API響應結果示例
    {
        "data":{
            "questions":[
                {
                    "score":0.9875,
                    "answer":"D",
                    "subject":"歷史",
                    "id":"a9db8f1252778836c99204e5cf9d7738",
                    "analysis":"",
                    "type":"",
                    "content":"學者們認為,五四運動標誌著中國民族意識極大覺醒,表現在<br />A.近代民族民主思想開始傳播 B .揭開了反帝反封建鬥爭的序幕<br />C.開啟了近代中國的思想啟蒙 D .各階層民眾的廣泛動員和參與",
                    "knowledge":""
                },
                {
                    "score":0.9875,
                    "answer":"D <br>注意題幹“中國民族意識極大覺醒”,結合所學可知,在五四運動過程中,學生、工人、商人都投入到之中,即各階層民眾的廣泛動員和參與,符合題意,故D項正確. <br>在戊戌變法前後,中國民族民主思想就已經開始傳播,故A項錯誤. <br>五四運動是一次反帝反封的愛國運動,但不是反帝反封鬥爭的開始,在這時期還沒有明確提出反帝反封的革命綱領,故B項錯誤. <br>中國近代是在新文化運動期間傳播民主思想,進行了思想的啟蒙,故C項錯誤. <br>故選D.",
                    "subject":"歷史",
                    "id":"2f8b9c814f064131cc7b2bead54f15da",
                    "analysis":"本題考查五四運動.考查五四運動的意義.考查學生對基礎知識的識記、運用能力.",
                    "type":"選擇題",
                    "content":"學者們認為,五四運動標誌著中國民族意識極大覺醒,表現在(  ) <p class="question_option">A.近代民族民主思想開始傳播<br/>B.揭開了反帝反封建鬥爭的序幕<br/>C.開啟了近代中國的思想啟蒙<br/>D.各階層民眾的廣泛動員和參與 </p>",
                    "knowledge":"/人類政治文明發展史/近代中國反侵略、求民主的潮流/五四運動"
                },
                {
                    "score":0.9875,
                    "answer":"D <br>【答案】 <p>D</p> <p></p>",
                    "subject":"歷史",
                    "id":"48a28969d882f20b04ffec15c8f1b94c",
                    "analysis":"<p>題目問的五四運動,在戊戌變法前後,中國民族民主思想就已經開始傳播,所以A不正確。五四運動是一次反帝反封的愛國運動,但不是反帝反封鬥爭的開始,在這時期還沒有明確提出反帝反封的革命綱領,所以B不正確。中國近代是在新文化運動期間傳播民主思想,進行了思想的啟蒙,所以C不正確。在五四運動 過程中,學生、工人、商人都投入到之中,所以D正確。</p> <p> </p>",
                    "type":"選擇題",
                    "content":"<p>學者們認為,五四運動標誌著中國民族意識極大覺醒,表現在</p> <p></p> <p class="question_option">A.近代民族民主思想開始傳播<br/>B.揭開了反帝反封建鬥爭的序幕<br/>C.開啟了近代中國的思想啟蒙<br/>D.各階層民眾的廣泛動員和參與 </p>",
                    "knowledge":"/人類政治文明發展史/近代中國反侵略、求民主的潮流"
                }
            ],
            "text":"學者們認為,五四運動標誌著中國民族意識極大覺醒,表現在 A,近代民族民主思想開始傳播B揭開了反帝反封建鬥爭的序幕 C,開啟了近代中國的思想啟蒙D。 各階層民眾的廣泛動員和參與"
        },
        "errorCode":"0"
    }
    

    響應結果是以json形式輸出,包含欄位如下表所示:

欄位 含義
errorCode 識別結果錯誤碼,一定存在。 詳細資訊可參見 錯誤程式碼列表
data 資料
-text 圖片題目OCR結果
-questions 相關題目
--id 答案
--content 題目內容
--answer 答案
--analysis 解析
--knowledge 知識點

效果展示

demo操作演示
操作過程及結果
來看看結果吧:

數學題搜尋結果展示:

數學result

歷史題搜題結果

歷史result

總結

有道智雲的整體搜尋API文件清晰,題目範圍極廣而且可以自動判斷學科,搜尋結果可謂“舉一反三”,會返回幾個可能的相近題目,很具有參考價值,值得推薦!數學解答題會返回相關圖片、公式等,用在web專案中效果灰常不錯。

專案地址:https://github.com/LemonQH/OcrQuestion

相關文章