app.py con almacenamiento en memoria

app.py

@@ -1,29 +1,20 @@
 import gradio as gr
 import spacy
 from graphviz import Digraph
-import os
-import uuid
+import io
+from PIL import Image
 import pandas as pd
 
 # Cargar modelo de spaCy
 nlp = spacy.load("es_dep_news_trf")
 
-# Carpeta persistente para imágenes
-os.makedirs("outputs", exist_ok=True)
-
-def limpiar_outputs():
-    for archivo in os.listdir("outputs"):
-        ruta = os.path.join("outputs", archivo)
-        if os.path.isfile(ruta):
-            os.remove(ruta)
-
 def generar_grafico_dependencia(texto):
-    # Limpiar la carpeta de imágenes antes de generar nuevas
-    limpiar_outputs()
+    if not texto or texto.strip() == "":
+        return [], pd.DataFrame()
 
     doc = nlp(texto)
     raices = [token for token in doc if token.head == token]
-    rutas_imagenes = []
+    imagenes = []
 
     for i, raiz in enumerate(raices):
         dot = Digraph(comment=f"Árbol {i+1}")
@@ -42,10 +33,12 @@ def generar_grafico_dependencia(texto):
 
         agregar_nodo(raiz)
 
-        # Guardar archivo con nombre único
-        nombre_archivo = f"outputs/arbol_{uuid.uuid4().hex}.png"
-        dot.render(filename=nombre_archivo, format='png', cleanup=True)
-        rutas_imagenes.append(nombre_archivo + ".png")  # Graphviz añade extensión
+        # Generar imagen en memoria (bytes)
+        png_bytes = dot.pipe(format='png')
+        
+        # Convertir bytes a imagen PIL
+        imagen = Image.open(io.BytesIO(png_bytes))
+        imagenes.append(imagen)
 
     # Crear tabla de atributos
     df = pd.DataFrame([{
@@ -68,7 +61,7 @@ def generar_grafico_dependencia(texto):
         "is_sent_start": str(token.is_sent_start) if token.is_sent_start is not None else "None"
     } for token in doc])
 
-    return rutas_imagenes, df
+    return imagenes, df
 
 # Ejemplos
 ejemplos = [
@@ -93,32 +86,31 @@ with gr.Blocks(title="Visualización de Dependencias Sintácticas", theme=gr.the
 
     with gr.Row():
         with gr.Column(scale=1):
-          texto_input = gr.Textbox(lines=4, label="Texto en español", placeholder="Escribe aquí tu frase...")
-          boton = gr.Button("Generar gráfico")
-          gr.Examples(
-              examples=ejemplos,
-              inputs=texto_input,
-              # outputs=galeria,
-              fn=generar_grafico_dependencia,
-              label="Ejemplos de texto"
-          )
+            texto_input = gr.Textbox(lines=4, label="Texto en español", placeholder="Escribe aquí tu frase...")
+            boton = gr.Button("Generar gráfico")
+            gr.Examples(
+                examples=ejemplos,
+                inputs=texto_input,
+                fn=generar_grafico_dependencia,
+                label="Ejemplos de texto"
+            )
 
         with gr.Column(scale=2):
-          with gr.Tab("visualización"):
-            galeria = gr.Gallery(label="Gráfico(s) generado(s)", show_label=True, columns=4, height="auto", elem_id="gallery")
-          with gr.Tab("tabla"):
-            tabla = gr.Dataframe(
-                headers=[
-                    "idx", "text", "lemma_", "pos_", "tag_", "dep_", "head", "morph",
-                    "ent_type_", "ent_iob_", "shape_", "is_alpha", "is_ascii",
-                    "is_digit", "is_punct", "like_num", "is_sent_start"
-                ],
-                label="Atributos de tokens",
-                interactive=False,
-                type="pandas",
-                row_count=5,
-                max_height="600"
-            )
+            with gr.Tab("visualización"):
+                galeria = gr.Gallery(label="Gráfico(s) generado(s)", show_label=True, columns=4, height="auto", elem_id="gallery")
+            with gr.Tab("tabla"):
+                tabla = gr.Dataframe(
+                    headers=[
+                        "idx", "text", "lemma_", "pos_", "tag_", "dep_", "head", "morph",
+                        "ent_type_", "ent_iob_", "shape_", "is_alpha", "is_ascii",
+                        "is_digit", "is_punct", "like_num", "is_sent_start"
+                    ],
+                    label="Atributos de tokens",
+                    interactive=False,
+                    type="pandas",
+                    row_count=5,
+                    max_height="600"
+                )
 
     boton.click(fn=generar_grafico_dependencia, inputs=texto_input, outputs=[galeria, tabla])