Update app.py

app.py

@@ -99,9 +99,9 @@ def process_input(text, rag, prompt_template):
     --------------------------------------------------------------
 
     Identifica y clasifica los argumentos expuestos en el anterior fragmento de texto siguiendo estos pasos:
-     - Primero: Determina cúantos argumentos se exponen en el fragmento proporcionado.
-     - Segundo: Determina el contenido de cada uno de los argumentos encontrados.
-     - Tercero: Clasifica cada uno de los argumentos encontrados en el fragmento con el tipo de argumento que le corresponda según la 'Lista Argumentos''
+     - Primero: Determina cúantos argumentos se exponen en el fragmento proporcionado. No escribas nada sobre este apartado.
+     - Segundo: Determina el contenido de cada uno de los argumentos encontrados. No escribas nada sobre este apartado.
+     - Tercero: Clasifica cada uno de los argumentos encontrados en el fragmento con el tipo de argumento que le corresponda según la 'Lista Argumentos''. No escribas nada sobre este apartado.
      - Cuarto: Este es el paso más importante. Devuelve una lista de JSON, cuya longitud será igual al número de argumentos encontrados, que contenga los argumentos clasificados con el siguiente formato:
         {format_text}
 
@@ -132,8 +132,8 @@ def create_interface():
     def fn(text, rag, prompt_template):
         
         output = process_input(text, rag, prompt_template)
-        matches = re.findall(r"tipo: '([^']+)'", output)
-        classification_types = matches
+        data = json.loads(output)
+        classification_types = [item['tipo'] for item in data]
         classification_types_options = [(option, option) for option in classification_types]
         classification_types_checkboxes = gr.CheckboxGroup(label="Clasificacion Tipo", choices=classification_types_options, interactive = True)