HMI_Translate_Helper_wMaster_in_Excel/menu_pasos_traduccion.py

import tkinter as tk
from tkinter import ttk, filedialog, messagebox
import PyLibrary.funciones_comunes as fc
import x1_importar_to_master
import x2_master_export2translate
import x3_llm_auto_translate
import x4_integrate_translates_to_master
import x4B_integrate_manual_translates_to_master
import x5_complete_empty_cells_master
import x6_update_from_master
from translation_config import TranslationConfig
import sys
import threading
import os
import subprocess

class TranslationApp:
    def __init__(self, master):
        self.master = master
        self.master.title("Ayuda para traducir textos de TIA Portal y Allen Bradley")
        self.master.geometry("1200x800")

        self.config = TranslationConfig.load()

        self.create_variables()
        self.create_widgets()
        self.place_widgets()

    def create_variables(self):
        self.codigo_tipo_PLC_var = tk.StringVar(value=self.config.codigo_tipo_PLC)
        self.codigo_columna_maestra_var = tk.StringVar(value=self.config.codigo_columna_maestra)
        self.codigo_idioma_var = tk.StringVar(value=self.config.codigo_idioma_seleccionado)
        self.codigo_idioma_var2 = tk.StringVar(value=self.config.codigo_idioma_secundario)
        self.work_dir_var = tk.StringVar(value=self.config.work_dir)

        self.master_name_var = tk.StringVar(value=self.config.master_name)
        self.translate_name_var = tk.StringVar(value=self.config.translate_name)
        self.auto_translate_name_var = tk.StringVar(value=self.config.auto_translate_name)

    def create_widgets(self):
        self.main_frame = ttk.Frame(self.master)
        self.main_frame.grid(row=0, column=0, sticky='nsew')
        self.master.grid_rowconfigure(0, weight=1)
        self.master.grid_columnconfigure(0, weight=1)

        self.frame_combos = ttk.Frame(self.main_frame)
        
        self.label_tipo_PLC = ttk.Label(self.frame_combos, text="Selecciona el tipo de PLC:")
        self.combo_tipo_PLC = ttk.Combobox(self.frame_combos, state="readonly")
        self.combo_tipo_PLC["values"] = [nombre for nombre, _ in fc.PLCs.values()]
        self.combo_tipo_PLC.bind('<<ComboboxSelected>>', self.update_tipo_PLC)
        self.set_combo_value(self.combo_tipo_PLC, self.codigo_tipo_PLC_var.get())

        self.label_codigo_maestra = ttk.Label(self.frame_combos, text="Idioma Columna Maestra:")
        self.combo_codigo_maestra = ttk.Combobox(self.frame_combos, state="readonly")
        self.combo_codigo_maestra["values"] = [nombre for nombre, _ in fc.IDIOMAS.values()]
        self.combo_codigo_maestra.bind('<<ComboboxSelected>>', self.update_codigo_maestra)
        self.set_combo_value(self.combo_codigo_maestra, self.codigo_columna_maestra_var.get())

        self.label1 = ttk.Label(self.frame_combos, text="Idioma de Traducción:")
        self.combo = ttk.Combobox(self.frame_combos, state="readonly")
        self.combo["values"] = [nombre for nombre, _ in fc.IDIOMAS.values()]
        self.combo.bind('<<ComboboxSelected>>', self.update_idioma)
        self.set_combo_value(self.combo, self.codigo_idioma_var.get())

        self.label2 = ttk.Label(self.frame_combos, text="Selecciona segundo idioma:")
        self.combo2 = ttk.Combobox(self.frame_combos, state="readonly")
        self.combo2["values"] = [nombre for nombre, _ in fc.IDIOMAS.values()]
        self.combo2.bind('<<ComboboxSelected>>', self.update_idioma2)
        self.set_combo_value(self.combo2, self.codigo_idioma_var2.get())

        self.work_dir_frame = ttk.Frame(self.main_frame)
        self.work_dir_label = ttk.Label(self.work_dir_frame, text="Directorio de trabajo:")
        self.work_dir_entry = ttk.Entry(self.work_dir_frame, textvariable=self.work_dir_var, width=50)
        self.work_dir_button = ttk.Button(self.work_dir_frame, text="Seleccionar", command=self.select_work_dir)
        self.open_explorer_button = ttk.Button(self.work_dir_frame, text="Abrir Explorador", command=self.open_explorer)

        style = ttk.Style()
        style.configure('TButton', padding=(10, 10), font=('Helvetica', 10, 'bold'))
        style.configure('Excel.TButton', padding=(10, 10), font=('Helvetica', 10), background='#69cf81')

        self.button_width = 60
        self.entry_width = 50
        
        self.paso1 = ttk.Button(self.main_frame, text="1 - Importar al Master", command=lambda: self.ejecutar_run(x1_importar_to_master.run), width=self.button_width)
        self.master_name_button = ttk.Button(self.main_frame, textvariable=self.master_name_var, command=lambda: self.open_file(self.config.get_master_path()), width=self.entry_width, style='Excel.TButton')

        self.paso2 = ttk.Button(self.main_frame, text="2 - Exportar Idioma a 2_master_export2translate.xlsx", command=lambda: self.ejecutar_run(x2_master_export2translate.run), width=self.button_width)
        self.translate_name_button = ttk.Button(self.main_frame, textvariable=self.translate_name_var, command=lambda: self.open_file(self.config.get_translate_path()), width=self.entry_width, style='Excel.TButton')

        self.paso3 = ttk.Button(self.main_frame, text="3 - Traducir y generar 3_master_export2translate_translated.xlsx", command=lambda: self.ejecutar_run(x3_llm_auto_translate.run), width=self.button_width)
        self.auto_translate_name_button = ttk.Button(self.main_frame, textvariable=self.auto_translate_name_var, command=lambda: self.open_file(self.config.get_auto_translate_path()), width=self.entry_width, style='Excel.TButton')

        self.paso4 = ttk.Button(self.main_frame, text="4 - Integrar las traducciones al 1_hmi_master_translates", command=lambda: self.ejecutar_run(x4_integrate_translates_to_master.run), width=self.button_width)

        self.paso4B = ttk.Button(self.main_frame, text="4B - Integrar traducciones manuales al 1_hmi_master_translates", command=lambda: self.ejecutar_run(x4B_integrate_manual_translates_to_master.run), width=self.button_width)

        self.paso5 = ttk.Button(self.main_frame, text="5 - Completar en 1_hmi_master_translates el idioma seleccionado usando el segundo idioma", command=lambda: self.ejecutar_run(x5_complete_empty_cells_master.run), width=self.button_width)

        self.paso6 = ttk.Button(self.main_frame, text="6 - Exportar usando un archivo exportado desde TIA Portal usando 1_hmi_master_translates", command=lambda: self.ejecutar_run(x6_update_from_master.run), width=self.button_width)

        self.output_text = tk.Text(self.main_frame, wrap='none', height=20)
        self.scrollbar_y = ttk.Scrollbar(self.main_frame, orient='vertical', command=self.output_text.yview)
        self.scrollbar_x = ttk.Scrollbar(self.main_frame, orient='horizontal', command=self.output_text.xview)
        self.output_text.configure(yscrollcommand=self.scrollbar_y.set, xscrollcommand=self.scrollbar_x.set)

        self.clear_button = ttk.Button(self.main_frame, text="Limpiar salida", command=self.clear_output)
        self.open_log_button = ttk.Button(self.main_frame, text="Abrir Log", command=self.open_log_file)

    def place_widgets(self):
        self.main_frame.grid_columnconfigure(0, weight=1)
        self.main_frame.grid_columnconfigure(1, weight=1)

        self.frame_combos.grid(row=0, column=0, columnspan=2, pady=10, sticky='ew')
        self.label_tipo_PLC.grid(row=0, column=0, padx=5, pady=5, sticky='e')
        self.combo_tipo_PLC.grid(row=0, column=1, padx=5, pady=5, sticky='w')
        self.label_codigo_maestra.grid(row=1, column=0, padx=5, pady=5, sticky='e')
        self.combo_codigo_maestra.grid(row=1, column=1, padx=5, pady=5, sticky='w')
        self.label1.grid(row=0, column=2, padx=5, pady=5, sticky='e')
        self.combo.grid(row=0, column=3, padx=5, pady=5, sticky='w')
        self.label2.grid(row=1, column=2, padx=5, pady=5, sticky='e')
        self.combo2.grid(row=1, column=3, padx=5, pady=5, sticky='w')

        self.work_dir_frame.grid(row=1, column=0, columnspan=2, pady=10, sticky='ew')
        self.work_dir_label.grid(row=0, column=0, padx=5, sticky='w')
        self.work_dir_entry.grid(row=0, column=1, padx=5, sticky='ew')
        self.work_dir_button.grid(row=0, column=2, padx=5, sticky='e')
        self.open_explorer_button.grid(row=0, column=3, padx=5, sticky='e')
        self.work_dir_frame.grid_columnconfigure(1, weight=1)

        current_row = 2
        self.paso1.grid(row=current_row, column=0, pady=5, padx=5, sticky='ew')
        self.master_name_button.grid(row=current_row, column=1, pady=5, padx=5, sticky='ew')
        
        current_row += 1
        self.paso2.grid(row=current_row, column=0, pady=5, padx=5, sticky='ew')
        self.translate_name_button.grid(row=current_row, column=1, pady=5, padx=5, sticky='ew')
        
        current_row += 1
        self.paso3.grid(row=current_row, column=0, pady=5, padx=5, sticky='ew')
        self.auto_translate_name_button.grid(row=current_row, column=1, pady=5, padx=5, sticky='ew')
        
        current_row += 1
        self.paso4.grid(row=current_row, column=0, pady=5, padx=5, sticky='ew')
        
        current_row += 1
        self.paso4B.grid(row=current_row, column=0, pady=5, padx=5, sticky='ew')
        
        current_row += 1
        self.paso5.grid(row=current_row, column=0, columnspan=2, pady=5, padx=5, sticky='ew')
        
        current_row += 1
        self.paso6.grid(row=current_row, column=0, columnspan=2, pady=5, padx=5, sticky='ew')

        current_row += 1
        self.output_text.grid(row=current_row, column=0, columnspan=2, padx=(10, 0), pady=10, sticky='nsew')
        self.scrollbar_y.grid(row=current_row, column=2, pady=10, sticky='ns')
        self.scrollbar_x.grid(row=current_row+1, column=0, columnspan=2, padx=10, sticky='ew')

        current_row += 2
        self.clear_button.grid(row=current_row, column=0, pady=10, padx=5, sticky='e')
        self.open_log_button.grid(row=current_row, column=1, pady=10, padx=5, sticky='w')

        self.main_frame.grid_rowconfigure(current_row-2, weight=1)

    def set_combo_value(self, combo, codigo):
        for nombre, code in fc.IDIOMAS.values():
            if code == codigo:
                combo.set(nombre)
                return
        for nombre, code in fc.PLCs.values():
            if code == codigo:
                combo.set(nombre)
                return

    def update_tipo_PLC(self, event):
        nombre_seleccionado = self.combo_tipo_PLC.get()
        for nombre, codigo in fc.PLCs.values():
            if nombre == nombre_seleccionado:
                self.codigo_tipo_PLC_var.set(codigo)
                break
        self.update_file_names()

    def update_codigo_maestra(self, event):
        nombre_seleccionado = self.combo_codigo_maestra.get()
        for nombre, codigo in fc.IDIOMAS.values():
            if nombre == nombre_seleccionado:
                self.codigo_columna_maestra_var.set(codigo)
                break
        self.update_file_names()

    def update_idioma(self, event):
        nombre_seleccionado = self.combo.get()
        for nombre, codigo in fc.IDIOMAS.values():
            if nombre == nombre_seleccionado:
                self.codigo_idioma_var.set(codigo)
                break
        self.update_file_names()

    def update_idioma2(self, event):
        nombre_seleccionado = self.combo2.get()
        for nombre, codigo in fc.IDIOMAS.values():
            if nombre == nombre_seleccionado:
                self.codigo_idioma_var2.set(codigo)
                break
        self.update_file_names()

    def save_config(self):
        self.config.update(
            codigo_tipo_PLC=self.codigo_tipo_PLC_var.get(),
            codigo_columna_maestra=self.codigo_columna_maestra_var.get(),
            codigo_idioma_seleccionado=self.codigo_idioma_var.get(),
            codigo_idioma_secundario=self.codigo_idioma_var2.get(),
            work_dir=self.work_dir_var.get(),
            master_name=self.master_name_var.get(),
            translate_name=self.translate_name_var.get(),
            auto_translate_name=self.auto_translate_name_var.get(),
        )
        self.config.save()

    def select_work_dir(self):
        dir_path = filedialog.askdirectory()
        if dir_path:
            self.work_dir_var.set(dir_path)
            self.save_config()
            self.update_file_names()

    def open_file(self, file_path):
        if os.path.exists(file_path):
            os.startfile(file_path)
        else:
            messagebox.showerror("Error", f"El archivo {file_path} no existe.")

    def open_explorer(self):
        if os.path.exists(self.config.work_dir):
            os.startfile(self.config.work_dir)
        else:
            messagebox.showerror("Error", f"El directorio {self.config.work_dir} no existe.")

    def open_log_file(self):
        log_file_path = os.path.join(self.config.work_dir, "logs", "translate_log.log")
        if os.path.exists(log_file_path):
            os.startfile(log_file_path)
        else:
            messagebox.showerror("Error", f"El archivo de log {log_file_path} no existe.")

    def update_file_names(self, event=None):
        codigo_tipo_PLC = self.codigo_tipo_PLC_var.get()
        codigo_idioma_seleccionado = self.codigo_idioma_var.get()
        
        self.master_name_var.set(f"1_hmi_master_translates_{codigo_tipo_PLC}.xlsx")
        self.translate_name_var.set(f'2_master_export2translate_{codigo_tipo_PLC}_{codigo_idioma_seleccionado}.xlsx')
        self.auto_translate_name_var.set(f"3_master_export2translate_translated_{codigo_tipo_PLC}_{codigo_idioma_seleccionado}.xlsx")
        
        self.save_config()

    def ejecutar_run(self, funcion_run):
        self.save_config()
        
        original_stdout = sys.stdout
        sys.stdout = self.RedirectText(self.output_text)

        try:
            if funcion_run == x1_importar_to_master.run:
                funcion_run(self.config)
            elif funcion_run == x2_master_export2translate.run:
                funcion_run(self.config)
            elif funcion_run == x3_llm_auto_translate.run:
                traducir_todo = messagebox.askyesno("Traducir todo", "¿Desea traducir todas las celdas?")
                self.config.traducir_todo = traducir_todo
                funcion_run(self.config)
            elif funcion_run == x4_integrate_translates_to_master.run:
                funcion_run(self.config)
            elif funcion_run == x4B_integrate_manual_translates_to_master.run:
                funcion_run(self.config)
            elif funcion_run == x5_complete_empty_cells_master.run:
                funcion_run(self.config)
            elif funcion_run == x6_update_from_master.run:
                archivo_to_update = filedialog.askopenfilename(title="Seleccione el archivo a actualizar", filetypes=[("Excel files", "*.xlsx")])
                if archivo_to_update:
                    funcion_run(self.config, archivo_to_update)
        except Exception as e:
            messagebox.showerror("Error", str(e))
        finally:
            sys.stdout = original_stdout
            self.save_config()
    def clear_output(self):
        self.output_text.delete('1.0', tk.END)

    class RedirectText:
        def __init__(self, text_widget):
            self.text_widget = text_widget

        def write(self, string):
            self.text_widget.insert(tk.END, string)
            self.text_widget.see(tk.END)

        def flush(self):
            pass

def main():
    root = tk.Tk()
    app = TranslationApp(root)
    root.mainloop()

if __name__ == "__main__":
    main()