封装识别接口

gui/main.py

@@ -1,4 +1,5 @@
 import ctypes
+from predict import solve
 
 
 def shot(target, width, height):
@@ -20,9 +21,11 @@ def shot(target, width, height):
     if error_ptr:
         print(ctypes.string_at(error_ptr).decode())
     else:
-        result = ctypes.string_at(data, length.value)
-        with open('./screen/0.png', 'wb') as f:
-            f.write(result)
+        image_bytes = ctypes.string_at(data, length.value)
+        result = solve(image_bytes)
+
+        # with open('./screen/0.png', 'wb') as f:
+        #     f.write(result)
 
         lib.Free(data)