main, lsb и dct под все типы сообщений и картинок

main.py

@@ -81,7 +81,7 @@ def encode_command(args) -> None:
         print("Ошибка: укажите сообщение через -m или -f")
         sys.exit(1)
 
-    # Проверяем, поместится ли сообщение (только для LSB)
+    # Проверяем вместимость (только для LSB)
     if args.method == 'lsb':
         try:
             capacity = calculate_capacity(args.image)
@@ -104,14 +104,12 @@ def encode_command(args) -> None:
         success = encode_lsb(args.image, message, args.output)
     elif args.method == 'dct':
         success = encode_dct(args.image, message, args.output)
-        success = False
     else:
         print(f"Неизвестный метод: {args.method}")
         success = False
 
     if success:
         print("Кодирование завершено успешно!")
-
         if args.psnr:
             show_psnr(args.image, args.output)
     else:
@@ -127,7 +125,6 @@ def decode_command(args) -> None:
         message = decode_lsb(args.image)
     elif args.method == 'dct':
         message = decode_dct(args.image)
-        message = None
     else:
         print(f"Неизвестный метод: {args.method}")
         sys.exit(1)