Update app.py

app.py

@@ -4,21 +4,20 @@ import numpy as np
 from PIL import Image
 import os
 from huggingface_hub import hf_hub_download 
-
 import logic 
 
-# --- CONFIGURACIÓN INICIAL DE LA PÁGINA ---
-st.set_page_config(page_title="DesignIA - Recomendador de Muebles inteligente", layout="wide")
-
 # --- CONSTANTES Y RUTAS DE RECURSOS ---
 csv_path = "data/furniture_data.csv"
-cache_path = "vectores_cache.pkl"
+cache_path = "vectores_cache.pkl" 
 
-# --- CONSTANTES DE DESCARGA DE MODELOS (USANDO TU REPOSITORIO) ---
+# --- CONSTANTES DE DESCARGA DE MODELOS ---
 MODEL_REPO_ID = "agerhund/DesignIA_models" 
 MODEL_FILE_BERT = "bert_style_encoder.pth"
 MODEL_FILE_HORIZON = "horizonnet_model.pth"
 
+# --- CONFIGURACIÓN INICIAL DE LA PÁGINA ---
+st.set_page_config(page_title="DesignIA - Recomendador de Muebles inteligente", layout="wide")
+
 # --- CARGAR ESTILOS CSS ---
 def cargar_estilo():
     """Define y aplica estilos CSS para la UI de Streamlit."""
@@ -51,7 +50,7 @@ def cargar_estilo():
             box-shadow: 0 2px 5px rgba(0,0,0,0.05); 
             margin-bottom: 15px; 
             border: 1px solid #eee;
-            color: black !important; /* Forzar texto negro dentro de la tarjeta blanca */
+            color: black !important; 
         }
         </style>
     """, unsafe_allow_html=True)
@@ -64,8 +63,8 @@ if 'room_data' not in st.session_state: st.session_state.room_data = None
 if 'muebles_df' not in st.session_state: st.session_state.muebles_df = None
 if 'data_manager' not in st.session_state: st.session_state.data_manager = None
 if 'horizon_model_path' not in st.session_state: st.session_state.horizon_model_path = None
-if 'selected_example_path' not in st.session_state: st.session_state.selected_example_path = None
-if 'selected_example_name' not in st.session_state: st.session_state.selected_example_name = None
+if 'source_file_path' not in st.session_state: st.session_state.source_file_path = None
+if 'is_example' not in st.session_state: st.session_state.is_example = False
 
 
 # --- FUNCIONES DE CARGA CON CACHE ---
@@ -75,10 +74,7 @@ def download_models():
     """Descarga ambos modelos grandes desde el Model Repository y devuelve las rutas locales temporales."""
     st.info("Descargando modelos grandes desde Hugging Face Hub (¡Solo la primera vez!)...")
     
-    # 1. Descarga del modelo BERT/Style (para DataManager)
     bert_path = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_BERT)
-    
-    # 2. Descarga del modelo HorizonNet (para RoomLayoutDetector)
     horizon_path = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_HORIZON)
 
     return bert_path, horizon_path
@@ -86,7 +82,6 @@ def download_models():
 @st.cache_resource
 def init_backend(csv, cache, bert_model_path):
     """Inicializa DataManager y carga el DataFrame de muebles, usando la ruta local del modelo BERT."""
-    # El DataManager (logic.py) utiliza la ruta del modelo BERT para sus operaciones.
     dm = logic.DataManager(csv, cache, bert_model_path)
     df = dm.cargar_datos()
     return dm, df
@@ -95,15 +90,12 @@ def init_backend(csv, cache, bert_model_path):
 try:
     if st.session_state.data_manager is None:
         
-        # 1. Descarga los modelos y obtiene las rutas locales temporales
         bert_path_downloaded, horizon_path_downloaded = download_models()
         
         with st.spinner("Cargando base de datos de muebles y modelos IA..."):
             
-            # 2. Inicializa el DataManager (que solo necesita el modelo BERT)
             dm, df = init_backend(csv_path, cache_path, bert_path_downloaded)
             
-            # 3. Guarda todos los objetos y la RUTA DE HORIZONNET para el PASO 1
             st.session_state.data_manager = dm
             st.session_state.muebles_df = df
             st.session_state.horizon_model_path = horizon_path_downloaded 
@@ -115,7 +107,7 @@ except Exception as e:
     st.stop()
 
 
-# --- SIDEBAR: INFORMACIÓN DEL PROYECTO ---
+# --- SIDEBAR: INFORMACIÓN DEL PROYECTO (SOLO INFORMACIÓN ESTÁTICA) ---
 with st.sidebar:
     st.title("DesignIA - Asistente de Diseño")
     st.markdown("---")
@@ -126,7 +118,7 @@ with st.sidebar:
     st.markdown("Desarrollado por **Andrés Gerlotti Slusnys**")
     st.markdown("© 2025")
     
-    # Indicador de estado para debugging
+    # Indicador de estado 
     with st.expander("Estado del Sistema", expanded=False):
         st.success("Motor Gráfico: Activo")
         st.success("Modelo NLP (BERT): Cargado")
@@ -134,43 +126,9 @@ with st.sidebar:
             st.success("HorizonNet: Conectado (Remoto)")
         else:
             st.warning("HorizonNet: Pendiente de descarga/inicialización")
-
-    # Sección de Ejemplos
-    st.markdown("---")
-    st.subheader("📸 Imágenes de Ejemplo")
-    st.caption("Descarga estas imágenes para probar la app:")
     
-    examples_dir = os.path.join(os.path.dirname(__file__), "examples")
-    if os.path.exists(examples_dir):
-        example_files = [f for f in os.listdir(examples_dir) if f.endswith(('.jpg', '.png'))]
-        example_files.sort()
-        
-        # Guardar el nombre del archivo seleccionado en la sesión
-        selected_example = st.selectbox("Selecciona un ejemplo:", example_files, key='example_select_box')
-        
-        if selected_example:
-            file_path = os.path.join(examples_dir, selected_example)
-            
-            # 🛑 Corrección: Guardar la ruta y el nombre del ejemplo seleccionado en session_state
-            st.session_state.selected_example_path = file_path
-            st.session_state.selected_example_name = selected_example
-
-            with open(file_path, "rb") as file:
-                btn = st.download_button(
-                    label="⬇️ Descargar Imagen",
-                    data=file,
-                    file_name=selected_example,
-                    mime="image/jpeg"
-                )
-            
-            # Mostrar miniatura
-            st.image(file_path, caption="Vista previa", use_container_width=True)
-        else:
-             # Limpiar la ruta si no hay nada seleccionado
-            if 'selected_example_path' in st.session_state:
-                del st.session_state.selected_example_path
-    else:
-        st.info("No hay ejemplos disponibles.")
+    st.markdown("---")
+    st.info("El selector de imágenes de ejemplo se encuentra en el Paso 1 de la página principal.")
 
 
 # --- INTERFAZ PRINCIPAL ---
@@ -180,49 +138,95 @@ st.markdown("Sube una panorámica, detecta el espacio y obtén el diseño ideal
 # --- PASO 1: CARGA DE IMAGEN Y DETECCIÓN ---
 st.header("1. Escaneo de Habitación")
 
-# 1. Selector de carga manual
-uploaded_file = st.file_uploader("Sube tu imagen panorámica (360)", type=['jpg', 'png', 'jpeg'])
+# --- LÓGICA DE SELECCIÓN DE EJEMPLOS EN EL CUERPO PRINCIPAL ---
+examples_dir = os.path.join(os.path.dirname(__file__), "examples")
 
-# Obtener rutas de archivos de ejemplo desde session_state (definidas en la barra lateral)
-selected_example_path = st.session_state.get('selected_example_path', None)
-selected_example_name = st.session_state.get('selected_example_name', None)
+if os.path.exists(examples_dir):
+    example_files = [f for f in os.listdir(examples_dir) if f.endswith(('.jpg', '.png'))]
+    example_files.sort()
+    
+    col_select, col_download = st.columns([3, 1])
+
+    with col_select:
+        # Añadimos la opción de carga manual como la principal
+        display_files = ["--- Cargar imagen manualmente (prioridad) ---"] + example_files 
+        
+        selected_example_name = st.selectbox(
+            "O usa una imagen de ejemplo:", 
+            display_files,
+            key='example_select_box'
+        )
+
+    # Manejo de la selección del ejemplo
+    if selected_example_name != display_files[0]:
+        file_path = os.path.join(examples_dir, selected_example_name)
+        st.session_state.source_file_path = file_path 
+        st.session_state.is_example = True
+
+        with col_download:
+            st.markdown(" ") 
+            with open(file_path, "rb") as file:
+                st.download_button(
+                    label=f"⬇️ Descargar: {selected_example_name}",
+                    data=file,
+                    file_name=selected_example_name,
+                    mime="image/jpeg"
+                )
+        
+        # Mostrar miniatura del ejemplo
+        st.image(file_path, caption="Vista previa del ejemplo", use_container_width=True)
+
+    else:
+        # Limpiar las variables si se selecciona la opción nula
+        if 'source_file_path' in st.session_state:
+             del st.session_state.source_file_path
+             st.session_state.is_example = False
+
+
+# 1. Selector de carga manual (Se mantiene aquí, pero solo para subir el archivo)
+uploaded_file = st.file_uploader("Sube tu imagen panorámica (360)", type=['jpg', 'png', 'jpeg'])
 
 # Determinar qué archivo usar (la carga manual tiene prioridad)
+source_file_path = st.session_state.get('source_file_path', None)
+is_example = st.session_state.get('is_example', False)
+
 if uploaded_file is not None:
     source_file = uploaded_file
     file_caption = 'Imagen subida'
-elif selected_example_path is not None:
-    # Usar el archivo de la carpeta 'examples' como fuente
-    source_file = selected_example_path
-    file_caption = f'Imagen de ejemplo: {selected_example_name}'
+    is_example = False
+elif source_file_path is not None and is_example == True:
+    source_file = source_file_path
+    file_caption = f'Imagen de ejemplo: {source_file_path.split(os.sep)[-1]}'
 else:
     source_file = None
 
 
 if source_file is not None:
-
-    if isinstance(source_file, str):
+    # Abrir la imagen
+    if is_example:
         image = Image.open(source_file)
     else:
         image = Image.open(source_file)
 
-    st.image(image, caption=file_caption, use_container_width=True)
+    # Mostrar la imagen subida (solo si es subida, ya que el ejemplo se mostró arriba)
+    if uploaded_file is not None:
+        st.image(image, caption=file_caption, use_container_width=True)
     
     if st.button("Analizar la habitación"):
         with st.spinner("Detectando la geometría..."):
             
-            if isinstance(source_file, str):
+            # --- MANEJO DEL ARCHIVO TEMPORAL PARA EL DETECTOR ---
+            if is_example:
                 temp_file_path = source_file
             else:
                 temp_file_path = "temp_pano.jpg"
                 with open(temp_file_path, "wb") as f:
                     f.write(source_file.getbuffer())
-
+            # --- FIN MANEJO ARCHIVO TEMPORAL ---
 
 
             # Instanciar y ejecutar el detector de layout (HorizonNet)
             try:
-                # 🛑 USAR LA RUTA TEMPORAL GUARDADA EN session_state
                 detector = logic.RoomLayoutDetector(st.session_state.horizon_model_path) 
                 room_data = detector.detect_layout(temp_file_path)
                 
@@ -394,7 +398,6 @@ if st.session_state.stage >= 2:
                     st.error("No se pudo generar una distribución válida para este espacio (demasiado pequeño o muchos obstáculos).")
                 else:
                     # 5. Recomendar productos (Knapsack para optimización de precio/estilo)
-                    # Las 'constraints' se definen por los muebles que el layout PUDO colocar
                     best_combo = recommender.buscar_combinacion(constraints, presupuesto, top_n=1)
                     
                     if not best_combo:
@@ -411,7 +414,7 @@ if st.session_state.stage == 3:
     st.divider()
     st.header("Tu salón ideal")
     
-    # --- VISUALIZACIÓN 3D Interactiva ---
+    # --- VISUALIZACIÓN 3D Interactiva (Plotly) ---
     st.subheader("Visualización 3D Interactiva")
     
     # Generar la figura 3D