работающая стеганография LSB и DCT

я устал... DCT убивает

main.py

@@ -17,6 +17,7 @@ import argparse
 import sys
 import os
 from core.lsb import encode_lsb, decode_lsb
+from core.dct import encode_dct, decode_dct
 from core.utils import calculate_capacity, calculate_psnr
 
 
@@ -102,7 +103,7 @@ def encode_command(args) -> None:
     if args.method == 'lsb':
         success = encode_lsb(args.image, message, args.output)
     elif args.method == 'dct':
-        print("Метод DCT пока не реализован")
+        success = encode_dct(args.image, message, args.output)
         success = False
     else:
         print(f"Неизвестный метод: {args.method}")
@@ -125,7 +126,7 @@ def decode_command(args) -> None:
     if args.method == 'lsb':
         message = decode_lsb(args.image)
     elif args.method == 'dct':
-        print("Метод DCT пока не реализован")
+        message = decode_dct(args.image)
         message = None
     else:
         print(f"Неизвестный метод: {args.method}")