Force Error

app.py

@@ -1547,15 +1547,10 @@ def run_model(model_name):
                     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  
+                    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)
                     
                     # Definir el color mapper
                     if selected_feature in model_options:
@@ -1589,16 +1584,25 @@ def run_model(model_name):
                     heatmap_fig.border_fill_color = "white"
 
                     # Dibujar la imagen del heatmap
-                    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=[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 == "UMAP" and nan_mask.any():
+                        transparent_layer = np.full_like(heatmap_data, np.nan)
+                        transparent_layer[nan_mask] = np.nanmin(heatmap_data)
+                        transparent_mapper = LinearColorMapper(palette=["rgba(255,255,255,0)"], low=np.nanmin(heatmap_data), high=np.nanmax(heatmap_data))
                         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
+                            color_mapper=transparent_mapper
                         )
                     
                     # Agregar la barra de color
@@ -1945,6 +1949,7 @@ def main():
     config_style()
     tabs = st.tabs(["Donut", "Idefics2", "Idefics2-patient", "Paligemma", "Llava"])
     with tabs[0]:
+        print(ahhas)
         st.markdown('<h2 class="sub-title">Donut 🤗</h2>', unsafe_allow_html=True)
         run_model("Donut")
     with tabs[1]: