Fix Bug

app.py

@@ -1448,6 +1448,14 @@ def run_model(model_name):
         df_real_only['TSNE1'] = reduced_real[:, 0]
         df_real_only['TSNE2'] = reduced_real[:, 1]
 
+        # Construir df_all de forma consistente con PCA
+        df_all = {}
+        df_all["real"] = df_real_only.copy()
+        if "synthetic" in embeddings:
+            df_all["synthetic"] = embeddings["synthetic"].copy()
+        if "pretrained" in embeddings:
+            df_all["pretrained"] = embeddings["pretrained"].copy()
+
         unique_labels_real = sorted(df_real_only['label'].unique().tolist())
         
         # Mapeo de colores para las muestras reales usando la paleta Reds9
@@ -1520,8 +1528,8 @@ 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}")
+                # 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
@@ -1608,19 +1616,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)