Update app.py

app.py

@@ -140,32 +140,41 @@ 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
 
 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'
+            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]
+            )
         )
 
-    # 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
+    # 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)
 
+    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(file_path, "rb") as file:
+            with open(source_file_path, "rb") as file:
                 st.download_button(
                     label=f"⬇️ Descargar: {selected_example_name}",
                     data=file,
@@ -174,29 +183,23 @@ if os.path.exists(examples_dir):
                 )
         
         # 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
-
+        st.image(source_file_path, caption="Vista previa del ejemplo", use_container_width=True)
 
-# 1. Selector de carga manual (Se mantiene aquí, pero solo para subir el archivo)
+# 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'])
 
 # 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)
+source_file_path_session = st.session_state.get('source_file_path', None)
+is_example_session = 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 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]}'
+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
 else:
     source_file = None