Actualizado menu con opciones persistentes

This commit is contained in:
Miguel 2024-09-20 09:24:08 +02:00
parent 7fe8de819d
commit c037c001fc
10 changed files with 1762 additions and 107 deletions

10
config.ini Normal file
View File

@ -0,0 +1,10 @@
[Seleccion]
tipo_plc = AllenBradley
codigo_columna_maestra = English_US
idioma_var = Spanish
idioma_var2 = English_US
[TextContent]
output_text =

View File

@ -17,6 +17,13 @@ IDIOMAS = {
7: ("English_US", "en-US"),
}
# Función para obtener el shortcode a partir del código completo
def idiomas_shortcodefromcode(code):
if code:
return code.split('-')[0]
return None # O puedes lanzar una excepción si prefieres
# Función para obtener el idioma a partir del código
def idiomas_idiomafromcode(code):
for idx, (idioma, codigo) in IDIOMAS.items():

File diff suppressed because it is too large Load Diff

View File

@ -8,11 +8,15 @@ import x3_llm_auto_translate
import x4_integrate_translates_to_master
import x5_complete_empty_cells_master
import x6_update_from_master
import sys
import threading
import configparser # Para manejar el archivo de configuración
import os # Para verificar si el archivo existe
# Crear la ventana principal
ventana = tk.Tk()
ventana.title("Ayuda para traducir textos de TIA Portal y Allen Bradley")
ventana.geometry("600x550") # Aumentamos el tamaño para acomodar los nuevos widgets
ventana.geometry("800x600") # Ajustamos el tamaño de la ventana
# Diccionario para tipo_PLC
tipo_PLC_dict = {'Siemens': 'siemens', 'AllenBradley': 'allenbradley'}
@ -26,34 +30,84 @@ idioma_var2 = tk.StringVar()
# Crear listas de idiomas con sus códigos
idiomas_lista = [(nombre, codigo) for _, (nombre, codigo) in fc.IDIOMAS.items()]
# Crear un combobox para seleccionar el tipo de PLC
label_tipo_PLC = tk.Label(ventana, text="Selecciona el tipo de PLC:")
combo_tipo_PLC = ttk.Combobox(ventana, textvariable=tipo_PLC_var, state="readonly")
# Ruta del archivo de configuración
config_file = 'config.ini'
# Función para cargar las opciones guardadas
def cargar_configuracion():
config = configparser.ConfigParser()
if os.path.exists(config_file):
config.read(config_file)
# Cargar tipo_PLC
tipo_PLC_var.set(config.get('Seleccion', 'tipo_PLC', fallback='Siemens'))
# Cargar codigo_columna_maestra
codigo_columna_maestra_var.set(config.get('Seleccion', 'codigo_columna_maestra', fallback=idiomas_lista[0][0]))
# Cargar idioma_var
idioma_var.set(config.get('Seleccion', 'idioma_var', fallback=idiomas_lista[4][0]))
# Cargar idioma_var2
idioma_var2.set(config.get('Seleccion', 'idioma_var2', fallback=idiomas_lista[1][0]))
else:
# Valores por defecto si no existe el archivo
tipo_PLC_var.set('Siemens')
codigo_columna_maestra_var.set(idiomas_lista[0][0])
idioma_var.set(idiomas_lista[4][0])
idioma_var2.set(idiomas_lista[1][0])
# Función para guardar las opciones seleccionadas
def guardar_configuracion():
config = configparser.ConfigParser()
config['Seleccion'] = {
'tipo_PLC': tipo_PLC_var.get(),
'codigo_columna_maestra': codigo_columna_maestra_var.get(),
'idioma_var': idioma_var.get(),
'idioma_var2': idioma_var2.get()
}
# Guardar el contenido del cuadro de texto
config['TextContent'] = {
'output_text': output_text.get("1.0", tk.END)
}
with open(config_file, 'w') as configfile:
config.write(configfile)
# Crear un Frame para la grilla de comboboxes
frame_combos = tk.Frame(ventana)
frame_combos.pack(pady=10)
# Crear los labels y comboboxes
label_tipo_PLC = tk.Label(frame_combos, text="Selecciona el tipo de PLC:")
combo_tipo_PLC = ttk.Combobox(frame_combos, textvariable=tipo_PLC_var, state="readonly")
combo_tipo_PLC["values"] = list(tipo_PLC_dict.keys())
combo_tipo_PLC.current(0) # Selecciona 'Siemens' por defecto
# Crear combobox para seleccionar el código de columna maestra
label_codigo_maestra = tk.Label(ventana, text="Idioma Columna Maestra:")
combo_codigo_maestra = ttk.Combobox(ventana, textvariable=codigo_columna_maestra_var, state="readonly")
label_codigo_maestra = tk.Label(frame_combos, text="Idioma Columna Maestra:")
combo_codigo_maestra = ttk.Combobox(frame_combos, textvariable=codigo_columna_maestra_var, state="readonly")
combo_codigo_maestra["values"] = [nombre for nombre, _ in idiomas_lista]
combo_codigo_maestra.current(0) # Selecciona el primer idioma por defecto
# Crear comboboxes para seleccionar los idiomas
label1 = tk.Label(ventana, text="Idioma de Traducción:")
combo = ttk.Combobox(ventana, textvariable=idioma_var, state="readonly")
label1 = tk.Label(frame_combos, text="Idioma de Traducción:")
combo = ttk.Combobox(frame_combos, textvariable=idioma_var, state="readonly")
combo["values"] = [nombre for nombre, _ in idiomas_lista]
combo.current(4) # Selecciona el quinto idioma por defecto
label2 = tk.Label(ventana, text="Selecciona segundo idioma:")
combo2 = ttk.Combobox(ventana, textvariable=idioma_var2, state="readonly")
label2 = tk.Label(frame_combos, text="Selecciona segundo idioma:")
combo2 = ttk.Combobox(frame_combos, textvariable=idioma_var2, state="readonly")
combo2["values"] = [nombre for nombre, _ in idiomas_lista]
combo2.current(1) # Selecciona el segundo idioma por defecto
# Organizar los labels y comboboxes en una grilla 2x2
label_tipo_PLC.grid(row=0, column=0, padx=5, pady=5, sticky='e')
combo_tipo_PLC.grid(row=0, column=1, padx=5, pady=5, sticky='w')
label_codigo_maestra.grid(row=1, column=0, padx=5, pady=5, sticky='e')
combo_codigo_maestra.grid(row=1, column=1, padx=5, pady=5, sticky='w')
label1.grid(row=0, column=2, padx=5, pady=5, sticky='e')
combo.grid(row=0, column=3, padx=5, pady=5, sticky='w')
label2.grid(row=1, column=2, padx=5, pady=5, sticky='e')
combo2.grid(row=1, column=3, padx=5, pady=5, sticky='w')
# Función para actualizar combo_codigo_maestra según la selección en combo_tipo_PLC
def actualizar_codigo_maestra(event):
plc_seleccionado = tipo_PLC_var.get()
if plc_seleccionado == 'Siemens':
# Buscar el índice de 'Italiano' en idiomas_lista
# Buscar el índice de 'Italian' en idiomas_lista
indice_italiano = next((i for i, (nombre, _) in enumerate(idiomas_lista) if nombre == 'Italian'), None)
if indice_italiano is not None:
combo_codigo_maestra.current(indice_italiano)
@ -62,102 +116,88 @@ def actualizar_codigo_maestra(event):
indice_ingles = next((i for i, (nombre, _) in enumerate(idiomas_lista) if nombre == 'English_US'), None)
if indice_ingles is not None:
combo_codigo_maestra.current(indice_ingles)
guardar_configuracion() # Guardar la configuración cuando se cambia el tipo de PLC
# Vincular la función al evento de cambio de selección en combo_tipo_PLC
combo_tipo_PLC.bind('<<ComboboxSelected>>', actualizar_codigo_maestra)
# Ubicar los widgets en la ventana
label_tipo_PLC.pack(pady=5)
combo_tipo_PLC.pack(pady=5)
label_codigo_maestra.pack(pady=5)
combo_codigo_maestra.pack(pady=5)
label1.pack(pady=5)
combo.pack(pady=5)
label2.pack(pady=5)
combo2.pack(pady=5)
# Vincular eventos de cambio en los comboboxes para guardar la configuración
def on_combobox_changed(event):
guardar_configuracion()
# Función para obtener el código del idioma seleccionado a partir del índice
def obtener_codigo_idioma(indice):
return idiomas_lista[indice][1]
combo_codigo_maestra.bind('<<ComboboxSelected>>', on_combobox_changed)
combo.bind('<<ComboboxSelected>>', on_combobox_changed)
combo2.bind('<<ComboboxSelected>>', on_combobox_changed)
# Funciones que se llamarán cuando se presionen los botones
def accion_boton1():
# Clase para redirigir la salida estándar a la caja de texto
class RedirectText(object):
def __init__(self, text_widget):
self.output = text_widget
def write(self, string):
self.output.insert(tk.END, string)
self.output.see(tk.END) # Desplaza hacia el final
def flush(self):
pass # No se requiere implementación para flush
# Función para obtener el código del idioma seleccionado a partir del nombre
def obtener_codigo_idioma_por_nombre(nombre_idioma):
for nombre, codigo in idiomas_lista:
if nombre == nombre_idioma:
return codigo
return None
# Función para ejecutar las funciones run() y redirigir la salida
def ejecutar_run(funcion_run):
# Obtener los parámetros necesarios
tipo_PLC = tipo_PLC_dict[tipo_PLC_var.get()]
indice_codigo_maestra = combo_codigo_maestra.current()
codigo_columna_maestra = obtener_codigo_idioma(indice_codigo_maestra)
print("Ejecutando x1_importar_to_master.py")
x1_importar_to_master.run(tipo_PLC, codigo_columna_maestra)
codigo_columna_maestra = obtener_codigo_idioma_por_nombre(codigo_columna_maestra_var.get())
codigo_idioma = obtener_codigo_idioma_por_nombre(idioma_var.get())
codigo_idioma2 = obtener_codigo_idioma_por_nombre(idioma_var2.get())
def accion_boton2():
tipo_PLC = tipo_PLC_dict[tipo_PLC_var.get()]
indice_codigo_maestra = combo_codigo_maestra.current()
codigo_columna_maestra = obtener_codigo_idioma(indice_codigo_maestra)
indice_seleccionado = combo.current()
x2_master_export2translate.run(tipo_PLC, codigo_columna_maestra, indice_seleccionado)
# Obtener los índices de los idiomas
indice_seleccionado = idiomas_lista.index((idioma_var.get(), codigo_idioma))
indice_seleccionado2 = idiomas_lista.index((idioma_var2.get(), codigo_idioma2))
def accion_boton3():
tipo_PLC = tipo_PLC_dict[tipo_PLC_var.get()]
indice_codigo_maestra = combo_codigo_maestra.current()
codigo_columna_maestra = obtener_codigo_idioma(indice_codigo_maestra)
indice_seleccionado = combo.current()
# Redirigir stdout
original_stdout = sys.stdout
sys.stdout = RedirectText(output_text)
try:
# Ejecutar la función correspondiente
if funcion_run == x1_importar_to_master.run:
print("Ejecutando x1_importar_to_master.run()")
funcion_run(tipo_PLC, codigo_columna_maestra)
elif funcion_run == x2_master_export2translate.run:
print("Ejecutando x2_master_export2translate.run()")
funcion_run(tipo_PLC, codigo_columna_maestra, indice_seleccionado)
elif funcion_run == x3_llm_auto_translate.run:
traducir_todo = messagebox.askyesno("Traducir todo", "¿Desea traducir todas las celdas?")
x3_llm_auto_translate.run(tipo_PLC, codigo_columna_maestra, indice_seleccionado, traducir_todo)
def accion_boton4():
tipo_PLC = tipo_PLC_dict[tipo_PLC_var.get()]
indice_codigo_maestra = combo_codigo_maestra.current()
codigo_columna_maestra = obtener_codigo_idioma(indice_codigo_maestra)
indice_seleccionado = combo.current()
x4_integrate_translates_to_master.run(tipo_PLC, codigo_columna_maestra, indice_seleccionado, 0.5)
def accion_boton5():
tipo_PLC = tipo_PLC_dict[tipo_PLC_var.get()]
indice_codigo_maestra = combo_codigo_maestra.current()
codigo_columna_maestra = obtener_codigo_idioma(indice_codigo_maestra)
indice_seleccionado = combo.current()
indice_seleccionado2 = combo2.current()
x5_complete_empty_cells_master.run(tipo_PLC, codigo_columna_maestra, indice_seleccionado, indice_seleccionado2)
def accion_boton6():
tipo_PLC = tipo_PLC_dict[tipo_PLC_var.get()]
indice_codigo_maestra = combo_codigo_maestra.current()
codigo_columna_maestra = obtener_codigo_idioma(indice_codigo_maestra)
indice_seleccionado = combo.current()
x6_update_from_master.run(tipo_PLC, codigo_columna_maestra, indice_seleccionado)
print("Ejecutando x3_llm_auto_translate.run()")
funcion_run(tipo_PLC, codigo_columna_maestra, indice_seleccionado, traducir_todo)
elif funcion_run == x4_integrate_translates_to_master.run:
print("Ejecutando x4_integrate_translates_to_master.run()")
funcion_run(tipo_PLC, codigo_columna_maestra, indice_seleccionado, 0.5)
elif funcion_run == x5_complete_empty_cells_master.run:
print("Ejecutando x5_complete_empty_cells_master.run()")
funcion_run(tipo_PLC, codigo_columna_maestra, indice_seleccionado, indice_seleccionado2)
elif funcion_run == x6_update_from_master.run:
print("Ejecutando x6_update_from_master.run()")
funcion_run(tipo_PLC, codigo_columna_maestra, indice_seleccionado)
finally:
# Restaurar stdout
sys.stdout = original_stdout
guardar_configuracion() # Guardar el contenido del cuadro de texto
# Crear los botones con el mismo ancho
button_width = 70
paso1 = tk.Button(ventana, text="1 - Importar al Master", command=accion_boton1, width=button_width)
paso2 = tk.Button(
ventana,
text="2 - Exportar Idioma a 2_master_export2translate.xlsx.",
command=accion_boton2,
width=button_width
)
paso3 = tk.Button(
ventana,
text="3 - Traducir y generar 3_master_export2translate_translated.xlsx.",
command=accion_boton3,
width=button_width
)
paso4 = tk.Button(
ventana,
text="4 - Integrar las traducciones al 1_hmi_master_translates.",
command=accion_boton4,
width=button_width
)
paso5 = tk.Button(
ventana,
text="5 - Completar en 1_hmi_master_translates el idioma seleccionado usando el segundo idioma.",
command=accion_boton5,
width=button_width
)
paso6 = tk.Button(
ventana,
text="6 - Exportar usando un archivo exportado desde TIA Portal usando 1_hmi_master_translates.",
command=accion_boton6,
width=button_width
)
paso1 = tk.Button(ventana, text="1 - Importar al Master", command=lambda: threading.Thread(target=ejecutar_run, args=(x1_importar_to_master.run,)).start(), width=button_width)
paso2 = tk.Button(ventana, text="2 - Exportar Idioma a 2_master_export2translate.xlsx.", command=lambda: threading.Thread(target=ejecutar_run, args=(x2_master_export2translate.run,)).start(), width=button_width)
paso3 = tk.Button(ventana, text="3 - Traducir y generar 3_master_export2translate_translated.xlsx.", command=lambda: threading.Thread(target=ejecutar_run, args=(x3_llm_auto_translate.run,)).start(), width=button_width)
paso4 = tk.Button(ventana, text="4 - Integrar las traducciones al 1_hmi_master_translates.", command=lambda: threading.Thread(target=ejecutar_run, args=(x4_integrate_translates_to_master.run,)).start(), width=button_width)
paso5 = tk.Button(ventana, text="5 - Completar en 1_hmi_master_translates el idioma seleccionado usando el segundo idioma.", command=lambda: threading.Thread(target=ejecutar_run, args=(x5_complete_empty_cells_master.run,)).start(), width=button_width)
paso6 = tk.Button(ventana, text="6 - Exportar usando un archivo exportado desde TIA Portal usando 1_hmi_master_translates.", command=lambda: threading.Thread(target=ejecutar_run, args=(x6_update_from_master.run,)).start(), width=button_width)
# Ubicar los botones en la ventana
paso1.pack(pady=(10,2))
@ -167,5 +207,63 @@ paso4.pack(pady=2)
paso5.pack(pady=2)
paso6.pack(pady=2)
# Crear un Frame para el cuadro de texto, los scrollbars y el botón
frame_texto = tk.Frame(ventana)
frame_texto.pack(fill='both', expand=True, pady=10)
# Crear los scrollbars
scrollbar_vertical = tk.Scrollbar(frame_texto, orient='vertical')
scrollbar_horizontal = tk.Scrollbar(frame_texto, orient='horizontal')
# Crear la caja de texto para mostrar la salida
output_text = tk.Text(
frame_texto,
wrap='none',
yscrollcommand=scrollbar_vertical.set,
xscrollcommand=scrollbar_horizontal.set
)
# Configurar los scrollbars
scrollbar_vertical.config(command=output_text.yview)
scrollbar_horizontal.config(command=output_text.xview)
# Crear el botón para borrar el contenido
boton_limpiar = tk.Button(frame_texto, text="Borrar contenido", command=lambda: limpiar_texto())
# Ubicar los widgets en frame_texto usando grid
output_text.grid(row=0, column=0, sticky='nsew')
scrollbar_vertical.grid(row=0, column=1, sticky='ns')
scrollbar_horizontal.grid(row=1, column=0, sticky='ew')
boton_limpiar.grid(row=1, column=1, padx=5, pady=5)
# Configurar la expansión de filas y columnas
frame_texto.grid_rowconfigure(0, weight=1)
frame_texto.grid_columnconfigure(0, weight=1)
# Función para borrar el contenido del cuadro de texto
def limpiar_texto():
output_text.delete('1.0', tk.END)
guardar_configuracion()
# Cargar las opciones guardadas al iniciar
cargar_configuracion()
# Cargar el contenido del cuadro de texto desde la configuración
def cargar_texto():
config = configparser.ConfigParser()
if os.path.exists(config_file):
config.read(config_file)
texto = config.get('TextContent', 'output_text', fallback='')
output_text.insert('1.0', texto)
cargar_texto()
# Guardar la configuración al cerrar la ventana
def on_closing():
guardar_configuracion()
ventana.destroy()
ventana.protocol("WM_DELETE_WINDOW", on_closing)
# Iniciar el bucle principal de la interfaz
ventana.mainloop()

View File

@ -110,13 +110,13 @@ def translate_batch_openai(texts_dict, source_lang, target_lang):
return dict(zip(texts_dict.keys(), translations))
def affinity_batch_openai(texts_dict):
def affinity_batch_openai(tipo_PLC, texts_dict):
system_prompt = (
"Evaluate the semantic similarity between the following table of pairs of texts in json format on a scale from 0 to 1. "
"Return the similarity scores for every row in JSON format as a list of numbers, without any additional text or formatting."
)
original_list = [
fc.compactar_celda_traducida(key) for key in texts_dict.keys()
fc.compactar_celda_traducida(tipo_PLC, key) for key in texts_dict.keys()
]
re_translated_list = list(texts_dict.values())
@ -167,13 +167,13 @@ def affinity_batch_openai(texts_dict):
# Función que calcula la afinidad entre dos textos
def calcular_afinidad(texto1, texto2):
def calcular_afinidad(tipo_PLC, texto1, texto2):
system_prompt = (
"Evaluate the semantic similarity between the following pair of texts on a scale from 0 to 1. "
"Return the similarity score as a single number."
)
original_text = fc.compactar_celda_traducida(texto1)
original_text = fc.compactar_celda_traducida(tipo_PLC, texto1)
compared_text = texto2
request_payload = json.dumps({"original": original_text, "compared": compared_text})
@ -295,7 +295,7 @@ def main(tipo_PLC, codigo_columna_maestra, file_path, target_lang_code, target_l
df.at[index, target_col] = translations[celda_clave]
# Realizar la traducción de verificación con Google Translate
try:
google_translation = google_translate(translations[celda_clave], "it")
google_translation = google_translate(translations[celda_clave], fc.idiomas_shortcodefromcode(codigo_columna_maestra))
df.at[index, check_translate_col] = google_translation
except Exception as e:
logger.error(
@ -316,7 +316,7 @@ def main(tipo_PLC, codigo_columna_maestra, file_path, target_lang_code, target_l
retries = 2 # Número de intentos totales (1 inicial + 1 reintento)
for attempt in range(retries):
try:
batch_affinities = affinity_batch_openai(batch_texts)
batch_affinities = affinity_batch_openai(tipo_PLC, batch_texts)
affinities.update(batch_affinities)
break # Si la llamada es exitosa, salimos del bucle de reintentos
except Exception as e:
@ -338,7 +338,7 @@ def main(tipo_PLC, codigo_columna_maestra, file_path, target_lang_code, target_l
# Intentar individualmente si falla en batch
for key, value in batch_texts.items():
try:
score = calcular_afinidad(key, value)
score = calcular_afinidad(tipo_PLC, key, value)
affinities[key] = score
except Exception as ind_e:
affinities[key] = "0"