Прошу помощи зала!

import os
import csv
import tkinter as tk
from tkinter import filedialog, messagebox, simpledialog
from PIL import Image, ImageTk, ImageDraw, ImageFont

class ImageViewer:
    def __init__(self, root):
        self.root = root
        self.root.title("Просмотрщик изображений")
        self.root.geometry("1200x800")
        
        # Переменные
        self.image_dir = ""
        self.image_files = []
        self.current_index = 0
        self.descriptions = {}
        self.show_desc = False
        self.fullscreen = False
        self.original_size = None
        
        # Загрузка CSV файла
        self.csv_file = None
        
        # Создание интерфейса
        self.create_menu()
        self.create_widgets()
        
        # Привязка горячих клавиш
        self.bind_keys()
        
    def create_menu(self):
        """Создание меню"""
        menubar = tk.Menu(self.root)
        self.root.config(menu=menubar)
        
        file_menu = tk.Menu(menubar, tearoff=0)
        menubar.add_cascade(label="Файл", menu=file_menu)
        file_menu.add_command(label="Выбрать директорию", command=self.choose_directory)
        file_menu.add_command(label="Обновить список", command=self.scan_directory)
        file_menu.add_separator()
        file_menu.add_command(label="Выход", command=self.root.quit)
        
        help_menu = tk.Menu(menubar, tearoff=0)
        menubar.add_cascade(label="Справка", menu=help_menu)
        help_menu.add_command(label="Горячие клавиши", command=self.show_help)
        
    def create_widgets(self):
        """Создание виджетов"""
        # Основные фреймы
        self.main_frame = tk.Frame(self.root)
        self.main_frame.pack(fill=tk.BOTH, expand=True, padx=5, pady=5)
        
        # Фрейм для списка файлов
        self.list_frame = tk.Frame(self.main_frame, width=250)
        self.list_frame.pack(side=tk.LEFT, fill=tk.Y, padx=(0, 5))
        self.list_frame.pack_propagate(False)
        
        # Фрейм для изображения
        self.image_frame = tk.Frame(self.main_frame)
        self.image_frame.pack(side=tk.RIGHT, fill=tk.BOTH, expand=True)
        
        # Список файлов
        self.file_listbox = tk.Listbox(self.list_frame, font=("Arial", 10))
        self.file_listbox.pack(side=tk.LEFT, fill=tk.BOTH, expand=True)
        
        # Скроллбар для списка
        scrollbar = tk.Scrollbar(self.list_frame)
        scrollbar.pack(side=tk.RIGHT, fill=tk.Y)
        
        self.file_listbox.config(yscrollcommand=scrollbar.set)
        scrollbar.config(command=self.file_listbox.yview)
        
        # Canvas для изображения
        self.canvas = tk.Canvas(self.image_frame, bg='gray20')
        self.canvas.pack(fill=tk.BOTH, expand=True)
        
        # Привязка событий мыши
        self.canvas.bind("<Double-Button-1>", self.toggle_fullscreen)
        
        # Информационная метка
        self.info_label = tk.Label(self.root, text="Выберите директорию через меню Файл", 
                                   bd=1, relief=tk.SUNKEN, anchor=tk.W)
        self.info_label.pack(side=tk.BOTTOM, fill=tk.X)
        
    def bind_keys(self):
        """Привязка горячих клавиш"""
        self.root.bind("<Up>", lambda e: self.move_selection(-1))
        self.root.bind("<Down>", lambda e: self.move_selection(1))
        self.root.bind("<Left>", lambda e: self.navigate_images(-1))
        self.root.bind("<Right>", lambda e: self.navigate_images(1))
        self.root.bind("<Return>", lambda e: self.edit_description())
        self.root.bind("<space>", lambda e: self.toggle_description())
        self.root.bind("<Escape>", lambda e: self.exit_fullscreen())
        self.root.bind("<F1>", lambda e: self.show_help())
        self.root.bind("<F5>", lambda e: self.scan_directory())
        
    def choose_directory(self):
        """Выбор директории"""
        directory = filedialog.askdirectory(title="Выберите директорию с изображениями")
        if directory:
            self.image_dir = directory
            self.csv_file = os.path.join(directory, "image_descriptions.csv")
            self.scan_directory()
            
    def scan_directory(self):
        """Сканирование директории на наличие изображений"""
        if not self.image_dir:
            messagebox.showinfo("Информация", "Сначала выберите директорию")
            return
            
        # Поддерживаемые форматы
        image_extensions = ('.jpg', '.jpeg', '.png', '.gif', '.bmp', '.tiff', '.webp')
        
        # Загружаем существующие описания ДО сканирования директории
        self.load_existing_csv()
        
        # Поиск изображений
        found_files = []
        for file in os.listdir(self.image_dir):
            if file.lower().endswith(image_extensions):
                found_files.append(file)
        
        if not found_files:
            messagebox.showwarning("Нет изображений", "В выбранной директории не найдено изображений")
            return
            
        # Сортировка файлов
        found_files.sort()
        
        # Обновляем список файлов
        self.image_files = found_files
        
        # Сохраняем CSV файл с новыми файлами
        self.update_csv_with_new_files()
        
        # Обновление списка файлов
        self.update_file_list()
        
        # Обновляем информационную метку
        self.info_label.config(text=f"Директория: {os.path.basename(self.image_dir)} | Файлов: {len(self.image_files)}")
        
        # Показать первое изображение
        if self.image_files:
            self.current_index = 0
            self.display_image()
            
    def load_existing_csv(self):
        """Загрузка существующего CSV файла с описаниями"""
        self.descriptions = {}
        
        if os.path.exists(self.csv_file):
            try:
                with open(self.csv_file, 'r', newline='', encoding='utf-8') as file:
                    reader = csv.reader(file)
                    for row in reader:
                        if len(row) >= 2:
                            filename = row[0].strip()
                            description = row[1].strip()
                            self.descriptions[filename] = description
            except Exception as e:
                print(f"Ошибка при загрузке CSV: {e}")
                
    def update_csv_with_new_files(self):
        """Обновление CSV файла с добавлением новых файлов"""
        try:
            # Сначала загружаем текущий CSV (если есть)
            existing_data = {}
            if os.path.exists(self.csv_file):
                with open(self.csv_file, 'r', newline='', encoding='utf-8') as file:
                    reader = csv.reader(file)
                    for row in reader:
                        if len(row) >= 2:
                            existing_data[row[0]] = row[1]
            
            # Объединяем старые описания с новыми файлами
            updated_data = {}
            
            # Для каждого файла в директории
            for filename in self.image_files:
                # Если файл уже был в CSV, сохраняем его описание
                if filename in existing_data:
                    updated_data[filename] = existing_data[filename]
                else:
                    # Иначе добавляем новый файл без описания
                    updated_data[filename] = ""
            
            # Записываем обновленные данные в CSV
            with open(self.csv_file, 'w', newline='', encoding='utf-8') as file:
                writer = csv.writer(file)
                for filename, description in updated_data.items():
                    writer.writerow([filename, description])
                    
            # Обновляем словарь описаний
            self.descriptions = updated_data
            
        except Exception as e:
            messagebox.showerror("Ошибка", f"Не удалось обновить CSV файл: {e}")
            
    def update_file_list(self):
        """Обновление списка файлов"""
        self.file_listbox.delete(0, tk.END)
        
        for i, filename in enumerate(self.image_files):
            # Добавляем звездочку если есть описание
            display_name = filename
            if filename in self.descriptions and self.descriptions[filename]:
                display_name = f"* {filename}"
                
            self.file_listbox.insert(tk.END, display_name)
            
        # Выделяем текущий файл
        if self.current_index < len(self.image_files):
            self.file_listbox.selection_set(self.current_index)
            self.file_listbox.see(self.current_index)
            
    def display_image(self):
        """Отображение текущего изображения"""
        if not self.image_files or self.current_index >= len(self.image_files):
            return
            
        filename = self.image_files[self.current_index]
        filepath = os.path.join(self.image_dir, filename)
        
        try:
            # Открываем изображение
            self.image = Image.open(filepath)
            
            # Если нужно показать описание
            if self.show_desc and filename in self.descriptions and self.descriptions[filename]:
                self.add_description_to_image()
            
            # Масштабирование изображения
            self.resize_image()
            
            # Обновление списка файлов
            self.update_file_list()
            
        except Exception as e:
            messagebox.showerror("Ошибка", f"Не удалось открыть изображение {filename}: {e}")
            
    def resize_image(self):
        """Масштабирование изображения под размер окна"""
        if not hasattr(self, 'image'):
            return
            
        # Получаем размеры области для отображения
        if self.fullscreen:
            canvas_width = self.root.winfo_screenwidth()
            canvas_height = self.root.winfo_screenheight()
        else:
            canvas_width = self.image_frame.winfo_width()
            canvas_height = self.image_frame.winfo_height()
            
        # Если размеры неизвестны, используем стандартные
        if canvas_width <= 1:
            canvas_width = 800
        if canvas_height <= 1:
            canvas_height = 600
            
        # Сохраняем оригинальные размеры
        if not self.original_size:
            self.original_size = self.image.size
            
        # Масштабируем с сохранением пропорций
        img_width, img_height = self.image.size
        ratio = min(canvas_width / img_width, canvas_height / img_height)
        
        if ratio < 1 or self.fullscreen:
            new_size = (int(img_width * ratio), int(img_height * ratio))
            resized_image = self.image.resize(new_size, Image.Resampling.LANCZOS)
        else:
            resized_image = self.image
            
        # Конвертируем для Tkinter
        self.tk_image = ImageTk.PhotoImage(resized_image)
        
        # Очищаем canvas и отображаем изображение
        self.canvas.delete("all")
        
        # Центрируем изображение
        x = (canvas_width - self.tk_image.width()) // 2
        y = (canvas_height - self.tk_image.height()) // 2
        
        self.canvas.create_image(x, y, anchor=tk.NW, image=self.tk_image)
        
    def add_description_to_image(self):
        """Добавление описания на изображение"""
        if not hasattr(self, 'image'):
            return
            
        filename = self.image_files[self.current_index]
        description = self.descriptions.get(filename, "")
        
        if not description:
            return
            
        # Создаем копию изображения для рисования
        img_with_text = self.image.copy()
        draw = ImageDraw.Draw(img_with_text)
        
        # Параметры текста
        try:
            # Пробуем загрузить системный шрифт
            font = ImageFont.truetype("arial.ttf", 20)
        except:
            # Используем стандартный шрифт если системный не найден
            font = ImageFont.load_default()
            
        # Вычисляем размеры текста
        text_bbox = draw.textbbox((0, 0), description, font=font)
        text_width = text_bbox[2] - text_bbox[0]
        text_height = text_bbox[3] - text_bbox[1]
        
        # Позиция текста (внизу по центру)
        img_width, img_height = img_with_text.size
        text_x = (img_width - text_width) // 2
        text_y = img_height - text_height - 20
        
        # Рисуем фон для текста
        padding = 10
        draw.rectangle(
            [text_x - padding, text_y - padding, 
             text_x + text_width + padding, text_y + text_height + padding],
            fill="black"
        )
        
        # Рисуем текст
        draw.text((text_x, text_y), description, font=font, fill="white")
        
        self.image = img_with_text
        
    def move_selection(self, direction):
        """Перемещение выделения в списке файлов"""
        if not self.image_files:
            return
            
        new_index = self.current_index + direction
        
        if 0 <= new_index < len(self.image_files):
            self.current_index = new_index
            self.display_image()
            
    def navigate_images(self, direction):
        """Навигация по изображениям (для полноэкранного режима)"""
        if not self.fullscreen:
            return
            
        self.move_selection(direction)
        
    def edit_description(self):
        """Редактирование описания изображения"""
        if not self.image_files:
            return
            
        filename = self.image_files[self.current_index]
        current_desc = self.descriptions.get(filename, "")
        
        # Открываем диалоговое окно
        new_desc = simpledialog.askstring(
            "Описание изображения",
            f"Введите описание для {filename}:",
            initialvalue=current_desc
        )
        
        if new_desc is not None:
            self.descriptions[filename] = new_desc
            self.save_to_csv()
            self.update_file_list()
            
            # Если описание показывается, обновляем изображение
            if self.show_desc:
                self.display_image()
                
    def save_to_csv(self):
        """Сохранение описаний в CSV файл"""
        try:
            with open(self.csv_file, 'w', newline='', encoding='utf-8') as file:
                writer = csv.writer(file)
                for filename in self.image_files:
                    description = self.descriptions.get(filename, "")
                    writer.writerow([filename, description])
        except Exception as e:
            messagebox.showerror("Ошибка", f"Не удалось сохранить файл: {e}")
                
    def toggle_description(self):
        """Показать/скрыть описание"""
        if not self.image_files:
            return
            
        self.show_desc = not self.show_desc
        self.display_image()
        
    def toggle_fullscreen(self, event=None):
        """Включение/выключение полноэкранного режима"""
        self.fullscreen = not self.fullscreen
        
        if self.fullscreen:
            # Сохраняем текущую геометрию
            self.normal_geometry = self.root.geometry()
            
            # Скрываем рамки и меню
            self.root.attributes('-fullscreen', True)
            self.list_frame.pack_forget()
            
            # Обновляем изображение
            self.display_image()
        else:
            self.exit_fullscreen()
            
    def exit_fullscreen(self, event=None):
        """Выход из полноэкранного режима"""
        if self.fullscreen:
            self.fullscreen = False
            self.root.attributes('-fullscreen', False)
            self.root.geometry(self.normal_geometry)
            
            # Восстанавливаем интерфейс
            self.list_frame.pack(side=tk.LEFT, fill=tk.Y, padx=(0, 5))
            
            # Обновляем изображение
            self.display_image()
            
    def show_help(self, event=None):
        """Показ справки о горячих клавишах"""
        help_text = """Горячие клавиши:

Обычный режим:
↑ / ↓ - Перемещение по списку файлов
Enter - Редактировать описание изображения
Пробел - Показать/скрыть описание
F1 - Эта справка
F5 - Обновить список файлов
Escape - Выход из полноэкранного режима
Двойной щелчок по изображению - Полноэкранный режим

Полноэкранный режим:
← / → - Переключение изображений
Enter - Редактировать описание изображения
Пробел - Показать/скрыть описание
Escape - Выход из полноэкранного режима

Работа с CSV файлом:
- При первом сканировании создается файл image_descriptions.csv
- При добавлении новых файлов в директорию используйте F5 для обновления
- Существующие описания сохраняются
- Новые файлы добавляются в CSV без описаний"""
        
        messagebox.showinfo("Справка - Горячие клавиши", help_text)
        
    def on_resize(self, event):
        """Обработка изменения размера окна"""
        if hasattr(self, 'image'):
            self.resize_image()

def main():
    root = tk.Tk()
    app = ImageViewer(root)
    
    # Привязка события изменения размера
    root.bind("<Configure>", app.on_resize)
    
    root.mainloop()

if __name__ == "__main__":
    main()