Update app.py

app.py

@@ -104,7 +104,8 @@ def process_input(text, rag, prompt_template):
     """
     output = generate(prompt)
     
-    return output
+    return output,prompt
+
 
 def create_interface():
     # Definimos los componentes de la interfaz
@@ -126,17 +127,18 @@ def create_interface():
 
     # Definimos la función que se ejecutará cuando se envíe la entrada
     def fn(text, rag, prompt_template):
-        
+
+        output = []
         output = process_input(text, rag, prompt_template)
-        matches = re.findall(r'"tipo"\s*:\s*"(.*?)"', output)
+        matches = re.findall(r'"tipo"\s*:\s*"(.*?)"', output[0])
         classification_types = matches
         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)
 
         # Resalta las partes del texto que se encuentran en la variable output en rojo
-        highlighted_text = highlight_text(text, output, color="red")
+        highlighted_text = highlight_text(text, output[0], color="red")
 
-        return highlighted_text, classification_types_checkboxes, output
+        return highlighted_text, classification_types_checkboxes, output[0], output[1]
         
     examples = [
         [recorte, False, True]
@@ -149,7 +151,7 @@ def create_interface():
     iface = gr.Interface(
         fn=fn,
         inputs=[input_text, rag_checkbox, prompt_template],
-        outputs=["html", classification_types_checkboxes, output_text],
+        outputs=["html", classification_types_checkboxes, "text", "text"],
         examples=examples,
         allow_flagging="manual",
         flagging_options=["👍", "👎"],