Fix Bug

app.py

@@ -1501,19 +1501,28 @@ def run_model(model_name):
                 feature_options = [col for col in df_heat.columns if col != "name"]
                 selected_feature = st.selectbox("Select heatmap feature:", 
                                                 options=feature_options, key=f"heatmap_{model_name}")
-                # select_extra_dataset_hm = st.selectbox("Select a dataset:", 
-                #                                 options=model_options_with_default, key=f"heatmap_extra_dataset_{model_name}")
-                
-                # Definir un rango fijo y rejilla
-                x_min, x_max = -1, 1
-                y_min, y_max = -1, 1
-                grid_size = 50
-                x_bins = np.linspace(x_min, x_max, grid_size + 1)
-                y_bins = np.linspace(y_min, y_max, grid_size + 1)
+                select_extra_dataset_hm = st.selectbox("Select a dataset:", 
+                                                options=model_options_with_default, key=f"heatmap_extra_dataset_{model_name}")        
                 
                 # Solo una combinación: TSNE1 vs TSNE2
                 x_comp, y_comp = "TSNE1", "TSNE2"
                 st.markdown(f"### Heatmap: {x_comp} vs {y_comp}")
+
+                # Rango real de los datos
+                x_min, x_max = df_heatmap[x_comp].min(), df_heatmap[x_comp].max()
+                y_min, y_max = df_heatmap[y_comp].min(), df_heatmap[y_comp].max()
+
+                # Opcional: ampliar un poco
+                padding = 0.05
+                x_min, x_max = x_min - padding, x_max + padding
+                y_min, y_max = y_min - padding, y_max + padding
+
+                # Definir rejilla
+                grid_size = st.slider("Grid size (resolución del heatmap)", min_value=10, max_value=100, value=40, step=5)
+                x_bins = np.linspace(x_min, x_max, grid_size + 1)
+                y_bins = np.linspace(y_min, y_max, grid_size + 1)
+                x_range = (x_min - padding, x_max + padding)
+                y_range = (y_min - padding, y_max + padding)
                 
                 df_heatmap = df_heatmap_base.copy()
                 df_heatmap["x"] = df_heatmap[x_comp]
@@ -1555,13 +1564,7 @@ def run_model(model_name):
                         nan_color='rgba(0, 0, 0, 0)'
                     )
                 
-                x_min, x_max = df_heatmap[x_comp].min(), df_heatmap[x_comp].max()
-                y_min, y_max = df_heatmap[y_comp].min(), df_heatmap[y_comp].max()
-
-                # opcional: ampliar un poco para que no quede todo pegado al borde
-                padding = 0.05
-                x_range = (x_min - padding, x_max + padding)
-                y_range = (y_min - padding, y_max + padding)
+                
 
                 # Figura heatmap
                 heatmap_fig = figure(title=f"Heatmap de '{selected_feature}' ({x_comp} vs {y_comp})",
@@ -1603,19 +1606,19 @@ def run_model(model_name):
                 })
                 invisible_renderer = heatmap_fig.circle('x', 'y', size=10, source=source_points, fill_alpha=0, line_alpha=0.5)
                 
-                # 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]
-                #     df_extra["y"] = df_extra[y_comp]
-                #     if 'name' not in df_extra.columns:
-                #         df_extra["name"] = df_extra["img"].apply(lambda x: x.split("/")[-1].replace(".png", "") if isinstance(x, str) else x)
-                #     source_extra_points = ColumnDataSource(data={
-                #         'x': df_extra['x'],
-                #         'y': df_extra['y'],
-                #         'img': df_extra['img'],
-                #         'label': df_extra['name']
-                #     })
-                #     heatmap_fig.circle('x', 'y', size=5, source=source_extra_points, fill_alpha=0, line_alpha=0.5, color="purple")
+                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]
+                    df_extra["y"] = df_extra[y_comp]
+                    if 'name' not in df_extra.columns:
+                        df_extra["name"] = df_extra["img"].apply(lambda x: x.split("/")[-1].replace(".png", "") if isinstance(x, str) else x)
+                    source_extra_points = ColumnDataSource(data={
+                        'x': df_extra['x'],
+                        'y': df_extra['y'],
+                        'img': df_extra['img'],
+                        'label': df_extra['name']
+                    })
+                    heatmap_fig.circle('x', 'y', size=5, source=source_extra_points, fill_alpha=0, line_alpha=0.5, color="purple")
                 
                 hover_tool_points = HoverTool(renderers=[invisible_renderer], tooltips=TOOLTIPS)
                 heatmap_fig.add_tools(hover_tool_points)