precompute model

src/tea3/cli.py

@@ -7,6 +7,7 @@ from tea3.sbox import run_sbox
 from tea3.variable_xor import run_variable_xor, run_exhaustive_xor
 from tea3.f31f32 import run_f31f32
 from tea3.cube_attack import run_cube_attack
+from tea3.precompute import run_precompute_cli, run_load_cli
 
 
 def run_classic_cli():
@@ -214,11 +215,13 @@ def main():
         print("  6) F31, F32 analysis")
         print("  7) Exhaustive XOR search")
         print("  8) Cube attack search")
+        print("  9) Precompute and save model")
+        print("  10) Load model")
         print("  0) Exit")
 
         mode = prompt_choice(
-            "Your choice (0-8): ",
-            {0, 1, 2, 3, 4, 5, 6, 7, 8}
+            "Your choice (0-10): ",
+            {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
         )
 
         if mode == 0:
@@ -240,6 +243,10 @@ def main():
             run_exhaustive_xor_cli()
         elif mode == 8:
             run_cube_attack_cli()
+        elif mode == 9:
+            run_precompute_cli()
+        elif mode == 10:
+            model = run_load_cli()
 
         input("\nPress Enter to return to the main menu...")
         print()