Update shuffle_image.py

shuffle_image.py

@@ -1,285 +1,285 @@
-import tkinter as tk
-from tkinter import filedialog, ttk, messagebox
-import os
-import threading
-import queue
-import random
-import shutil
-
-# Biến toàn cục để điều khiển việc dừng và lưu lỗi
-stop_processing = False
-error_messages = []
-error_window = None
-selected_files = []
-save_directory = ""
-random_selected_files = []
-shuffle_enabled = True  # Biến để lưu trạng thái xáo trộn
-
-def open_image_shuffle():
-    global stop_processing, error_messages, error_window, selected_files, save_dir_var, status_var, num_files_var, errors_var, thread_count_var, progress, q, random_selected_files, random_file_count_var, shuffle_enabled
-
-    # Tạo cửa sổ Tkinter
-    root = tk.Tk()
-    root.title("Image Shuffle")
-
-    # Khởi tạo các biến Tkinter
-    save_dir_var = tk.StringVar()
-    status_var = tk.StringVar()
-    num_files_var = tk.StringVar()
-    errors_var = tk.StringVar(value="Errors: 0")
-    thread_count_var = tk.StringVar(value="4")
-    random_file_count_var = tk.StringVar(value="0")
-    progress = tk.IntVar()
-    q = queue.Queue()
-
-    def center_window(window):
-        window.update_idletasks()
-        width = window.winfo_width() + 120
-        height = window.winfo_height()
-        x = (window.winfo_screenwidth() // 2) - (width // 2)
-        y = (window.winfo_screenheight() // 2) - (height // 2)
-        window.geometry(f'{width}x{height}+{x}+{y}')
-
-    def select_files():
-        global selected_files
-        filetypes = [
-            ("All Image files", "*.jpg;*.jpeg;*.png;*.gif;*.bmp;*.tiff;*.tif;*.svg;*.webp"),
-            ("JPEG files", "*.jpg;*.jpeg"),
-            ("PNG files", "*.png"),
-            ("GIF files", "*.gif"),
-            ("BMP files", "*.bmp"),
-            ("TIFF files", "*.tiff;*.tif"),
-            ("SVG files", "*.svg"),
-            ("WEBP files", "*.webp")
-        ]
-        filepaths = filedialog.askopenfilenames(title="Select Image Files", filetypes=filetypes)
-        if filepaths:
-            selected_files.clear()
-            selected_files.extend(filepaths)
-            num_files_var.set(f"{len(selected_files)} files selected.")
-
-    def choose_directory():
-        global save_directory
-        directory = filedialog.askdirectory()
-        if directory:
-            save_directory = directory
-            save_dir_var.set(directory)
-
-    def shuffle_and_save_images(num_random):
-        """Chọn ngẫu nhiên và lưu ảnh."""
-        try:
-            if num_random > len(selected_files):
-                messagebox.showerror("Error", "Not enough images to select.")
-                return
-            
-            random_selected_files = random.sample(selected_files, num_random)
-            if shuffle_enabled:
-                random.shuffle(random_selected_files)  # Xáo trộn danh sách file nếu được bật
-
-            for i, original_path in enumerate(random_selected_files):
-                filename = os.path.basename(original_path)
-                new_filename = f"shuffled_{i+1:03d}_{filename}"  # Tạo tên file mới với tiền tố shuffled và số thứ tự
-                new_path = os.path.join(save_directory, new_filename)
-                shutil.copy(original_path, new_path)
-                q.put(i + 1)  # Gửi tiến trình hoàn thành cho hàng đợi
-
-            q.put(None)  # Thông báo hoàn thành
-        except Exception as e:
-            q.put(e)  # Gửi lỗi vào hàng đợi
-
-    def select_and_save_sequentially(num_random):
-        """Chọn và lưu ảnh theo thứ tự."""
-        try:
-            if num_random > len(selected_files):
-                messagebox.showerror("Error", "Not enough images to select.")
-                return
-
-            step = len(selected_files) // num_random
-            sequential_selected_files = selected_files[::step]
-
-            for i, original_path in enumerate(sequential_selected_files[:num_random]):
-                filename = os.path.basename(original_path)
-                new_filename = f"sequential_{i+1:03d}_{filename}"  # Tạo tên file mới với tiền tố sequential và số thứ tự
-                new_path = os.path.join(save_directory, new_filename)
-                shutil.copy(original_path, new_path)
-                q.put(i + 1)  # Gửi tiến trình hoàn thành cho hàng đợi
-
-            q.put(None)  # Thông báo hoàn thành
-        except Exception as e:
-            q.put(e)  # Gửi lỗi vào hàng đợi
-
-    def worker(num_random):
-        try:
-            progress.set(0)
-            if shuffle_enabled:
-                shuffle_and_save_images(num_random)
-            else:
-                select_and_save_sequentially(num_random)
-        except Exception as e:
-            q.put(e)  # Gửi lỗi vào hàng đợi
-
-    def update_progress():
-        try:
-            completed = 0
-            total_files = int(random_file_count_var.get())
-            while True:
-                item = q.get()
-                if item is None:
-                    break
-                if isinstance(item, Exception):
-                    raise item
-                completed = item
-                progress.set(int((completed / total_files) * 100))
-                if not stop_processing:
-                    root.after(0, status_var.set, f"Processed {completed} of {total_files} files")
-                    root.after(0, root.update_idletasks)
-            if not stop_processing:
-                root.after(0, progress.set(100))
-                show_completion_message(completed)
-        except Exception as e:
-            root.after(0, status_var.set, f"Error: {e}")
-
-    def show_completion_message(completed):
-        message = f"Processing complete. {completed} files processed."
-        if error_messages:
-            message += f" {len(error_messages)} errors occurred."
-        messagebox.showinfo("Process Complete", message)
-
-    def validate_and_process():
-        global stop_processing, error_messages
-        stop_processing = False
-        error_messages.clear()
-        errors_var.set("Errors: 0")
-        if not selected_files or not save_directory:
-            status_var.set("Please select images and save location.")
-            return
-
-        num_random = int(random_file_count_var.get())
-
-        threading.Thread(target=worker, args=(num_random,)).start()
-        threading.Thread(target=update_progress).start()
-
-    def validate_number(P):
-        return P.isdigit() or P == ""
-
-    def toggle_shuffle():
-        global shuffle_enabled
-        shuffle_enabled = shuffle_var.get() == 1
-        if shuffle_enabled:
-            shuffle_and_save_button.config(text="Shuffle")
-            random_file_count_label.config(text="Number of Random Files:")
-        else:
-            shuffle_and_save_button.config(text="Sequentially")
-            random_file_count_label.config(text="Number of Files to Select:")
-
-    def stop_processing_func():
-        global stop_processing
-        stop_processing = True
-        status_var.set("Processing stopped.")
-
-    def return_to_menu():
-        stop_processing_func()
-        root.destroy()
-        import main
-        main.open_main_menu()
-
-    def on_closing():
-        return_to_menu()
-
-    def show_errors():
-        global error_window
-        if error_window is not None:
-            return
-
-        error_window = tk.Toplevel(root)
-        error_window.title("Error Details")
-        error_window.geometry("500x400")
-
-        error_text = tk.Text(error_window, wrap='word')
-        error_text.pack(expand=True, fill='both')
-
-        if error_messages:
-            for error in error_messages:
-                error_text.insert('end', error + '\n')
-        else:
-            error_text.insert('end', "No errors recorded.")
-
-        error_text.config(state='disabled')
-
-        def on_close_error_window():
-            global error_window
-            error_window.destroy()
-            error_window = None
-
-        error_window.protocol("WM_DELETE_WINDOW", on_close_error_window)
-
-    # Tạo các thành phần giao diện
-    validate_command = root.register(validate_number)
-
-    back_button = tk.Button(root, text="<-", font=('Helvetica', 14), command=return_to_menu)
-    back_button.pack(anchor='nw', padx=10, pady=10)
-
-    title_label = tk.Label(root, text="Image Shuffle", font=('Helvetica', 16))
-    title_label.pack(pady=10)
-
-    select_files_button = tk.Button(root, text="Select Files", command=select_files)
-    select_files_button.pack(pady=5)
-
-    num_files_label = tk.Label(root, textvariable=num_files_var)
-    num_files_label.pack(pady=5)
-
-    choose_dir_button = tk.Button(root, text="Choose Save Directory", command=choose_directory)
-    choose_dir_button.pack(pady=10)
-
-    save_dir_entry = tk.Entry(root, textvariable=save_dir_var, state='readonly', justify='center')
-    save_dir_entry.pack(pady=5, fill=tk.X)
-
-    # Mô tả về các lựa chọn Shuffle và Sequential
-    description_label = tk.Label(root, text="Choose to shuffle or select sequentially the images before saving them:", font=('Helvetica', 10))
-    description_label.pack(pady=5)
-
-    # Khung cho hai lựa chọn Shuffle và Sequential
-    option_frame = tk.Frame(root)
-    option_frame.pack(pady=5)
-
-    shuffle_var = tk.IntVar(value=1)  # Biến để lưu trạng thái xáo trộn
-
-    shuffle_radio_button = tk.Radiobutton(option_frame, text="Shuffle", variable=shuffle_var, value=1, command=toggle_shuffle)
-    shuffle_radio_button.pack(side='left', padx=5)
-
-    sequential_radio_button = tk.Radiobutton(option_frame, text="Sequential", variable=shuffle_var, value=0, command=toggle_shuffle)
-    sequential_radio_button.pack(side='left', padx=5)
-
-    random_file_count_label = tk.Label(root, text="Number of Random Files:")
-    random_file_count_label.pack(pady=5)
-
-    random_file_count_entry = tk.Entry(root, textvariable=random_file_count_var, width=10, justify='center', validate="key", validatecommand=(validate_command, '%P'))
-    random_file_count_entry.pack(pady=5)
-
-    shuffle_and_save_button = tk.Button(root, text="Shuffle", command=validate_and_process)
-    shuffle_and_save_button.pack(pady=10)
-
-    thread_count_label = tk.Label(root, text="Number of Threads:")
-    thread_count_label.pack(pady=5)
-
-    thread_count_entry = tk.Entry(root, textvariable=thread_count_var, width=5, justify='center', validate="key", validatecommand=(validate_command, '%P'))
-    thread_count_entry.pack(pady=5)
-
-    stop_button = tk.Button(root, text="Stop", command=stop_processing_func)
-    stop_button.pack(pady=5)
-
-    errors_button = tk.Button(root, textvariable=errors_var, command=show_errors)
-    errors_button.pack(pady=5)
-
-    progress_bar = ttk.Progressbar(root, variable=progress, maximum=100)
-    progress_bar.pack(pady=5, fill=tk.X)
-
-    status_label = tk.Label(root, textvariable=status_var, fg="green")
-    status_label.pack(pady=5)
-
-    center_window(root)
-    root.protocol("WM_DELETE_WINDOW", on_closing)
-    root.mainloop()
-
-if __name__ == "__main__":
-    open_image_shuffle()
+import tkinter as tk
+from tkinter import filedialog, ttk, messagebox
+import os
+import threading
+import queue
+import random
+import shutil
+
+# Biến toàn cục để điều khiển việc dừng và lưu lỗi
+stop_processing = False
+error_messages = []
+error_window = None
+selected_files = []
+save_directory = ""
+random_selected_files = []
+shuffle_enabled = True  # Biến để lưu trạng thái xáo trộn
+
+def open_image_shuffle():
+    global stop_processing, error_messages, error_window, selected_files, save_dir_var, status_var, num_files_var, errors_var, thread_count_var, progress, q, random_selected_files, random_file_count_var, shuffle_enabled
+
+    # Tạo cửa sổ Tkinter
+    root = tk.Tk()
+    root.title("Image Shuffle")
+
+    # Khởi tạo các biến Tkinter
+    save_dir_var = tk.StringVar()
+    status_var = tk.StringVar()
+    num_files_var = tk.StringVar()
+    errors_var = tk.StringVar(value="Errors: 0")
+    thread_count_var = tk.StringVar(value="1")
+    random_file_count_var = tk.StringVar(value="0")
+    progress = tk.IntVar()
+    q = queue.Queue()
+
+    def center_window(window):
+        window.update_idletasks()
+        width = window.winfo_width() + 120
+        height = window.winfo_height()
+        x = (window.winfo_screenwidth() // 2) - (width // 2)
+        y = (window.winfo_screenheight() // 2) - (height // 2)
+        window.geometry(f'{width}x{height}+{x}+{y}')
+
+    def select_files():
+        global selected_files
+        filetypes = [
+            ("All Image files", "*.jpg;*.jpeg;*.png;*.gif;*.bmp;*.tiff;*.tif;*.svg;*.webp"),
+            ("JPEG files", "*.jpg;*.jpeg"),
+            ("PNG files", "*.png"),
+            ("GIF files", "*.gif"),
+            ("BMP files", "*.bmp"),
+            ("TIFF files", "*.tiff;*.tif"),
+            ("SVG files", "*.svg"),
+            ("WEBP files", "*.webp")
+        ]
+        filepaths = filedialog.askopenfilenames(title="Select Image Files", filetypes=filetypes)
+        if filepaths:
+            selected_files.clear()
+            selected_files.extend(filepaths)
+            num_files_var.set(f"{len(selected_files)} files selected.")
+
+    def choose_directory():
+        global save_directory
+        directory = filedialog.askdirectory()
+        if directory:
+            save_directory = directory
+            save_dir_var.set(directory)
+
+    def shuffle_and_save_images(num_random):
+        """Chọn ngẫu nhiên và lưu ảnh."""
+        try:
+            if num_random > len(selected_files):
+                messagebox.showerror("Error", "Not enough images to select.")
+                return
+            
+            random_selected_files = random.sample(selected_files, num_random)
+            if shuffle_enabled:
+                random.shuffle(random_selected_files)  # Xáo trộn danh sách file nếu được bật
+
+            for i, original_path in enumerate(random_selected_files):
+                filename = os.path.basename(original_path)
+                new_filename = f"shuffled_{i+1:03d}_{filename}"  # Tạo tên file mới với tiền tố shuffled và số thứ tự
+                new_path = os.path.join(save_directory, new_filename)
+                shutil.copy(original_path, new_path)
+                q.put(i + 1)  # Gửi tiến trình hoàn thành cho hàng đợi
+
+            q.put(None)  # Thông báo hoàn thành
+        except Exception as e:
+            q.put(e)  # Gửi lỗi vào hàng đợi
+
+    def select_and_save_sequentially(num_random):
+        """Chọn và lưu ảnh theo thứ tự."""
+        try:
+            if num_random > len(selected_files):
+                messagebox.showerror("Error", "Not enough images to select.")
+                return
+
+            step = len(selected_files) // num_random
+            sequential_selected_files = selected_files[::step]
+
+            for i, original_path in enumerate(sequential_selected_files[:num_random]):
+                filename = os.path.basename(original_path)
+                new_filename = f"sequential_{i+1:03d}_{filename}"  # Tạo tên file mới với tiền tố sequential và số thứ tự
+                new_path = os.path.join(save_directory, new_filename)
+                shutil.copy(original_path, new_path)
+                q.put(i + 1)  # Gửi tiến trình hoàn thành cho hàng đợi
+
+            q.put(None)  # Thông báo hoàn thành
+        except Exception as e:
+            q.put(e)  # Gửi lỗi vào hàng đợi
+
+    def worker(num_random):
+        try:
+            progress.set(0)
+            if shuffle_enabled:
+                shuffle_and_save_images(num_random)
+            else:
+                select_and_save_sequentially(num_random)
+        except Exception as e:
+            q.put(e)  # Gửi lỗi vào hàng đợi
+
+    def update_progress():
+        try:
+            completed = 0
+            total_files = int(random_file_count_var.get())
+            while True:
+                item = q.get()
+                if item is None:
+                    break
+                if isinstance(item, Exception):
+                    raise item
+                completed = item
+                progress.set(int((completed / total_files) * 100))
+                if not stop_processing:
+                    root.after(0, status_var.set, f"Processed {completed} of {total_files} files")
+                    root.after(0, root.update_idletasks)
+            if not stop_processing:
+                root.after(0, progress.set(100))
+                show_completion_message(completed)
+        except Exception as e:
+            root.after(0, status_var.set, f"Error: {e}")
+
+    def show_completion_message(completed):
+        message = f"Processing complete. {completed} files processed."
+        if error_messages:
+            message += f" {len(error_messages)} errors occurred."
+        messagebox.showinfo("Process Complete", message)
+
+    def validate_and_process():
+        global stop_processing, error_messages
+        stop_processing = False
+        error_messages.clear()
+        errors_var.set("Errors: 0")
+        if not selected_files or not save_directory:
+            status_var.set("Please select images and save location.")
+            return
+
+        num_random = int(random_file_count_var.get())
+
+        threading.Thread(target=worker, args=(num_random,)).start()
+        threading.Thread(target=update_progress).start()
+
+    def validate_number(P):
+        return P.isdigit() or P == ""
+
+    def toggle_shuffle():
+        global shuffle_enabled
+        shuffle_enabled = shuffle_var.get() == 1
+        if shuffle_enabled:
+            shuffle_and_save_button.config(text="Shuffle")
+            random_file_count_label.config(text="Number of Random Files:")
+        else:
+            shuffle_and_save_button.config(text="Sequentially")
+            random_file_count_label.config(text="Number of Files to Select:")
+
+    def stop_processing_func():
+        global stop_processing
+        stop_processing = True
+        status_var.set("Processing stopped.")
+
+    def return_to_menu():
+        stop_processing_func()
+        root.destroy()
+        import main
+        main.open_main_menu()
+
+    def on_closing():
+        return_to_menu()
+
+    def show_errors():
+        global error_window
+        if error_window is not None:
+            return
+
+        error_window = tk.Toplevel(root)
+        error_window.title("Error Details")
+        error_window.geometry("500x400")
+
+        error_text = tk.Text(error_window, wrap='word')
+        error_text.pack(expand=True, fill='both')
+
+        if error_messages:
+            for error in error_messages:
+                error_text.insert('end', error + '\n')
+        else:
+            error_text.insert('end', "No errors recorded.")
+
+        error_text.config(state='disabled')
+
+        def on_close_error_window():
+            global error_window
+            error_window.destroy()
+            error_window = None
+
+        error_window.protocol("WM_DELETE_WINDOW", on_close_error_window)
+
+    # Tạo các thành phần giao diện
+    validate_command = root.register(validate_number)
+
+    back_button = tk.Button(root, text="<-", font=('Helvetica', 14), command=return_to_menu)
+    back_button.pack(anchor='nw', padx=10, pady=10)
+
+    title_label = tk.Label(root, text="Image Shuffle", font=('Helvetica', 16))
+    title_label.pack(pady=10)
+
+    select_files_button = tk.Button(root, text="Select Files", command=select_files)
+    select_files_button.pack(pady=5)
+
+    num_files_label = tk.Label(root, textvariable=num_files_var)
+    num_files_label.pack(pady=5)
+
+    choose_dir_button = tk.Button(root, text="Choose Save Directory", command=choose_directory)
+    choose_dir_button.pack(pady=10)
+
+    save_dir_entry = tk.Entry(root, textvariable=save_dir_var, state='readonly', justify='center')
+    save_dir_entry.pack(pady=5, fill=tk.X)
+
+    # Mô tả về các lựa chọn Shuffle và Sequential
+    description_label = tk.Label(root, text="Choose to shuffle or select sequentially the images before saving them:", font=('Helvetica', 10))
+    description_label.pack(pady=5)
+
+    # Khung cho hai lựa chọn Shuffle và Sequential
+    option_frame = tk.Frame(root)
+    option_frame.pack(pady=5)
+
+    shuffle_var = tk.IntVar(value=1)  # Biến để lưu trạng thái xáo trộn
+
+    shuffle_radio_button = tk.Radiobutton(option_frame, text="Shuffle", variable=shuffle_var, value=1, command=toggle_shuffle)
+    shuffle_radio_button.pack(side='left', padx=5)
+
+    sequential_radio_button = tk.Radiobutton(option_frame, text="Sequential", variable=shuffle_var, value=0, command=toggle_shuffle)
+    sequential_radio_button.pack(side='left', padx=5)
+
+    random_file_count_label = tk.Label(root, text="Number of Random Files:")
+    random_file_count_label.pack(pady=5)
+
+    random_file_count_entry = tk.Entry(root, textvariable=random_file_count_var, width=10, justify='center', validate="key", validatecommand=(validate_command, '%P'))
+    random_file_count_entry.pack(pady=5)
+
+    shuffle_and_save_button = tk.Button(root, text="Shuffle", command=validate_and_process)
+    shuffle_and_save_button.pack(pady=10)
+
+    thread_count_label = tk.Label(root, text="Number of Threads:")
+    thread_count_label.pack(pady=5)
+
+    thread_count_entry = tk.Entry(root, textvariable=thread_count_var, width=5, justify='center', validate="key", validatecommand=(validate_command, '%P'))
+    thread_count_entry.pack(pady=5)
+
+    stop_button = tk.Button(root, text="Stop", command=stop_processing_func)
+    stop_button.pack(pady=5)
+
+    errors_button = tk.Button(root, textvariable=errors_var, command=show_errors)
+    errors_button.pack(pady=5)
+
+    progress_bar = ttk.Progressbar(root, variable=progress, maximum=100)
+    progress_bar.pack(pady=5, fill=tk.X)
+
+    status_label = tk.Label(root, textvariable=status_var, fg="green")
+    status_label.pack(pady=5)
+
+    center_window(root)
+    root.protocol("WM_DELETE_WINDOW", on_closing)
+    root.mainloop()
+
+if __name__ == "__main__":
+    open_image_shuffle()