好友 阅读权限 10
听众 最后登录 1970-1-1
有兴趣的大神，一起讨论优化一下。可以将源码或者成品共享学习学习。import tkinter as tkfrom tkinter import filedialog, simpledialogfrom tkinter import ttkfrom PIL import ImageGrab, Imageimport pytesseractimport requestsimport jsonimport threadingimport timeimport osimport pytesseractr'C:\Program Files\Tesseract-OCR\tesseract.exe' # OCR 文字识别 def extract_text_from_image (image):, lang ='chi_sim' )return text.strip()# Kimi API 查询 def query_kimi (question, api_key, callback=None ):"Authorization" : f"Bearer { api_key} " , "Content-Type" : "application/json" }"model" : "qwen-plus" , "prompt" : f" 请回答以下问题： { question} " }try :"https://api.kimi.ai/v1/chat/completions" , json =data, headers =headers)"choices" , [{}])[0 ].get("message" , {}).get("content" , " 无法获取答案 " )if callback:except Exception as e:if callback:f"API 请求失败 : { str (e)} " )# 加载题库 def load_question_bank (file_path):try :with open (file_path, 'r' , encoding ='utf-8' ) as f:return json.load(f)except Exception as e:print (f" 加载题库失败 : { str (e)} " )return {}# 加载题库（ txt 格式） def load_question_bank_from_txt (file_path):""" 从 TXT 文件加载题库，每行格式为 ' 问题 : 答案 '""" try :with open (file_path, 'r' , encoding ='utf-8' ) as f:for line in f:if ":" in line:, answer = line.strip().split(":" , 1 )except Exception as e:print (f" 加载 TXT 题库失败 : { str (e)} " )return question_bank# 截图功能 def capture_screen (region=None ):if region:, y1, x2, y2 = regionreturn ImageGrab.grab(bbox =(x1, y1, x2, y2))return ImageGrab.grab()class FullScreenApp:def __init__ (self , master, on_capture):self .master = masterself .on_capture = on_captureself .root = tk.Toplevel(master)self .root.attributes('-fullscreen' , True )self .root.attributes('-alpha' , 0.3 )self .root.configure(bg ='black' )self .canvas = tk.Canvas(self .root, cursor ="cross" , highlightthickness =0 )self .canvas.pack(fill =tk.BOTH, expand =True )self .rect = None self .start_x = self .start_y = 0 self .end_x = self .end_y = 0 self .setup_bindings()def setup_bindings (self ):self .canvas.bind("<ButtonPress-1>" , self .on_button_press)self .canvas.bind("<B1-Motion>" , self .on_mouse_drag)self .canvas.bind("<ButtonRelease-1>" , self .on_button_release)def on_button_press (self , event):self .start_x = self .end_x = event.xself .start_y = self .end_y = event.yif not self .rect:self .rect = self .canvas.create_rectangle(0 , 0 , 0 , 0 , outline ='red' , width =2 )def on_mouse_drag (self , event):self .end_x = event.xself .end_y = event.yself .canvas.coords(self .rect, self .start_x, self .start_y, self .end_x, self .end_y)def on_button_release (self , event):self .root.destroy()min (self .start_x, self .end_x)min (self .start_y, self .end_y)max (self .start_x, self .end_x)max (self .start_y, self .end_y)if (x2 - x1) > 5 and (y2 - y1) > 5 :self .on_capture((x1, y1, x2, y2))# 主界面类 class AutoAnswerApp:def __init__ (self , root):self .root = rootself .root.title(" 自动答题助手 " )self .root.geometry("600x400" )self .root.resizable(False, False )self .style = ttk.Style()self .style.theme_use('clam' )self .style.configure("TButton" , padding =6 , relief ="flat" , background ="#4CAF50" )self .style.map("TButton" , background =[('active' , '#45a049' )])self .question_bank = {}self .kimi_api_key = os.getenv("KIMI_API_KEY" , "your_kimi_api_key_here" )# 创建菜单栏 menubar = tk.Menu(root), tearoff =0 )label =" 设置 Kimi API 密钥 " , command =self .set_api_key)label =" 退出 " , command =root.quit)label =" 文件 " , menu =filemenu)menu =menubar)# 主框架 main_frame = ttk.Frame(root, padding ="20" )fill =tk.BOTH, expand =True )# 选择题库文件按钮 self .select_button = ttk.Button(main_frame, text =" 选择题库文件 " , command =self .select_file, style ="TButton" )self .select_button.pack(pady =10 , fill =tk.X)# 截图识别题目按钮 self .capture_button = ttk.Button(main_frame, text =" 截图识别题目 " , command =self .start_fullscreen_capture, style ="TButton" )self .capture_button.pack(pady =10 , fill =tk.X)# 题目显示区域 self .question_label = tk.Label(main_frame, text =" 识别到的题目 :" , justify =tk.LEFT, wraplength =500 , font =(" 微软雅黑 " , 12 ))self .question_label.pack(pady =10 )# 答案显示区域 self .answer_label = tk.Label(main_frame, text =" 答案将在这里显示 :" , justify =tk.LEFT, wraplength =500 , font =(" 微软雅黑 " , 12 ), fg ="blue" )self .answer_label.pack(pady =10 )def set_api_key (self ):" 输入 " , " 请输入 Kimi API 密钥 :" , initialvalue =self .kimi_api_key)if new_key:self .kimi_api_key = new_keydef select_file (self ):filetypes =[(" 支持的文件格式 " , "*.json;*.txt" )])if file_path:, ext = os.path.splitext(file_path)if ext == ".json" :self .question_bank = load_question_bank(file_path)elif ext == ".txt" :self .question_bank = load_question_bank_from_txt(file_path)def start_fullscreen_capture (self ):self .root, self .process_capture)def process_capture (self , region):self .question_label.config(text =f" 识别到的题目 : \n{ question} " )# 异步查询 AI 答案 if question and self .kimi_api_key != "your_kimi_api_key_here" :target =query_kimi, args =(question, self .kimi_api_key, self .update_answer_label)).start()# 查找题库答案 answer_db = search_answer_in_db(question, self .question_bank)self .answer_label.config(text =f" 题库答案 : \n{ answer_db or ' 未找到 ' } ..." )def update_answer_label (self , answer):self .answer_label.config(text =f"AI 答案 : \n{ answer} " )# 启动程序 def main ():if __name__ == "__main__" :
查看全部评分
[复制链接]
发表于 2025-7-17 09:57
