Update app.py

app.py

@@ -40,17 +40,9 @@ prompt_template_text = """
 """
 
 recorte = """
-    SEGUNDA.- Que como consecuencia de lo manifestado con anterioridad, intereso la apertura del
-    periodo de prueba, conforme al el art. 13 del Real Decreto 320/1994 en relación con los arts. 77 y 78
-    de la Ley 39/2015 y propongo la práctica de los medios de prueba, conforme a lo dispuesto en el 13.1
-    del RD. 320/1994, que esta parte considera imprescindibles para poder ejercer su derecho a la
-    defensa, con expresa petición de la notificación de su resultado y sin perjuicio de que una vez recibidas
-    pueda realizar cuantas alegaciones estime pertinentes, que se deja interesada como sigue:
-
-    A- Documental, para que se incorpore al expediente INFORME DEL AGENTE DENUNCIANTE
-    relativo a las circunstancias de la presunta infracción, con expresión en concreto de su consideración
-    acerca de la relación entre la conducta del presunto infractor y su determinación como infracción
-    tipificada
+CUARTA .- Que conforme al art.53.2 a) de la Ley 39/2015 de 1 de Octubre del Procedimiento
+Administrativo Común de las Administraciones Públicas, SOLICITO la comunicación de la identidad
+de la autoridad competente para imponerla sanción y de la norma que le atribuya tal competencia.
 """
 
 def format_prompt(message):
@@ -119,6 +111,16 @@ def create_interface():
     output_text = gr.Textbox(label="Output")
     classification_types_checkboxes = gr.CheckboxGroup(label="Clasificacion Tipo")
 
+    def highlight_text(text, output, color="red"):
+    # Encuentra las partes del texto que se encuentran en la variable output
+    matches = re.findall(r"'texto_referencia'\s*:\s*\"(.*?)\"", output)
+    # Resalta las partes del texto que se encuentran en la variable output en rojo
+    for match in matches:
+        if match in text:
+            text = text.replace(match, f'<span style="color: {color};">{match}</span>')
+    return text
+
+
     # Definimos la función que se ejecutará cuando se envíe la entrada
     def fn(text, rag, prompt_template):
         
@@ -128,7 +130,11 @@ def create_interface():
         classification_types_options = [(option, option) for option in classification_types]
         classification_types_checkboxes = gr.CheckboxGroup(label="Clasificacion Tipo", choices=classification_types_options,value=[option[0] for option in classification_types_options], interactive = True)
 
-        return output, classification_types_checkboxes
+        # Resalta las partes del texto que se encuentran en la variable output en rojo
+        highlighted_text = highlight_text(text, output, color="red")
+
+
+        return highlighted_text, classification_types_checkboxes
         
     examples = [
         [recorte, False, True]