import tkinter as tk from tkinter import filedialog, ttk, messagebox from PIL import Image as PILImage, ImageTk from wand.image import Image import os import queue import hashlib import threading import subprocess import time # Thêm import time # Biến toàn cục để điều khiển việc dừng và lưu lỗi stop_conversion = False error_messages = [] selected_files = [] converted_hashes = set() save_directory = "" pause_event = threading.Event() # Thêm biến kiểm soát tạm dừng def open_image_converter(): global stop_conversion, error_messages, selected_files, save_dir_var, format_var, status_var, num_files_var, errors_var, thread_count_var, filename_var, filter_var, progress, q, root global total_dimension_var, delete_original_var, stop_button_text # Thêm các biến mới # Tạo cửa sổ Tkinter root = tk.Tk() root.title("Image Converter") # Khởi tạo các biến Tkinter save_dir_var = tk.StringVar() format_var = tk.StringVar(value='png') status_var = tk.StringVar() num_files_var = tk.StringVar(value="0 files selected.") errors_var = tk.StringVar(value="Errors: 0") thread_count_var = tk.StringVar(value="1") filename_var = tk.StringVar() filter_var = tk.BooleanVar(value=False) progress = tk.IntVar(value=0) q = queue.Queue() total_dimension_var = tk.StringVar() # Biến cho tổng chiều dài và chiều rộng delete_original_var = tk.BooleanVar(value=False) # Biến cho ô tích xóa ảnh gốc stop_button_text = tk.StringVar(value="Stop") # Biến để thay đổi text nút Stop/Continue # Định nghĩa các định dạng được hỗ trợ SUPPORTED_TO_PS = ['pdf', 'jpeg', 'jpg', 'tiff', 'pnm'] SUPPORTED_TO_EPS = ['pdf', 'jpeg', 'jpg', 'tiff', 'pnm'] NEEDS_INTERMEDIATE_CONVERSION = { 'psd': 'pdf', 'svg': 'pdf', 'png': 'pdf', 'bmp': 'pdf', 'gif': 'pdf', 'ico': 'png' } # Các hàm trợ giúp def center_window(window): window.update_idletasks() width = window.winfo_width() 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(): filepaths = filedialog.askopenfilenames( title="Select Files", filetypes=[("All Image files", "*.jpg;*.jpeg;*.png;*.gif;*.bmp;*.tiff;*.tif;*.svg;*.webp;*.pdf;*.psd;*.ico"), ("JPEG files", "*.jpg;*.jpeg"), ("PNG files", "*.png"), ("GIF files", "*.gif"), ("BMP files", "*.bmp"), ("TIFF files", "*.tiff;*.tif"), ("SVG files", "*.svg"), ("WEBP files", "*.webp"), ("PDF files", "*.pdf"), ("PSD files", "*.psd"), ("ICO files", "*.ico")] ) if filepaths: selected_files.clear() selected_files.extend(filepaths) num_files_var.set(f"{len(selected_files)} files selected.") update_image_preview() def choose_save_directory(): global save_directory directory = filedialog.askdirectory() if directory: save_directory = directory save_dir_var.set(directory) save_dir_entry.config(state='normal') save_dir_entry.delete(0, tk.END) save_dir_entry.insert(0, directory) save_dir_entry.config(state='readonly') def hash_image(file_path): """Tạo hàm băm SHA-256 từ nội dung ảnh.""" hash_sha256 = hashlib.sha256() try: file_path = os.path.normpath(file_path) with open(file_path, "rb") as f: for chunk in iter(lambda: f.read(4096), b""): hash_sha256.update(chunk) except Exception as e: print(f"Error hashing file {file_path}: {e}") return None return hash_sha256.hexdigest() def filter_duplicate_images(filepaths): unique_images = {} filtered_files = [] for filepath in filepaths: image_hash = hash_image(filepath) if image_hash and image_hash not in unique_images: if image_hash not in converted_hashes: unique_images[image_hash] = filepath filtered_files.append(filepath) return filtered_files def can_convert_directly(input_format, output_format): """Kiểm tra xem có thể chuyển đổi trực tiếp từ định dạng nguồn sang định dạng đích không.""" if output_format == 'ps': return input_format in SUPPORTED_TO_PS elif output_format == 'eps': return input_format in SUPPORTED_TO_EPS elif output_format == 'ico': return input_format not in ['ps', 'eps', 'pdf'] elif input_format == 'ico': return output_format not in ['ps', 'eps'] return True def notify_conversion_path(input_format, output_format): """Hiển thị thông báo nhắc nhở người dùng về bước chuyển đổi trung gian cần thiết.""" intermediate_format = NEEDS_INTERMEDIATE_CONVERSION.get(input_format, None) if intermediate_format: if output_format in ['ps', 'eps']: message = (f"Cannot convert directly from {input_format.upper()} to {output_format.upper()}. " f"Please convert to PDF first, then convert to {output_format.upper()}.") else: message = (f"Cannot convert directly from {input_format.upper()} to {output_format.upper()}. " f"Please convert to {intermediate_format.upper()} first, then convert to {output_format.upper()}.") else: message = (f"Conversion from {input_format.upper()} to {output_format.upper()} is not supported. " "Please use a different format for conversion.") messagebox.showwarning("Conversion Notice", message) def convert_image_with_wand(input_path, output_format, output_path, total_dimension=None): """Chuyển đổi ảnh sử dụng ImageMagick thông qua Wand.""" try: with Image(filename=input_path) as img: # Thay đổi kích thước nếu total_dimension được cung cấp if total_dimension: width = img.width height = img.height aspect_ratio = width / height total_dim = float(total_dimension) new_height = int(total_dim / (aspect_ratio + 1)) new_width = int(aspect_ratio * new_height) img.resize(new_width, new_height) img.format = output_format img.save(filename=output_path) except Exception as e: raise RuntimeError(f"Error converting {input_path} to {output_format}: {e}") def convert_image_with_ghostscript(input_path, output_format, output_path): """Chuyển đổi ảnh sử dụng Ghostscript.""" try: gs_command = [ "gswin64c", "-dBATCH", "-dNOPAUSE", "-sDEVICE=" + ("ps2write" if output_format == "ps" else "eps2write"), f"-sOutputFile={output_path}", input_path ] subprocess.run(gs_command, check=True, creationflags=subprocess.CREATE_NO_WINDOW) except subprocess.CalledProcessError as e: raise RuntimeError(f"Ghostscript error: {e}") def convert_image(input_path, save_directory, output_format, output_filename, q, total_dimension, delete_original): if stop_conversion: return filename = os.path.basename(input_path) name, ext = os.path.splitext(filename) ext = ext[1:].lower() original_name = name try: if output_filename: name = output_filename output_path = os.path.join(save_directory, f"{name}.{output_format}") counter = 1 while os.path.exists(output_path): new_name = f"{name} ({counter})" output_path = os.path.join(save_directory, f"{new_name}.{output_format}") counter += 1 if ext in NEEDS_INTERMEDIATE_CONVERSION and output_format in ["ps", "eps"]: intermediate_format = NEEDS_INTERMEDIATE_CONVERSION[ext] intermediate_path = os.path.join(save_directory, f"{name}.{intermediate_format}") # Chuyển đổi sang định dạng trung gian convert_image_with_wand(input_path, intermediate_format, intermediate_path, total_dimension) # Chuyển đổi từ định dạng trung gian sang định dạng cuối if output_format in ["ps", "eps"]: convert_image_with_ghostscript(intermediate_path, output_format, output_path) else: convert_image_with_wand(intermediate_path, output_format, output_path, total_dimension) os.remove(intermediate_path) else: if output_format in ["ps", "eps"]: convert_image_with_ghostscript(input_path, output_format, output_path) else: convert_image_with_wand(input_path, output_format, output_path, total_dimension) q.put(input_path) converted_hashes.add(hash_image(output_path)) # Xóa ảnh gốc nếu được chọn if delete_original: os.remove(input_path) except Exception as e: error_message = f"Error converting {filename}: {str(e)}" q.put(error_message) error_messages.append(error_message) def worker(save_directory, output_format, num_threads, output_filename, q, filter_duplicates, total_dimension, delete_original): try: total_files = selected_files if filter_duplicates: total_files = filter_duplicate_images(selected_files) progress.set(0) for i, input_path in enumerate(total_files, 1): if stop_conversion: break while pause_event.is_set(): time.sleep(0.1) if stop_conversion: break input_format = os.path.splitext(input_path)[1][1:].lower() if not can_convert_directly(input_format, output_format): notify_conversion_path(input_format, output_format) q.put(None) return thread = threading.Thread(target=convert_image, args=(input_path, save_directory, output_format, output_filename, q, total_dimension, delete_original)) thread.start() thread.join() q.put(None) except Exception as e: if not stop_conversion: q.put(e) def update_progress(): try: completed = 0 while True: item = q.get() if item is None: break if isinstance(item, str): if "Error" in item: root.after(0, errors_var.set, f"Errors: {len(error_messages)}") continue completed += 1 progress.set(int((completed / len(selected_files)) * 100)) if not stop_conversion: root.after(0, status_var.set, f"Converted {completed} files") root.after(0, root.update_idletasks) if not stop_conversion: root.after(0, progress.set, 100) show_completion_message(completed) except Exception as e: if not stop_conversion: root.after(0, status_var.set, f"Error: {e}") def show_completion_message(completed): message = f"Conversion complete. {completed} files converted." if error_messages: message += f" {len(error_messages)} errors occurred." messagebox.showinfo("Conversion Complete", message) def convert_files(): global stop_conversion, error_messages stop_conversion = False pause_event.clear() stop_button_text.set("Stop") error_messages.clear() errors_var.set("Errors: 0") save_directory = save_dir_var.get() output_format = format_var.get() try: num_threads = int(thread_count_var.get() or 4) except ValueError: messagebox.showerror("Input Error", "Threads must be a number.") return output_filename = filename_var.get() filter_duplicates = filter_var.get() total_dimension = total_dimension_var.get() or None delete_original = delete_original_var.get() if not selected_files or not save_directory or not output_format: status_var.set("Please select images, output format, and save location.") return threading.Thread(target=worker, args=(save_directory, output_format, num_threads, output_filename, q, filter_duplicates, total_dimension, delete_original)).start() threading.Thread(target=update_progress).start() def stop_conversion_func(): global stop_conversion if stop_button_text.get() == "Stop": pause_event.set() stop_button_text.set("Continue") status_var.set("Conversion paused.") else: pause_event.clear() stop_button_text.set("Stop") status_var.set("Conversion resumed.") def return_to_menu(): stop_conversion_func() root.destroy() import main main.open_main_menu() def on_closing(): return_to_menu() error_window = None # Thêm biến để kiểm soát cửa sổ lỗi 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) def update_image_preview(): for widget in image_preview_frame.winfo_children(): widget.destroy() for i, file_path in enumerate(selected_files): thumbnail_size = (100, 100) try: image = PILImage.open(file_path) image.thumbnail(thumbnail_size) thumbnail = ImageTk.PhotoImage(image) tk.Label(image_preview_frame, image=thumbnail).grid(row=i, column=0, padx=5, pady=5) tk.Label(image_preview_frame, text=os.path.basename(file_path)).grid(row=i, column=1, padx=5, pady=5) tk.Label(image_preview_frame, text="Caption placeholder", wraplength=300).grid(row=i, column=2, padx=5, pady=5) image_preview_frame.image = thumbnail # Giữ tham chiếu để tránh bị xóa except Exception as e: tk.Label(image_preview_frame, text="Error loading image").grid(row=i, column=0, columnspan=3, padx=5, pady=5) def validate_number(P): if P.isdigit() or P == "": return True else: messagebox.showerror("Input Error", "Please enter only numbers.") return False validate_command = root.register(validate_number) # Tạo các thành phần giao diện main_frame = tk.Frame(root) main_frame.pack(fill=tk.BOTH, expand=True) control_frame = tk.Frame(main_frame) control_frame.pack(fill=tk.X, padx=10, pady=10) back_button = tk.Button(control_frame, text="<-", font=('Helvetica', 14), command=return_to_menu) back_button.pack(side=tk.TOP, anchor='w', padx=5, pady=5) title_label = tk.Label(control_frame, text="Image Converter", font=('Helvetica', 16)) title_label.pack(side=tk.TOP, padx=5, pady=5) select_button = tk.Button(control_frame, text="Select Files", command=select_files) select_button.pack(side=tk.TOP, padx=5, pady=5) num_files_label = tk.Label(control_frame, textvariable=num_files_var) num_files_label.pack(side=tk.TOP, padx=5, pady=5) save_dir_button = tk.Button(control_frame, text="Choose Save Directory", command=choose_save_directory) save_dir_button.pack(side=tk.TOP, padx=5, pady=5) save_dir_entry = tk.Entry(control_frame, textvariable=save_dir_var, state='readonly', justify='center') save_dir_entry.pack(side=tk.TOP, padx=5, pady=5, fill=tk.X) format_frame = tk.Frame(main_frame) format_frame.pack(fill=tk.X, padx=10, pady=5) format_label = tk.Label(format_frame, text="Output Format:") format_label.pack(side=tk.LEFT, padx=5) format_dropdown = ttk.Combobox(format_frame, textvariable=format_var, values=['png', 'jpg', 'gif', 'bmp', 'tiff', 'svg', 'webp', 'pdf', 'psd', 'ps', 'eps', 'ico']) format_dropdown.pack(side=tk.LEFT, padx=5) thread_count_label = tk.Label(format_frame, text="Threads:") thread_count_label.pack(side=tk.LEFT, padx=5) thread_count_entry = tk.Entry(format_frame, textvariable=thread_count_var, width=5, validate="key", validatecommand=(validate_command, '%P'), justify='center') thread_count_entry.pack(side=tk.LEFT, padx=5) filename_label = tk.Label(main_frame, text="Output Filename (optional):") filename_label.pack(fill=tk.X, padx=10, pady=5) filename_entry = tk.Entry(main_frame, textvariable=filename_var, justify='center') filename_entry.pack(fill=tk.X, padx=10, pady=5) # Thêm ô nhập cho tổng chiều dài và chiều rộng total_dimension_label = tk.Label(main_frame, text="Nhập tổng chiều dài và chiều rộng (optional):") total_dimension_label.pack(fill=tk.X, padx=10, pady=5) total_dimension_entry = tk.Entry(main_frame, textvariable=total_dimension_var, justify='center', validate="key", validatecommand=(validate_command, '%P')) total_dimension_entry.pack(fill=tk.X, padx=10, pady=5) filter_frame = tk.Frame(main_frame) filter_frame.pack(fill=tk.X, padx=10, pady=5) filter_checkbox = tk.Checkbutton(filter_frame, text="Filter duplicate images", variable=filter_var) filter_checkbox.pack(side=tk.LEFT) # Thêm ô tích để xóa ảnh gốc sau khi chuyển đổi delete_original_checkbox = tk.Checkbutton(filter_frame, text="Xóa ảnh gốc sau khi chuyển đổi", variable=delete_original_var) delete_original_checkbox.pack(side=tk.LEFT, padx=5) convert_button = tk.Button(main_frame, text="Convert", command=convert_files) convert_button.pack(pady=10) stop_button = tk.Button(main_frame, textvariable=stop_button_text, command=stop_conversion_func) stop_button.pack(pady=5) errors_button = tk.Button(main_frame, textvariable=errors_var, command=show_errors) errors_button.pack(pady=5) progress_bar = ttk.Progressbar(main_frame, variable=progress, maximum=100) progress_bar.pack(pady=5, fill=tk.X) status_label = tk.Label(main_frame, textvariable=status_var, fg="green") status_label.pack(pady=5) # Khung hiển thị ảnh và caption image_preview_frame = tk.Frame(root) image_preview_frame.pack(side=tk.BOTTOM, fill=tk.BOTH, expand=True) center_window(root) root.protocol("WM_DELETE_WINDOW", on_closing) root.mainloop() if __name__ == "__main__": open_image_converter()