mejora este programa y hazlo mejor: def train_model(num, image): model_data[num] = image def get_number(image): closest_num = None smallest_difference = 9999999 for num, num_image in model_data.items(): difference = image_difference(num_image, image) if difference < smallest_difference: closest_num = num smallest_difference = difference return closest_num def image_difference(imageA, imageB): difference = 0 for x in range(width): for y in range(height): pixelA = imageA.getpixel((x, y)) pixelB = imageB.getpixel((x, y)) difference += abs(pixelA[0] - pixelB[0]) return difference def get_image(): with mouse.Listener(on_click=on_click) as listener: listener.join() image = ImageGrab.grab() image = image.crop((50, 100, 250

def get_image():
  with mouse.Listener(on_click=on_click) as listener:
    listener.join()
  image = ImageGrab.grab()
  image = image.crop((50, 100, 250, 300))
  return image
  
def save_model():
  with open("model.json", "w") as f:
    json.dump(model_data, f)

def load_model():
  with open("model.json", "r") as f:
    return json.load(f)


model_data = load_model()

while True:
  image = get_image()
  image.show()
  num = int(input("Enter number: "))
  train_model(num, image)
  save_model()