427 lines
21 KiB
Python
427 lines
21 KiB
Python
"""
|
||
GUI интерфейс для стеганографического приложения
|
||
"""
|
||
|
||
import tkinter as tk
|
||
from tkinter import ttk, filedialog, messagebox
|
||
import threading
|
||
import os
|
||
import sys
|
||
|
||
# Добавляем родительскую директорию для импорта модулей core
|
||
sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
||
|
||
from core.lsb import encode_lsb, decode_lsb
|
||
from core.dct import encode_dct, decode_dct
|
||
from core.utils import calculate_capacity
|
||
|
||
|
||
class StegoGUI:
|
||
"""Главное окно приложения"""
|
||
|
||
def __init__(self, root):
|
||
self.root = root
|
||
self.root.title("Stego App - Стеганографическое приложение")
|
||
self.root.geometry("800x700")
|
||
self.root.resizable(True, True)
|
||
|
||
# Переменные
|
||
self.encode_image_path = tk.StringVar()
|
||
self.encode_output_path = tk.StringVar()
|
||
self.decode_image_path = tk.StringVar()
|
||
self.method_encode = tk.StringVar(value="lsb")
|
||
self.method_decode = tk.StringVar(value="lsb")
|
||
|
||
self.setup_ui()
|
||
|
||
def setup_ui(self):
|
||
"""Создание интерфейса"""
|
||
# Вкладки
|
||
notebook = ttk.Notebook(self.root)
|
||
notebook.pack(fill='both', expand=True, padx=10, pady=10)
|
||
|
||
# Вкладка "Кодирование"
|
||
self.encode_frame = ttk.Frame(notebook)
|
||
notebook.add(self.encode_frame, text="Скрыть сообщение")
|
||
self.setup_encode_tab()
|
||
|
||
# Вкладка "Декодирование"
|
||
self.decode_frame = ttk.Frame(notebook)
|
||
notebook.add(self.decode_frame, text="Извлечь сообщение")
|
||
self.setup_decode_tab()
|
||
|
||
# Вкладка "О программе"
|
||
about_frame = ttk.Frame(notebook)
|
||
notebook.add(about_frame, text="О программе")
|
||
self.setup_about_tab(about_frame)
|
||
|
||
def setup_encode_tab(self):
|
||
"""Вкладка кодирования"""
|
||
row = 0
|
||
|
||
# Метод
|
||
ttk.Label(self.encode_frame, text="Метод стеганографии:").grid(row=row, column=0, sticky='w', padx=5, pady=5)
|
||
ttk.Radiobutton(self.encode_frame, text="LSB (рекомендуется)", variable=self.method_encode, value="lsb").grid(row=row, column=1, padx=5, sticky='w')
|
||
ttk.Radiobutton(self.encode_frame, text="DCT (для гладких/одноцветных изображений)", variable=self.method_encode, value="dct").grid(row=row, column=2, padx=5, sticky='w')
|
||
row += 1
|
||
|
||
# Разделитель
|
||
ttk.Separator(self.encode_frame, orient='horizontal').grid(row=row, column=0, columnspan=4, sticky='ew', padx=5, pady=5)
|
||
row += 1
|
||
|
||
# Исходное изображение
|
||
ttk.Label(self.encode_frame, text="Исходное изображение:").grid(row=row, column=0, sticky='w', padx=5, pady=5)
|
||
self.encode_image_entry = ttk.Entry(self.encode_frame, textvariable=self.encode_image_path, width=55)
|
||
self.encode_image_entry.grid(row=row, column=1, columnspan=2, padx=5, sticky='ew')
|
||
ttk.Button(self.encode_frame, text="Обзор...", command=self.select_encode_image).grid(row=row, column=3, padx=5)
|
||
row += 1
|
||
|
||
# Выходное изображение
|
||
ttk.Label(self.encode_frame, text="Сохранить как (PNG):").grid(row=row, column=0, sticky='w', padx=5, pady=5)
|
||
self.encode_output_entry = ttk.Entry(self.encode_frame, textvariable=self.encode_output_path, width=55)
|
||
self.encode_output_entry.grid(row=row, column=1, columnspan=2, padx=5, sticky='ew')
|
||
ttk.Button(self.encode_frame, text="Обзор...", command=self.select_encode_output).grid(row=row, column=3, padx=5)
|
||
row += 1
|
||
|
||
# Разделитель
|
||
ttk.Separator(self.encode_frame, orient='horizontal').grid(row=row, column=0, columnspan=4, sticky='ew', padx=5, pady=5)
|
||
row += 1
|
||
|
||
# Сообщение
|
||
ttk.Label(self.encode_frame, text="Сообщение для сокрытия:").grid(row=row, column=0, sticky='nw', padx=5, pady=5)
|
||
self.message_text = tk.Text(self.encode_frame, height=8, width=65, wrap=tk.WORD)
|
||
self.message_text.grid(row=row, column=1, columnspan=3, padx=5, pady=5, sticky='ew')
|
||
row += 1
|
||
|
||
# Кнопка очистки сообщения
|
||
ttk.Button(self.encode_frame, text="Очистить сообщение", command=self.clear_message).grid(row=row, column=1, sticky='w', padx=5, pady=5)
|
||
row += 1
|
||
|
||
# Разделитель
|
||
ttk.Separator(self.encode_frame, orient='horizontal').grid(row=row, column=0, columnspan=4, sticky='ew', padx=5, pady=5)
|
||
row += 1
|
||
|
||
# Вместимость
|
||
self.capacity_label = ttk.Label(self.encode_frame, text="", foreground="gray")
|
||
self.capacity_label.grid(row=row, column=1, columnspan=2, sticky='w', padx=5)
|
||
row += 1
|
||
|
||
# Прогресс
|
||
self.encode_progress = ttk.Progressbar(self.encode_frame, mode='indeterminate', length=400)
|
||
|
||
# Кнопка
|
||
self.encode_btn = ttk.Button(self.encode_frame, text="Скрыть сообщение", command=self.do_encode)
|
||
self.encode_btn.grid(row=row, column=1, columnspan=2, pady=15)
|
||
row += 1
|
||
|
||
# Статус
|
||
self.status_label = ttk.Label(self.encode_frame, text="", foreground="blue")
|
||
self.status_label.grid(row=row, column=0, columnspan=4, pady=5)
|
||
|
||
# Привязка событий
|
||
self.encode_image_path.trace('w', lambda *args: self.update_capacity())
|
||
|
||
def setup_decode_tab(self):
|
||
"""Вкладка декодирования"""
|
||
row = 0
|
||
|
||
# Метод
|
||
ttk.Label(self.decode_frame, text="Метод:").grid(row=row, column=0, sticky='w', padx=5, pady=5)
|
||
ttk.Radiobutton(self.decode_frame, text="LSB (рекомендуется)", variable=self.method_decode, value="lsb").grid(row=row, column=1, padx=5, sticky='w')
|
||
ttk.Radiobutton(self.decode_frame, text="DCT (для гладких/одноцветных изображений)", variable=self.method_decode, value="dct").grid(row=row, column=2, padx=5, sticky='w')
|
||
row += 1
|
||
|
||
# Разделитель
|
||
ttk.Separator(self.decode_frame, orient='horizontal').grid(row=row, column=0, columnspan=4, sticky='ew', padx=5, pady=5)
|
||
row += 1
|
||
|
||
# Изображение
|
||
ttk.Label(self.decode_frame, text="Изображение со скрытым сообщением:").grid(row=row, column=0, sticky='w', padx=5, pady=5)
|
||
self.decode_image_entry = ttk.Entry(self.decode_frame, textvariable=self.decode_image_path, width=55)
|
||
self.decode_image_entry.grid(row=row, column=1, columnspan=2, padx=5, sticky='ew')
|
||
ttk.Button(self.decode_frame, text="Обзор...", command=self.select_decode_image).grid(row=row, column=3, padx=5)
|
||
row += 1
|
||
|
||
# Разделитель
|
||
ttk.Separator(self.decode_frame, orient='horizontal').grid(row=row, column=0, columnspan=4, sticky='ew', padx=5, pady=5)
|
||
row += 1
|
||
|
||
# Прогресс
|
||
self.decode_progress = ttk.Progressbar(self.decode_frame, mode='indeterminate', length=400)
|
||
|
||
# Кнопка
|
||
self.decode_btn = ttk.Button(self.decode_frame, text="Извлечь сообщение", command=self.do_decode)
|
||
self.decode_btn.grid(row=row, column=1, columnspan=2, pady=15)
|
||
row += 1
|
||
|
||
# Результат
|
||
ttk.Label(self.decode_frame, text="Извлеченное сообщение:").grid(row=row, column=0, sticky='nw', padx=5, pady=5)
|
||
self.result_text = tk.Text(self.decode_frame, height=8, width=65, wrap=tk.WORD)
|
||
self.result_text.grid(row=row, column=1, columnspan=3, padx=5, pady=5, sticky='ew')
|
||
row += 1
|
||
|
||
# Кнопки
|
||
button_frame = ttk.Frame(self.decode_frame)
|
||
button_frame.grid(row=row, column=1, columnspan=2, pady=5)
|
||
|
||
self.save_btn = ttk.Button(button_frame, text="Сохранить в файл", command=self.save_result, state='disabled')
|
||
self.save_btn.pack(side='left', padx=5)
|
||
|
||
ttk.Button(button_frame, text="Очистить результат", command=self.clear_result).pack(side='left', padx=5)
|
||
row += 1
|
||
|
||
# Статус
|
||
self.decode_status = ttk.Label(self.decode_frame, text="", foreground="blue")
|
||
self.decode_status.grid(row=row, column=0, columnspan=4, pady=5)
|
||
|
||
self.decoded_message = ""
|
||
|
||
def setup_about_tab(self, frame):
|
||
"""Вкладка 'О программе'"""
|
||
# Основная информация
|
||
info_frame = ttk.Frame(frame)
|
||
info_frame.pack(pady=20)
|
||
|
||
ttk.Label(info_frame, text="Stego App", font=("Arial", 16, "bold")).pack(pady=5)
|
||
ttk.Label(info_frame, text="Стеганографическое приложение", font=("Arial", 10)).pack(pady=2)
|
||
ttk.Label(info_frame, text="").pack(pady=5)
|
||
ttk.Label(info_frame, text="Петров Игорь ИУ10-24").pack()
|
||
ttk.Label(info_frame, text="").pack(pady=5)
|
||
|
||
# Год
|
||
ttk.Label(info_frame, text="2026", font=("Arial", 10, "italic")).pack(pady=5)
|
||
|
||
# Разделитель
|
||
ttk.Separator(frame, orient='horizontal').pack(fill='x', padx=20, pady=10)
|
||
|
||
# Техническая информация
|
||
tech_frame = ttk.Frame(frame)
|
||
tech_frame.pack(pady=10)
|
||
|
||
ttk.Label(tech_frame, text="Методы стеганографии:", font=("Arial", 10, "bold")).pack(anchor='w')
|
||
ttk.Label(tech_frame, text=" • LSB (Least Significant Bit) - сокрытие в последних битах пикселей").pack(anchor='w')
|
||
ttk.Label(tech_frame, text=" Работает с любыми изображениями, высокая вместимость").pack(anchor='w', padx=20)
|
||
ttk.Label(tech_frame, text="").pack()
|
||
ttk.Label(tech_frame, text=" • DCT (Discrete Cosine Transform) - сокрытие в частотной области").pack(anchor='w')
|
||
ttk.Label(tech_frame, text=" Рекомендуется для гладких/однотонных изображений").pack(anchor='w', padx=20)
|
||
ttk.Label(tech_frame, text="").pack()
|
||
|
||
ttk.Label(tech_frame, text="Поддерживаемые форматы:", font=("Arial", 10, "bold")).pack(anchor='w')
|
||
ttk.Label(tech_frame, text=" • PNG, BMP, JPG, JPEG").pack(anchor='w')
|
||
ttk.Label(tech_frame, text="").pack()
|
||
|
||
ttk.Label(tech_frame, text="Рекомендации:", font=("Arial", 10, "bold")).pack(anchor='w')
|
||
ttk.Label(tech_frame, text=" • Для сохранения результата используйте PNG").pack(anchor='w')
|
||
ttk.Label(tech_frame, text=" • LSB - универсальный метод, рекомендуется по умолчанию").pack(anchor='w')
|
||
|
||
def select_encode_image(self):
|
||
"""Выбор изображения для кодирования"""
|
||
filename = filedialog.askopenfilename(
|
||
title="Выберите исходное изображение",
|
||
filetypes=[("Изображения", "*.png *.bmp *.jpg *.jpeg"), ("Все файлы", "*.*")]
|
||
)
|
||
if filename:
|
||
self.encode_image_path.set(filename)
|
||
|
||
def select_encode_output(self):
|
||
"""Выбор выходного файла"""
|
||
filename = filedialog.asksaveasfilename(
|
||
title="Сохранить результат как",
|
||
defaultextension=".png",
|
||
filetypes=[("PNG", "*.png"), ("BMP", "*.bmp"), ("Все файлы", "*.*")]
|
||
)
|
||
if filename:
|
||
self.encode_output_path.set(filename)
|
||
|
||
def select_decode_image(self):
|
||
"""Выбор изображения для декодирования"""
|
||
filename = filedialog.askopenfilename(
|
||
title="Выберите изображение со скрытым сообщением",
|
||
filetypes=[("Изображения", "*.png *.bmp *.jpg *.jpeg"), ("Все файлы", "*.*")]
|
||
)
|
||
if filename:
|
||
self.decode_image_path.set(filename)
|
||
|
||
def clear_message(self):
|
||
"""Очистка поля сообщения"""
|
||
self.message_text.delete("1.0", tk.END)
|
||
|
||
def clear_result(self):
|
||
"""Очистка поля результата"""
|
||
self.result_text.delete("1.0", tk.END)
|
||
self.decoded_message = ""
|
||
self.save_btn.config(state='disabled')
|
||
|
||
def update_capacity(self):
|
||
"""Обновление информации о вместимости"""
|
||
path = self.encode_image_path.get()
|
||
if path and os.path.exists(path):
|
||
try:
|
||
capacity = calculate_capacity(path)
|
||
kb = capacity / 1024
|
||
self.capacity_label.config(text=f"📊 Максимальная вместимость: {capacity} байт (~{kb:.1f} КБ)")
|
||
except Exception as e:
|
||
self.capacity_label.config(text=f"⚠️ Ошибка: {e}")
|
||
else:
|
||
self.capacity_label.config(text="")
|
||
|
||
def check_image_for_dct(self, image_path):
|
||
"""Проверка изображения для DCT метода"""
|
||
try:
|
||
from PIL import Image, ImageStat
|
||
|
||
img = Image.open(image_path)
|
||
w, h = img.size
|
||
|
||
# Проверка размеров
|
||
if w % 8 != 0 or h % 8 != 0:
|
||
return False, f"Размеры изображения ({w}x{h}) должны быть кратны 8 для DCT метода"
|
||
|
||
# Проверка "гладкости" изображения
|
||
stat = ImageStat.Stat(img)
|
||
# Если стандартное отклонение цветов большое (>50) — изображение пёстрое
|
||
if max(stat.stddev) > 50:
|
||
return False, "Изображение содержит много деталей/цветов.\nDCT рекомендуется только для гладких/однотонных изображений.\nИспользуйте LSB метод."
|
||
|
||
return True, ""
|
||
except Exception as e:
|
||
return False, str(e)
|
||
|
||
def do_encode(self):
|
||
"""Выполнение кодирования"""
|
||
image = self.encode_image_path.get()
|
||
output = self.encode_output_path.get()
|
||
message = self.message_text.get("1.0", tk.END).strip()
|
||
method = self.method_encode.get()
|
||
|
||
if not image:
|
||
messagebox.showerror("Ошибка", "Выберите исходное изображение")
|
||
return
|
||
if not output:
|
||
messagebox.showerror("Ошибка", "Укажите путь для сохранения результата")
|
||
return
|
||
if not message:
|
||
messagebox.showerror("Ошибка", "Введите сообщение для сокрытия")
|
||
return
|
||
|
||
# Проверка для DCT
|
||
if method == 'dct':
|
||
is_ok, error_msg = self.check_image_for_dct(image)
|
||
if not is_ok:
|
||
messagebox.showerror("Ошибка", error_msg)
|
||
return
|
||
|
||
# Запускаем в отдельном потоке
|
||
self.encode_btn.config(state='disabled')
|
||
self.encode_progress.grid(row=11, column=1, columnspan=2, sticky='ew', padx=5, pady=5)
|
||
self.encode_progress.start()
|
||
self.status_label.config(text="⏳ Кодирование... Пожалуйста, подождите", foreground="blue")
|
||
|
||
thread = threading.Thread(target=self._do_encode, args=(image, output, message, method))
|
||
thread.daemon = True
|
||
thread.start()
|
||
|
||
def _do_encode(self, image, output, message, method):
|
||
"""Фоновое кодирование"""
|
||
try:
|
||
if method == 'lsb':
|
||
success = encode_lsb(image, message, output)
|
||
else:
|
||
success = encode_dct(image, message, output)
|
||
|
||
self.root.after(0, self._encode_done, success, None)
|
||
except Exception as e:
|
||
self.root.after(0, self._encode_done, False, str(e))
|
||
|
||
def _encode_done(self, success, error=None):
|
||
"""Завершение кодирования"""
|
||
self.encode_progress.stop()
|
||
self.encode_progress.grid_forget()
|
||
self.encode_btn.config(state='normal')
|
||
|
||
if success:
|
||
self.status_label.config(text="✅ Кодирование успешно завершено!", foreground="green")
|
||
messagebox.showinfo("Успех", f"✅ Сообщение успешно скрыто в изображении!\n\nСохранено: {self.encode_output_path.get()}")
|
||
else:
|
||
error_msg = error or "Неизвестная ошибка"
|
||
self.status_label.config(text=f"❌ Ошибка кодирования", foreground="red")
|
||
messagebox.showerror("Ошибка", f"Не удалось скрыть сообщение:\n{error_msg}")
|
||
|
||
def do_decode(self):
|
||
"""Выполнение декодирования"""
|
||
image = self.decode_image_path.get()
|
||
method = self.method_decode.get()
|
||
|
||
if not image:
|
||
messagebox.showerror("Ошибка", "Выберите изображение")
|
||
return
|
||
|
||
# Запускаем в отдельном потоке
|
||
self.decode_btn.config(state='disabled')
|
||
self.decode_progress.grid(row=2, column=1, columnspan=2, sticky='ew', padx=5, pady=5)
|
||
self.decode_progress.start()
|
||
self.decode_status.config(text="⏳ Декодирование... Пожалуйста, подождите", foreground="blue")
|
||
|
||
thread = threading.Thread(target=self._do_decode, args=(image, method))
|
||
thread.daemon = True
|
||
thread.start()
|
||
|
||
def _do_decode(self, image, method):
|
||
"""Фоновое декодирование"""
|
||
try:
|
||
if method == 'lsb':
|
||
message = decode_lsb(image)
|
||
else:
|
||
message = decode_dct(image)
|
||
|
||
self.root.after(0, self._decode_done, message, None)
|
||
except Exception as e:
|
||
self.root.after(0, self._decode_done, None, str(e))
|
||
|
||
def _decode_done(self, message, error=None):
|
||
"""Завершение декодирования"""
|
||
self.decode_progress.stop()
|
||
self.decode_progress.grid_forget()
|
||
self.decode_btn.config(state='normal')
|
||
|
||
if message and not error and message != "":
|
||
self.decoded_message = message
|
||
self.result_text.delete("1.0", tk.END)
|
||
self.result_text.insert("1.0", message)
|
||
self.save_btn.config(state='normal')
|
||
self.decode_status.config(text="✅ Декодирование успешно завершено!", foreground="green")
|
||
messagebox.showinfo("Успех", f"✅ Сообщение успешно извлечено!\n\nДлина: {len(message)} символов")
|
||
else:
|
||
error_msg = error or "Сообщение не найдено или повреждено"
|
||
self.decode_status.config(text=f"❌ Ошибка декодирования", foreground="red")
|
||
self.save_btn.config(state='disabled')
|
||
messagebox.showerror("Ошибка", f"Не удалось извлечь сообщение:\n{error_msg}")
|
||
|
||
def save_result(self):
|
||
"""Сохранение извлеченного сообщения в файл"""
|
||
if not self.decoded_message:
|
||
return
|
||
|
||
filename = filedialog.asksaveasfilename(
|
||
title="Сохранить сообщение",
|
||
defaultextension=".txt",
|
||
filetypes=[("Текстовые файлы", "*.txt"), ("Все файлы", "*.*")]
|
||
)
|
||
if filename:
|
||
try:
|
||
with open(filename, 'w', encoding='utf-8') as f:
|
||
f.write(self.decoded_message)
|
||
messagebox.showinfo("Успех", f"✅ Сообщение сохранено в:\n{filename}")
|
||
except Exception as e:
|
||
messagebox.showerror("Ошибка", f"Не удалось сохранить файл:\n{e}")
|
||
|
||
|
||
def main():
|
||
"""Запуск GUI"""
|
||
root = tk.Tk()
|
||
app = StegoGUI(root)
|
||
root.mainloop()
|
||
|
||
|
||
if __name__ == '__main__':
|
||
main() |