Include Extra Dataset In Heatmap

app.py

@@ -1057,6 +1057,14 @@ def run_model(model_name):
 
         st.write(f"Regresión global (Nueva PCA): R² = {r2_new:.4f}, Slope = {slope_new:.4f}, Intercept = {intercept_new:.4f}")
 
+
+
+
+
+
+
+
+
         # --- INICIO DEL BLOQUE: Heatmap de características ---
         st.markdown("## Heatmap de Características")
 
@@ -1083,9 +1091,12 @@ def run_model(model_name):
                 
                 # Extraemos las características disponibles (excluyendo 'name')
                 feature_options = [col for col in df_heat.columns if col != "name"]
-                selected_feature = st.selectbox("Seleccione la característica para el heatmap:", 
+                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=["-", "es-digital-seq", "es-digital-rotation-degradation-seq", "es-digital-zoom-degradation-seq", "es-render-seq"], key=f"heatmap_extra_dataset_{model_name}")
+                
                 # Determinar el rango de las posiciones (x, y) de las muestras reales
                 x_min, x_max = df_heatmap['x'].min(), df_heatmap['x'].max()
                 y_min, y_max = df_heatmap['y'].min(), df_heatmap['y'].max()
@@ -1168,6 +1179,22 @@ def run_model(model_name):
                 # Dibujar círculos con transparencia total (no se verán)
                 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]
+                    # Asegurarse de que exista la columna 'name'
+                    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']
+                    })
+                    # Agregar renderer para el dataset extra
+                    extra_renderer = heatmap_fig.circle('x', 'y', size=10, source=source_extra_points, fill_alpha=0, line_alpha=0.5, color="red")
+
                 hover_tool_points = HoverTool(renderers=[invisible_renderer], tooltips=TOOLTIPS)
                 heatmap_fig.add_tools(hover_tool_points)