关于百度OCR截图识字 代码存档

1. import tkinter
   
2. # import asyncio
   
3. from PIL import ImageGrab
   
4. from aip import AipOcr
   
5. import time
   
6. import os
   
7. import http.client
   
8. import hashlib
   
9. import json
   
10. import urllib
    
11. import random
    
12. from tkinter import ttk
    
13. 
    
14. 
    
15. # def get_from_lang(self):
    
16. #     print(self.from_lang.get())
    
17. #     # return
    
18. #
    
19. #
    
20. # def get_to_lang(self):
    
21. #     print(self.to_lang.get())
    
22. #     # return
    
23. 
    
24. 
    
25. class MyCapture:
    
26. 
    
27.     def __init__(self):
    
28.         # 变量X和Y用来记录鼠标左键按下的位置
    
29.         self.X = tkinter.IntVar(value=0)
    
30.         self.Y = tkinter.IntVar(value=0)
    
31.         self.sel = False
    
32.         self.ocr_text = None
    
33.         self.capture_png_path = ''
    
34.         self.capture_text_box = tkinter.Text(window)  # 创建text容器用于存放截图识别的文字
    
35.         self.capture_text_box.place(x=10, y=40, anchor='nw', width=366+60, height=366)
    
36. 
    
37.         self.capture_btn = tkinter.Button(text='截图', command=self.capture_cmd)  # 创建一个按钮
    
38.         self.capture_btn.place(x=10, y=10, anchor='nw', width=60, height=26)  # 在创建的窗口的西北角x=20,y=10处放置按钮
    
39. 
    
40.         self.label_1 = tkinter.Label(text='点击截图，然后按住鼠标左键选取范围，松开即可识别所选区域')
    
41.         self.label_1.place(x=76,y=10)
    
42.         #self.label_1.pack(side='top')
    
43. 
    
44. 
    
45.         # 屏幕尺寸
    
46.         self.screenWidth = window.winfo_screenwidth()
    
47.         self.screenHeight = window.winfo_screenheight()
    
48.         self.temp_png = 'temp.png'
    
49.         # self.create_canvas()
    
50. 
    
51.     def create_canvas(self):
    
52.         time.sleep(0.2)
    
53.         im = ImageGrab.grab()
    
54.         im.save(self.temp_png)
    
55.         im.close()
    
56.         # 创建顶级组件容器
    
57.         self.top = tkinter.Toplevel(window, width=self.screenWidth, height=self.screenHeight)
    
58.         # 不显示最大化、最小化按钮
    
59.         self.top.overrideredirect(True)
    
60.         self.canvas = tkinter.Canvas(self.top, bg='white', width=self.screenWidth, height=self.screenHeight)
    
61.         # 显示全屏截图，在全屏截图上进行区域截图
    
62.         self.image = tkinter.PhotoImage(file=self.temp_png)
    
63.         self.canvas.create_image(self.screenWidth // 2, self.screenHeight // 2, image=self.image)
    
64. 
    
65.         # 鼠标左键按下的位置
    
66.         self.canvas.bind('<Button-1>', self.mouse_left_down)
    
67.         # 鼠标左键移动，显示选取的区域
    
68.         self.canvas.bind('<B1-Motion>', self.mouse_move)
    
69.         # 获取鼠标左键抬起的位置，保存区域截图
    
70.         self.canvas.bind('<ButtonRelease-1>', self.mouse_left_up)
    
71. 
    
72.         self.canvas.pack(fill=tkinter.BOTH, expand=tkinter.YES)
    
73. 
    
74.     def mouse_left_down(self, event):
    
75.         """鼠标左键按下的位置"""
    
76.         self.X.set(event.x)
    
77.         self.Y.set(event.y)
    
78.         self.sel = True  # 开始截图
    
79. 
    
80.     # 鼠标左键移动，显示选取的区域
    
81.     def mouse_move(self, event):
    
82.         if not self.sel:
    
83.             return
    
84.         try:
    
85.             # 删除刚画完的图形，要不然鼠标移动的时候是黑乎乎的一片矩形
    
86.             self.canvas.delete(self.lastDraw)
    
87.         except Exception as e:
    
88.             pass
    
89.         self.lastDraw = self.canvas.create_rectangle(self.X.get(), self.Y.get(), event.x, event.y, outline='red')
    
90. 
    
91.     def mouse_left_up(self, event):
    
92.         """获取鼠标左键抬起的位置，保存区域截图"""
    
93.         self.sel = False
    
94.         try:
    
95.             self.canvas.delete(self.lastDraw)
    
96.         except Exception as e:
    
97.             pass
    
98.         # 考虑鼠标左键从右下方按下而从左上方抬起的截图
    
99.         x1, x2 = sorted([self.X.get(), event.x])  # tkinter记录的坐标点
    
100.         y1, y2 = sorted([self.Y.get(), event.y])
     
101. 
     
102.         pic = ImageGrab.grab((x1 + 1, y1 + 1, x2, y2))
     
103.         # pic.show()
     
104.         self.capture_png_path = 'capture_png.png'
     
105.         pic.save(self.capture_png_path)
     
106. 
     
107.         # 关闭当前窗口
     
108.         self.top.destroy()
     
109. 
     
110.     def capture_cmd(self):
     
111.         """点击截图按钮触发函数"""
     
112.         window.iconify()  # 窗口最小化
     
113.         # 显示全屏幕截图
     
114.         self.create_canvas()
     
115.         self.capture_btn.wait_window(self.top)
     
116.         os.remove(self.temp_png)
     
117.         self.ocr_text = self.baidu_ocr(self.capture_png_path)
     
118.         print(self.ocr_text)
     
119.         if self.ocr_text:
     
120.             self.capture_text_box.delete('1.0', tkinter.END)  # 清空文本框
     
121.             #self.translate_text_box.delete('1.0', tkinter.END)
     
122.             self.capture_text_box.insert('end', self.ocr_text)
     
123.             window.deiconify()  # 窗口显示
     
124.             os.remove(self.capture_png_path)
     
125. 
     
126.     def baidu_ocr(self, file_path):
     
127.         """ 调用通用文字识别, 图片参数为本地图片 """
     
128.         app_id = '22939066'
     
129.         api_key = 'axVPGC3eid6ZTkbsVAo9xjGO'
     
130.         secret_key = 'kdySkPheRA7jAtCBGMCYTvMoRmIbmBG2'
     
131.         ocr_text = ''
     
132.         if os.path.isfile(file_path):
     
133.             with open(file_path, 'rb') as fp:
     
134.                 image = fp.read()
     
135.             ocr_ret = AipOcr(app_id, api_key, secret_key).basicGeneral(image)
     
136.             words = ocr_ret.get('words_result')
     
137.             if words is not None and len(words):
     
138.                 for word in words:
     
139.                     # print(word['words'], end='\n')
     
140.                     ocr_text += word['words'] + '\n'
     
141.                 return ocr_text
     
142.             else:
     
143.                 return None
     
144.         else:
     
145.             return None
     
146. 
     
147.     def baidu_translate(self, content):
     
148.         app_id = '22939066'
     
149.         secret_key = 'kdySkPheRA7jAtCBGMCYTvMoRmIbmBG2'
     
150.         http_client = None
     
151.         myurl = '/api/trans/vip/translate'
     
152.         q = content
     
153.         # from_lang = 'zh'  # 源语言
     
154.         from_lang = self.from_lang  # 源语言
     
155.         to_lang = self.to_lang  # 翻译后的语言
     
156.         # to_lang = 'en'  # 翻译后的语言
     
157.         salt = random.randint(32768, 65536)
     
158.         sign = app_id + q + str(salt) + secret_key
     
159.         sign = hashlib.md5(sign.encode()).hexdigest()
     
160.         myurl = myurl + '?appid=' + app_id + '&q=' + urllib.parse.quote(
     
161.             q) + '&from=' + from_lang + '&to=' + to_lang + '&salt=' + str(
     
162.             salt) + '&sign=' + sign
     
163. 
     
164.         try:
     
165.             http_client = http.client.HTTPConnection('api.fanyi.baidu.com')
     
166.             http_client.request('GET', myurl)
     
167.             # response是HTTPResponse对象
     
168.             response = http_client.getresponse()
     
169.             json_response = response.read().decode("utf-8")  # 获得返回的结果，结果为json格式
     
170.             js = json.loads(json_response)  # 将json格式的结果转换字典结构
     
171.             # print(js)
     
172.             dst = str(js["trans_result"][0]["dst"])  # 取得翻译后的文本结果
     
173.             # print(dst)  # 打印结果
     
174.             return dst
     
175.         except Exception as e:
     
176.             print(e)
     
177.             return None
     
178.         finally:
     
179.             if http_client:
     
180.                 http_client.close()
     
181. 
     
182.     def get_from_lang(self, event):
     
183.         # print(self.from_lang_box.get())
     
184.         self.from_lang = self.lang_dic[self.from_lang_box.get()]
     
185.         # return
     
186. 
     
187.     def get_to_lang(self, event):
     
188.         # print(self.to_lang_box.get())
     
189.         self.to_lang = self.lang_dic[self.to_lang_box.get()]
     
190.         # return
     
191. 
     
192. 
     
193. window = tkinter.Tk()
     
194. window.title('截图识字-Python')
     
195. # 创建tkinter主窗口
     
196. window.geometry('432x420')  # 指定主窗口位置与大小
     
197. window.minsize(366+86,420)
     
198. capture = MyCapture()
     
199. # capture.from_lang.bind("<<ComboboxSelected>>", get_from_lang(capture))
     
200. # capture.to_lang.bind("<<ComboboxSelected>>", get_to_lang(capture))
     
201. window.mainloop()

复制代码

• 本内容来源互联网，如涉及版权问题请及时联系站长
• 为防止代码解析，除代码框内<>为正常，其余请重新替换<>尖括号