Try Again

app.py

@@ -1072,7 +1072,6 @@ def run_model(model_name):
             )
 
         reduced_real = reducer_real.fit_transform(df_real_only[embedding_cols].values)
-        reduced_real = np.nan_to_num(reduced_real).astype(float)
 
         # Agregar columnas PC1, PC2, … a df_real_only
         for i in range(reduced_real.shape[1]):
@@ -1547,6 +1546,16 @@ def run_model(model_name):
                     # Transponer y limpiar valores inválidos (NaN, inf, -inf)
                     heatmap_data = heat_stat.T
                     # heatmap_data = np.nan_to_num(heat_stat.T, nan=0.0, posinf=0.0, neginf=0.0)
+
+                    if reduction_method == "UMAP":
+                        nan_mask = np.isnan(heatmap_data)
+                        valid_min = np.nanmin(heatmap_data)
+                        heatmap_data = np.nan_to_num(heatmap_data, nan=valid_min, posinf=valid_min, neginf=valid_min)
+                        heatmap_data = heatmap_data.astype(float)
+
+                        if nan_mask.any():
+                            transparent_layer = np.zeros_like(heatmap_data)
+                            transparent_layer[~nan_mask] = np.nan  
                     
                     # Definir el color mapper
                     if selected_feature in model_options:
@@ -1580,9 +1589,17 @@ def run_model(model_name):
                     heatmap_fig.border_fill_color = "white"
 
                     # 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,
-                                    color_mapper=color_mapper)
+                    if reduction_method == "PCA":
+                        heatmap_fig.image(image=[heatmap_data], x=x_min, y=y_min,
+                                        dw=x_max - x_min, dh=y_max - y_min,
+                                        color_mapper=color_mapper)
+                    else:
+                        heatmap_fig.image(
+                            image=[transparent_layer],
+                            x=x_min, y=y_min,
+                            dw=x_max - x_min, dh=y_max - y_min,
+                            color_mapper=color_mapper
+                        )
                     
                     # Agregar la barra de color
                     color_bar = ColorBar(color_mapper=color_mapper, location=(0, 0))