before frames update

ui/local_processing_tab.py

@@ -4,40 +4,39 @@ Module for the Local Processing Tab in the Image Processor application.
 
 import tempfile
 import threading
-import tkinter as tk
-from tkinter import ttk
+import customtkinter as ctk
 from tkinter.scrolledtext import ScrolledText
-from tkinter import Tk, Label, Button, Entry, Toplevel, StringVar, BooleanVar
+from tkinter import StringVar, BooleanVar
 from PIL import Image, ImageTk
 from utils.file_operations import FileProcessor
 from utils.image_processing import ImageProcessor
 from api.woocommerce_api import process_product_images, process_all_products
 from ui.options_window import OptionsWindow
-from pprint import pformat, pprint
+from pprint import pformat
 from config.encrypt_config import ConfigEncryptor
+import os
+
 
 class LocalProcessingTab:
     """
     Class for the Local Processing Tab in the Image Processor application.
     """
 
-    def __init__(self, tab_parent, text, log):
+    def __init__(self, tab_parent, log_window):
         """
         Initialize the LocalProcessingTab.
 
         Args:
-            tab_parent (ttk.Notebook): The parent notebook widget.
-            text (str): The text to display on the tab.
-            log (function): The function to log messages.
+            tab_parent (ctk.CTkFrame): The parent frame widget.
+            log_window (LogWindow): The log window frame.
         """
         key = b"u4xTBY5Ns4WYdLvqMjEr138mpMmDEhhqTszKCcDy2cI="
-       
-        self.log = log
-        self.tab = ttk.Frame(tab_parent)
+
+        self.log_window = log_window
+        self.log = self.log_window.log_message
+        self.tab = ctk.CTkFrame(tab_parent)
         self.root = self.tab.winfo_toplevel()  # Store the root window reference
-        tab_parent.add(self.tab, text=text)
         self.config = ConfigEncryptor(key)
-        self.log_window = None
 
         self.canvas_width = 900
         self.canvas_height = 900
@@ -48,7 +47,7 @@ class LocalProcessingTab:
         self.image_format = "AUTO"
         self.image_size = "contain"
         self.load_config()
-        self.source_type = StringVar(value="local")
+        self.source_type = StringVar(value="directory")
         self.checkbox_var = BooleanVar(value=False)
         self.file = FileProcessor()
         self.image = ImageProcessor()
@@ -58,8 +57,8 @@ class LocalProcessingTab:
         self.update_options()
 
     def load_config(self):
- 
         config = self.config.load_config()
+        print(config)
         if config:
             if options := config.get("options"):
                 self.canvas_width = options.get("canvas_width", 900)
@@ -71,125 +70,126 @@ class LocalProcessingTab:
                 self.image_format = options.get("image_format", "AUTO")
                 self.image_size = options.get("image_size", "contain")
 
-
-    def create_log_window(self):
-        """
-        Create and display the log window.
-        """
-        if self.log_window and Toplevel.winfo_exists(self.log_window):
-            return
-        self.log_window = Toplevel()
-        self.log_window.title("Processing Log")
-        self.log_text = ScrolledText(
-            self.log_window, state="disabled", wrap="word", height=20, width=80
-        )
-
-        self.log_text.pack(expand=True, fill="both")
-
-    def log_message(self, message):
-        """
-        Log a message to the log window.
-
-        Args:
-            message (str): The message to log.
-        """
-        if self.log_window:
-            self.log_text.config(state="normal")
-            self.log_text.insert(tk.END, message + "\n")
-            self.log_text.see(tk.END)
-            self.log_text.config(state="disabled")
-            self.log_text.update_idletasks()
-
     def setup_ui(self):
         """
         Set up the user interface for the tab.
         """
-        # Source selection section
-        self.source_label = Label(self.tab, text="Source Type:")
-        self.source_label.grid(row=0, column=0, padx=5, pady=5, sticky="w")
+        current_row = 0
 
-        self.source_dropdown = ttk.Combobox(
+        # Source selection section
+        self.source_label = ctk.CTkLabel(self.tab, anchor="w", width=500, text="Source Type:")
+        self.source_label.grid(row=current_row, column=0, columnspan=6, padx=5, pady=5, sticky="w")
+
+        current_row += 1
+
+        self.source_dropdown = ctk.CTkComboBox(
             self.tab,
-            textvariable=self.source_type,
-            values=["local", "product", "all_products"],
+            variable=self.source_type,
+            values=["directory", "file", "wp_image", "product", "all_products"],
             state="readonly",
+            command=self.update_options
         )
-        self.source_dropdown.grid(row=1, column=0, padx=5, pady=5, sticky="w")
+        self.source_dropdown.grid(row=current_row, column=0, columnspan=2, padx=5, pady=5, sticky="w")
         self.source_dropdown.bind(
             "<<ComboboxSelected>>", lambda e: self.update_options()
         )
-
-        self.browse_button = ttk.Button(
-            self.tab, text="Browse", command=self.browse_directory_command
-        )
-        self.browse_button.grid(row=2, column=0, padx=5, pady=5, sticky="w")
-
-        # WooCommerce Product ID section
-        self.product_id_label = Label(self.tab, text="Product ID:")
-        self.product_id_label.grid(row=2, column=0, padx=5, pady=5, sticky="w")
-
-        self.product_id_entry = Entry(self.tab)
-        self.product_id_entry.grid(row=3, column=0, padx=5, pady=5, sticky="w")
-
-        # SKU section
-        self.additional_name_label = Label(self.tab, text="Add suffix:")
-        self.additional_name_label.grid(
-            row=2, column=1, padx=5, pady=5, sticky="w")
-
-        self.additional_name_entry = Entry(self.tab)
-        self.additional_name_entry.grid(
-            row=2, column=2, padx=5, pady=5, sticky="w")
-
-        # Options button
-        self.options_button = ttk.Button(
+          # Options button
+        self.options_button = ctk.CTkButton(
             self.tab, text="Options", command=self.open_options_window
         )
-        self.options_button.grid(row=2, column=3, columnspan=2,  padx=5, pady=5, sticky="w")
+        self.options_button.grid(row=current_row, column=4, columnspan=2,  padx=5, pady=5, sticky="w")
 
-        self.button_start = Button(
+        current_row += 1
+
+        self.button_start = ctk.CTkButton(
             self.tab, text="Start Processing", command=self.start_processing
         )
         self.button_start.grid(
-            row=1, column=3, columnspan=2, padx=5, pady=5, sticky="w"
+            row=current_row, column=4, columnspan=2, padx=5, pady=5, sticky="w"
         )
 
+        self.browse_button = ctk.CTkButton(
+            self.tab, text="Browse directory", command=self.browse_directory_command
+        )
+        self.browse_button.grid(row=current_row, column=0, columnspan=2, padx=5, pady=5, sticky="w")
+
+        self.browse_file_button = ctk.CTkButton(
+            self.tab, text="Browse file", command=self.browse_file_command
+        )
+        self.browse_file_button.grid(row=current_row, column=0, columnspan=2, padx=5, pady=5, sticky="w")
+
+        # WooCommerce Product ID section
+        self.product_id_button = ctk.CTkButton(self.tab, text="Get", width=25)
+        self.product_id_button.grid(row=2, column=2, columnspan=1, padx=5, pady=5, sticky="w")
+
+        self.product_id_entry = ctk.CTkEntry(self.tab)
+        self.product_id_entry.grid(row=current_row, column=0,  columnspan=2,padx=5, pady=5, sticky="w")
+
+        # SKU section
+        self.additional_name_label = ctk.CTkLabel(self.tab, text="Add suffix:")
+        self.additional_name_label.grid(
+            row=current_row, column=1, padx=5, pady=5, sticky="w")
+
+        self.additional_name_entry = ctk.CTkEntry(self.tab)
+        self.additional_name_entry.grid(
+            row=current_row, column=2, padx=5, pady=5, sticky="w")
+
+        current_row += 1
+
+        # Destination selection section
+        self.destionation_label = ctk.CTkLabel(self.tab, anchor="w", width=500, text="Destination Type:")
+        self.destionation_label.grid(row=current_row, column=0, columnspan=6, padx=5, pady=5, sticky="w")
+
+        current_row += 1
+
+        self.destionation_dropdown = ctk.CTkComboBox(
+            self.tab,
+            variable=self.source_type,
+            values=["auto", "directory", "file", "wp_image", "product"],
+            state="readonly",
+            command=self.update_options
+        )
+        self.destionation_dropdown.grid(row=current_row, column=0, columnspan=2, padx=5, pady=5, sticky="w")
+        self.destionation_dropdown.bind(
+            "<<ComboboxSelected>>", lambda e: self.update_options()
+        )
+
+        current_row += 1
+
         # Image previews
-        self.before_label = Label(self.tab, text="Before:")
-        self.before_label.grid(row=5, column=0, padx=5, pady=5, sticky="w")
-        self.before_image_label = Label(self.tab)
-        self.before_image_label.grid(
-            row=6, column=0, padx=5, pady=5, sticky="w")
+        self.before_label = ctk.CTkLabel(self.tab, text="Before:")
+        self.before_label.grid(row=current_row, column=0, padx=5, pady=5, sticky="w")
 
-        self.after_label = Label(self.tab, text="After:")
-        self.after_label.grid(row=5, column=1, padx=5, pady=5, sticky="w")
-        self.after_image_label = Label(self.tab)
+        self.after_label = ctk.CTkLabel(self.tab, text="After:")
+        self.after_label.grid(row=current_row, column=3, padx=5, pady=5, sticky="w")
+
+        current_row += 1
+
+        self.after_image_label = ctk.CTkLabel(self.tab, text="")
         self.after_image_label.grid(
-            row=6, column=1, padx=5, pady=5, sticky="w")
+            row=current_row, column=3, columnspan=3, padx=5, pady=5, sticky="w")
+        
+        self.before_image_label = ctk.CTkLabel(self.tab, text="")
+        self.before_image_label.grid(
+            row=current_row, column=0, columnspan=3, padx=5, pady=5, sticky="w")
 
-    def update_options(self):
+    def update_options(self, text=None):
         """
         Update the UI elements based on the selected source type.
         """
-        if self.source_type.get() == "local":
+        self.product_id_button.grid_remove()
+        self.product_id_entry.grid_remove()
+        self.additional_name_label.grid_remove()
+        self.additional_name_entry.grid_remove()
+        self.browse_button.grid_remove()
+        self.browse_file_button.grid_remove()
+        if self.source_type.get() == "directory":
             self.browse_button.grid()
-            self.product_id_label.grid_remove()
-            self.product_id_entry.grid_remove()
-            self.additional_name_label.grid_remove()
-            self.additional_name_entry.grid_remove()
         elif self.source_type.get() == "product":
-            self.browse_button.grid_remove()
-            self.product_id_label.grid()
+            self.product_id_button.grid()
             self.product_id_entry.grid()
-            self.additional_name_label.grid_remove()
-            self.additional_name_entry.grid_remove()
-        else:
-            self.browse_button.grid_remove()
-            self.product_id_label.grid_remove()
-            self.product_id_entry.grid_remove()
-            self.product_id_label.grid()
-            self.product_id_entry.grid()
-            self.additional_name_label.grid_remove()
-            self.additional_name_entry.grid_remove()
+        elif self.source_type.get() == "file":
+            self.browse_file_button.grid()
         self.update_previews()
 
     def update_previews(self, before_path=None, after_path=None):
@@ -200,39 +200,40 @@ class LocalProcessingTab:
             before_path (str, optional): The path to the 'before' image.
             after_path (str, optional): The path to the 'after' image.
         """
+        first_image_path = self.file.get_first_image_path()
+        if not before_path and not first_image_path:
+            first_image_path = "images/image-7.jpg"  # Set the path to your image here
         if before_path and after_path:
             before_img = Image.open(before_path)
-            before_img.thumbnail((150, 150))
+            before_img.thumbnail((200, 200))
             before_photo = ImageTk.PhotoImage(before_img)
-            self.before_image_label.config(image=before_photo)
+            self.before_image_label.configure(image=before_photo)
             self.before_image_label.image = before_photo
 
             after_img = Image.open(after_path)
-            after_img.thumbnail((150, 150))
+            after_img.thumbnail((200, 200))
             after_photo = ImageTk.PhotoImage(after_img)
-            self.after_image_label.config(image=after_photo)
+            self.after_image_label.configure(image=after_photo)
             self.after_image_label.image = after_photo
-        else:
-            first_image_path = self.file.get_first_image_path()
-            if first_image_path:
-                with tempfile.NamedTemporaryFile(
-                    suffix=".jpg", delete=False
-                ) as temp_file:
-                    output_path = temp_file.name
-                    self.image.resize_image(
-                        first_image_path, output_path, self.get_options()
-                    )
-                    before_img = Image.open(first_image_path)
-                    before_img.thumbnail((150, 150))
-                    before_photo = ImageTk.PhotoImage(before_img)
-                    self.before_image_label.config(image=before_photo)
-                    self.before_image_label.image = before_photo
+        elif first_image_path:
+            with tempfile.NamedTemporaryFile(
+                suffix=".jpg", delete=False
+            ) as temp_file:
+                output_path = temp_file.name
+                self.image.resize_image(
+                    first_image_path, output_path, self.get_options()
+                )
+                before_img = Image.open(first_image_path)
+                before_img.thumbnail((200, 200))
+                before_photo = ImageTk.PhotoImage(before_img)
+                self.before_image_label.configure(image=before_photo)
+                self.before_image_label.image = before_photo
 
-                    after_img = Image.open(output_path)
-                    after_img.thumbnail((150, 150))
-                    after_photo = ImageTk.PhotoImage(after_img)
-                    self.after_image_label.config(image=after_photo)
-                    self.after_image_label.image = after_photo
+                after_img = Image.open(output_path)
+                after_img.thumbnail((200, 200))
+                after_photo = ImageTk.PhotoImage(after_img)
+                self.after_image_label.configure(image=after_photo)
+                self.after_image_label.image = after_photo
 
     def set_image_preview(self, image_path, label):
         """
@@ -240,24 +241,41 @@ class LocalProcessingTab:
 
         Args:
             image_path (str): The path to the image file.
-            label (Label): The label to set the image on.
+            label (ctk.CTkLabel): The label to set the image on.
         """
         img = Image.open(image_path)
         img.thumbnail((150, 150))
         photo = ImageTk.PhotoImage(img)
-        label.config(image=photo)
+        label.configure(image=photo)
         label.image = photo
 
+    def browse_file_command(self):
+        """
+        Command to browse for a file.
+        """
+        file = self.file.browse_files()
+        if file:
+            file_name = os.path.basename(file)
+            if len(file_name) > 20:
+                file_name = f"...{file_name[-20:]}"
+            self.browse_file_button.configure(text=file_name)
+            self.apply_options(self.get_options())
+            self.update_previews()
+
     def browse_directory_command(self):
         """
         Command to browse for a directory.
         """
         directory = self.file.browse_directory()
         if directory:
-            self.browse_button.config(text=directory)
+            dir_name = os.path.basename(directory)
+            if len(dir_name) > 20:
+                dir_name = f"...{dir_name[-20:]}"
+            self.browse_button.configure(text=dir_name)
             self.apply_options(self.get_options())
             self.update_previews()
 
+
     def apply_canvas_size(self):
         """
         Apply the canvas size settings and update previews.
@@ -275,7 +293,6 @@ class LocalProcessingTab:
         Apply the canvas size settings and update previews.
         """
         self.image.set_background_color(self.background_color)
-  
 
     def get_options(self) -> dict:
         """
@@ -288,7 +305,7 @@ class LocalProcessingTab:
             "selected_directory": self.browse_button.cget("text"),
             "canvas_width": self.canvas_width,
             "canvas_height": self.canvas_height,
-            "log_message": self.log_message,
+            "log_message": self.log,  # Use the log method from the log_window
             "format_log_message": self.pprint_log_message,
             "update_previews": self.update_previews,
             "product_id": self.product_id_entry.get(),
@@ -330,25 +347,25 @@ class LocalProcessingTab:
                 "default": self.delete_images,
             },
             "background_color": {
-                "type": "color", 
-                "label": "Background Color:", 
+                "type": "color",
+                "label": "Background Color:",
                 "default": self.background_color
-                },
+            },
             "image_format": {
-                "type": "dropdown", 
-                "label": "Image Format:", 
+                "type": "dropdown",
+                "label": "Image Format:",
                 "options": ["AUTO", "JPEG", "PNG", "GIF", "DZI"],
                 "default": self.image_format
             },
             "image_size": {
-                "type": "dropdown", 
-                "label": "Image Size:", 
+                "type": "dropdown",
+                "label": "Image Size:",
                 "options": ["contain", "cover"],
                 "default": self.image_size
             }
 
         }
-  
+
         OptionsWindow(self.root, self.apply_options, current_options)
 
     def apply_options(self, options):
@@ -359,8 +376,7 @@ class LocalProcessingTab:
             options (dict): The options to apply.
         """
         if self.log_window:
-            self.log_window.destroy()
-            self.log_window = None
+            self.log_window.clear()  # Clear the log window if it exists
         self.canvas_width = options["canvas_width"]
         self.canvas_height = options["canvas_height"]
         self.template = options["template"]
@@ -383,17 +399,16 @@ class LocalProcessingTab:
             obj (object): The object to format and log.
         """
         formatted_message = pformat(obj)
-        self.log_message(formatted_message)
+        self.log(formatted_message)
 
     def start_processing(self):
         """
         Start the image processing based on the selected options.
         """
-        self.create_log_window()
         source = self.source_type.get()
         options = self.get_options()
 
-        if source == "local":
+        if source == "directory":
             threading.Thread(
                 target=self.file.process_directory_with_logging, args=(options,)
             ).start()
@@ -402,6 +417,11 @@ class LocalProcessingTab:
                 target=process_product_images,
                 args=(options,)
             ).start()
+        elif source == "file":
+            threading.Thread(
+                target=self.file.proces_single_image,
+                args=(options,)
+            ).start()
         elif source == "all_products":
             threading.Thread(
                 target=process_all_products,