Update app.py

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
 import json
-import re
 
 # Función para cargar los datos del archivo JSON
 def cargar_datos_json(ruta_json):
@@ -22,18 +21,14 @@ def procesar_entrada(texto_entrada, rag_habilitado, prompt_template_habilitado):
     
     # Simulación de generación de respuesta
     respuesta = f"Texto de salida procesado: {texto_entrada}"
-    
-    # Cargar datos del archivo JSON
-    ruta_json = './archivo.json'
-    datos_json = cargar_datos_json(ruta_json)
-    
-    # Resaltar partes coincidentes
-    for dato in datos_json:
-        contenido = dato['contenido']
-        respuesta = re.sub(f"({contenido})", r"<mark>\1</mark>", respuesta)
-    
     return [respuesta] + [False] * len(mostrar_checkboxes(datos_json))
 
+# Función para guardar los checkbox marcados en un archivo JSON
+def guardar_checkbox_info(checkboxes, ruta_json):
+    info_checkbox = {key: value for key, value in checkboxes.items()}
+    with open(ruta_json, 'w') as file:
+        json.dump(info_checkbox, file, indent=4)
+
 # Ruta al archivo JSON
 ruta_json = './archivo.json'
 datos_json = cargar_datos_json(ruta_json)
@@ -48,10 +43,21 @@ interfaz = gr.Interface(
         gr.Checkbox(label="Habilitar Prompt Template")  
     ],
     outputs=[
-        gr.Textbox(label="Texto de Salida", type="text"),
-        *checkboxes_tipos.values()
+        gr.Textbox(label="Texto de Salida", type="auto"),
+        *checkboxes_tipos.values(),
+        gr.Button(label="Guardar en JSON", type="button", live=True)
     ]
 )
 
+# Creación de diccionario de checkboxes
+checkboxes = {key: False for key in checkboxes_tipos.keys()}
+
+# Definición de la acción del botón
+def accion_boton():
+    guardar_checkbox_info(checkboxes, 'rlhf.json')
+
+# Asignación de la acción al botón
+interfaz[3].onclick(accion_boton)
+
 # Ejecución de la interfaz
 interfaz.launch()