Test Chromium

app.py

@@ -16,6 +16,24 @@ import json
 import itertools
 import matplotlib.pyplot as plt
 import matplotlib.colors as mcolors
+import zipfile
+import tempfile
+import shutil
+from selenium.webdriver.firefox.options import Options
+from selenium.webdriver.firefox.service import Service
+from selenium import webdriver
+from bokeh.io import export_png
+import os
+
+
+from selenium import webdriver
+from selenium.webdriver.firefox.service import Service as ChromeService
+from selenium.webdriver.firefox.options import Options
+from webdriver_manager.chrome import ChromeDriverManager
+
+
+
+
 
 
 N_COMPONENTS = 3
@@ -1166,6 +1184,10 @@ def run_model(model_name):
                 x_bins = np.linspace(x_min, x_max, grid_size + 1)
                 y_bins = np.linspace(y_min, y_max, grid_size + 1)
                 
+                # Listas para almacenar las figuras de heatmap y sus nombres
+                heatmap_figures = []
+                heatmap_names = []
+
                 # Generar heatmaps para cada combinación de componentes
                 pairs = list(itertools.combinations(range(N_COMPONENTS), 2))
                 for (i, j) in pairs:
@@ -1202,7 +1224,7 @@ def run_model(model_name):
                     # Transponer la matriz para alinear correctamente los ejes
                     heatmap_data = heat_stat.T
                     
-                    # Definir el color mapper: si la feature es de un modelo (model_options) usar la paleta rojo-verde con rango 0 a 1
+                    # Definir el color mapper
                     if selected_feature in model_options:
                         color_mapper = LinearColorMapper(
                             palette=red_green_palette,
@@ -1226,8 +1248,10 @@ def run_model(model_name):
                                         sizing_mode="fixed")
                     heatmap_fig.match_aspect = True
                     
+                    # Asignar etiquetas a los ejes
                     heatmap_fig.xaxis.axis_label = x_comp
                     heatmap_fig.yaxis.axis_label = y_comp
+
                     # Dibujar la imagen del heatmap
                     heatmap_fig.image(image=[heatmap_data], x=x_min, y=y_min,
                                     dw=x_max - x_min, dh=y_max - y_min,
@@ -1250,7 +1274,7 @@ def run_model(model_name):
                         """)
                     heatmap_fig.add_layout(color_bar, 'right')
                     
-                    # Agregar renderer invisible para tooltips (usando puntos en cada bin)
+                    # Agregar renderer invisible para tooltips
                     source_points = ColumnDataSource(data={
                         'x': df_heatmap['x'],
                         'y': df_heatmap['y'],
@@ -1259,7 +1283,6 @@ def run_model(model_name):
                     })
                     invisible_renderer = heatmap_fig.circle('x', 'y', size=10, source=source_points, fill_alpha=0, line_alpha=0.5)
                     
-                    # Si se selecciona un dataset extra, proyectar sus puntos en la combinación actual
                     if select_extra_dataset_hm != "-":
                         df_extra = df_all["synthetic"][df_all["synthetic"]["source"] == select_extra_dataset_hm].copy()
                         df_extra["x"] = df_extra[x_comp]
@@ -1277,7 +1300,42 @@ def run_model(model_name):
                     hover_tool_points = HoverTool(renderers=[invisible_renderer], tooltips=TOOLTIPS)
                     heatmap_fig.add_tools(hover_tool_points)
                     
+                    # Mostrar el heatmap en la app
                     st.bokeh_chart(heatmap_fig)
+                    
+                    # Guardar la figura y el nombre correspondiente
+                    heatmap_figures.append(heatmap_fig)
+                    heatmap_names.append(f"heatmap_{selected_feature}_{x_comp}_vs_{y_comp}.png")
+
+
+                    # Inicializar el driver de Chrome en modo headless
+                    chrome_options = Options()
+                    chrome_options.add_argument("--headless")
+                    chrome_options.add_argument("--no-sandbox")
+                    chrome_options.add_argument("--disable-dev-shm-usage")
+                    driver = webdriver.Chrome(options=chrome_options, service=ChromeService(ChromeDriverManager().install()))
+
+                    if st.button("Descargar todos los heatmaps"):
+                        zip_buffer = io.BytesIO()
+                        with tempfile.TemporaryDirectory() as tmpdirname:
+                            with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zip_file:
+                                # Exporta cada figura a PNG y añádela al ZIP
+                                for fig, filename in zip(heatmap_figures, heatmap_names):
+                                    tmp_path = f"{tmpdirname}/{filename}"
+                                    export_png(fig, filename=tmp_path, driver=driver)
+                                    with open(tmp_path, "rb") as f:
+                                        zip_file.writestr(filename, f.read())
+                        zip_buffer.seek(0)
+                        st.download_button(
+                            label="Descargar ZIP de heatmaps",
+                            data=zip_buffer,
+                            file_name="heatmaps.zip",
+                            mime="application/zip",
+                            key=f"download_heatmaps_{model_name}"
+                        )
+
+                    driver.quit()
+
 
 def main():
     config_style()