Update app.py

app.py

@@ -72,8 +72,6 @@ if 'is_example' not in st.session_state: st.session_state.is_example = False
 @st.cache_resource
 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!)...")
-    
     bert_path = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_BERT)
     horizon_path = hf_hub_download(repo_id=MODEL_REPO_ID, filename=MODEL_FILE_HORIZON)
 
@@ -86,12 +84,18 @@ def init_backend(csv, cache, bert_model_path):
     df = dm.cargar_datos()
     return dm, df
 
-# --- 1. CARGA DE DATOS Y MODELOS (Cacheado) ---
+# --- 1. CARGA DE DATOS Y MODELOS (Cacheado y estabilizado) ---
 try:
     if st.session_state.data_manager is None:
         
+        # 1. Usar contenedor para el mensaje de descarga (Estabiliza el salto)
+        status_message = st.empty()
+        status_message.info("Descargando modelos grandes desde Hugging Face Hub (¡Solo la primera vez!)...")
+        
         bert_path_downloaded, horizon_path_downloaded = download_models()
         
+        # 2. Limpiar el placeholder de descarga y mostrar el spinner de carga de datos
+        status_message.empty() 
         with st.spinner("Cargando base de datos de muebles y modelos IA..."):
             
             dm, df = init_backend(csv_path, cache_path, bert_path_downloaded)
@@ -100,6 +104,8 @@ try:
             st.session_state.muebles_df = df
             st.session_state.horizon_model_path = horizon_path_downloaded 
             
+        st.toast("Modelos y datos cargados con éxito.", icon="✅") 
+        
     st.sidebar.success(f"Base de datos cargada: {len(st.session_state.muebles_df)} items")
 
 except Exception as e:
@@ -140,41 +146,33 @@ st.header("1. Escaneo de Habitación")
 
 # --- LÓGICA DE SELECCIÓN DE EJEMPLOS EN EL CUERPO PRINCIPAL ---
 examples_dir = os.path.join(os.path.dirname(__file__), "examples")
-source_file = None
-file_caption = None
+image_placeholder = st.empty() # Placeholder para la imagen (Estabiliza el temblor)
 
 if os.path.exists(examples_dir):
     example_files = [f for f in os.listdir(examples_dir) if f.endswith(('.jpg', '.png'))]
     example_files.sort()
     
-    # 1. Creamos el selector de ejemplos
-    display_files = ["--- Cargar imagen manualmente (prioridad) ---"] + example_files 
-    
-    # Usamos st.empty para evitar re-render innecesario al mostrar la imagen
     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',
-            # El on_change es clave para forzar la actualización solo cuando el valor cambia
-            on_change=lambda: st.session_state.update(
-                # Forzar el re-render solo si el valor ha cambiado
-                source_file_path=os.path.join(examples_dir, st.session_state.example_select_box) if st.session_state.example_select_box != display_files[0] else None,
-                is_example=st.session_state.example_select_box != display_files[0]
-            )
+            key='example_select_box'
         )
 
-    # 2. Manejo de la selección del ejemplo
-    is_example_selected = st.session_state.get('is_example', False)
-    source_file_path = st.session_state.get('source_file_path', None)
+    # 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
 
-    if is_example_selected and source_file_path:
         with col_download:
-            # Poner el botón de descarga en la columna derecha
             st.markdown(" ") 
-            with open(source_file_path, "rb") as file:
+            with open(file_path, "rb") as file:
                 st.download_button(
                     label=f"⬇️ Descargar: {selected_example_name}",
                     data=file,
@@ -182,24 +180,34 @@ if os.path.exists(examples_dir):
                     mime="image/jpeg"
                 )
         
-        # Mostrar miniatura del ejemplo
-        st.image(source_file_path, caption="Vista previa del ejemplo", use_container_width=True)
+        # Mostrar miniatura del ejemplo en el placeholder
+        with image_placeholder.container():
+            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
+             image_placeholder.empty() # Limpiar el placeholder si no hay selección de ejemplo
 
-# 2. 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'])
+
+# 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'], key='main_file_uploader')
 
 # Determinar qué archivo usar (la carga manual tiene prioridad)
-source_file_path_session = st.session_state.get('source_file_path', None)
-is_example_session = st.session_state.get('is_example', False)
+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'
     is_example = False
-elif source_file_path_session and is_example_session:
-    source_file = source_file_path_session
-    file_caption = f'Imagen de ejemplo: {source_file_path_session.split(os.sep)[-1]}'
-    is_example = True
+    st.session_state.is_example = False # Resetear por seguridad
+    image_placeholder.empty() # Limpiar el placeholder si hay una subida manual
+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
 
@@ -213,7 +221,8 @@ if source_file is not None:
 
     # 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)
+        with image_placeholder.container():
+             st.image(image, caption=file_caption, use_container_width=True)
     
     if st.button("Analizar la habitación"):
         with st.spinner("Detectando la geometría..."):
@@ -241,7 +250,8 @@ if source_file is not None:
                 else:
                     st.session_state.room_data = room_data
                     st.session_state.stage = 1
-                    st.success("Análisis completado")
+                    
+                    st.toast("Análisis completado", icon="✅") 
 
                     # Mostrar resultado de la detección de HorizonNet
                     st.header("Resultado del análisis visual")
@@ -359,14 +369,14 @@ if st.session_state.stage >= 2:
     presupuesto = st.number_input("Presupuesto Máximo (€)", min_value=100.0, value=1000.0, step=100.0)
     
     if st.button("Generar diseño"):
-        # Convertir dimensiones de m a cm para el LayoutEngine
-        w_cm = st.session_state.room_data.get('width', 0.0) * 100
-        l_cm = st.session_state.room_data.get('length', 0.0) * 100
-        
-        if w_cm < 200 or l_cm < 200:
-            st.error("Las dimensiones de la habitación son demasiado pequeñas (mínimo 2x2m) o no fueron capturadas correctamente.")
-        else:
-            with st.spinner("Calculando distribución óptima y seleccionando muebles..."):
+        with st.spinner("Calculando distribución óptima y seleccionando muebles..."):
+            # Convertir dimensiones de m a cm para el LayoutEngine
+            w_cm = st.session_state.room_data.get('width', 0.0) * 100
+            l_cm = st.session_state.room_data.get('length', 0.0) * 100
+            
+            if w_cm < 200 or l_cm < 200:
+                st.error("Las dimensiones de la habitación son demasiado pequeñas (mínimo 2x2m) o no fueron capturadas correctamente.")
+            else:
                 # 1. Inicializar motores
                 layout_engine = logic.LayoutEngine(st.session_state.data_manager.dimensiones_promedio)
                 recommender = logic.Recommender(st.session_state.muebles_df)