XA-vito commited on
Commit
01f2b02
·
verified ·
1 Parent(s): 4ba974f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -4
app.py CHANGED
@@ -123,14 +123,21 @@ def conversar_con_colmena(mensaje):
123
  """Genera una respuesta combinando el modelo de lenguaje con análisis de datos."""
124
  datos = obtener_datos_colmena()
125
  if "error" in datos:
126
- return datos["error"], None # Devolvemos texto y None para la imagen
127
-
128
  resultado = interpretar_instruccion(mensaje, datos)
129
  if isinstance(resultado, tuple):
130
  texto, grafico = resultado
131
- return texto, grafico
 
 
 
 
 
 
132
  else:
133
- return resultado, None # Si no hay imagen, devolvemos None
 
134
 
135
 
136
  iface = gr.Interface(
 
123
  """Genera una respuesta combinando el modelo de lenguaje con análisis de datos."""
124
  datos = obtener_datos_colmena()
125
  if "error" in datos:
126
+ return datos["error"], None # Retorna un string y `None` para evitar error en la imagen
127
+
128
  resultado = interpretar_instruccion(mensaje, datos)
129
  if isinstance(resultado, tuple):
130
  texto, grafico = resultado
131
+ if grafico:
132
+ # Convierte BytesIO en una imagen para Gradio
133
+ from PIL import Image
134
+ img = Image.open(grafico)
135
+ return texto, img
136
+ else:
137
+ return texto, None
138
  else:
139
+ return resultado, None # Asegurar que siempre haya dos valores de retorno
140
+
141
 
142
 
143
  iface = gr.Interface(