Implementado sistema de selección visual para botones: el botón activo se muestra en rojo y se añade mensaje de confirmación

utils/preprocessing_ui.py

@@ -38,6 +38,10 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
         # Initialize result
         result = {"success": True, "message": ""}
         
+        # Set default selection to improved image if not already set
+        if "selected_image_mode" not in st.session_state:
+            st.session_state["selected_image_mode"] = "improved"
+        
         # Create an expander for the suggested improvements
         with st.expander("Suggested improvements available", expanded=True):
             # Create a two-column layout for original and improved images
@@ -46,10 +50,16 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
             with col1:
                 st.markdown("**Original Image**")
                 st.image(img, use_column_width=True)
+                
+                # Determine button style based on current selection
+                original_button_type = "secondary"
+                if st.session_state["selected_image_mode"] == "original":
+                    original_button_type = "primary"
+                
                 # Botón "Continue with Original" debajo de la imagen original
                 continue_original = st.button("Continue with Original", 
                                           key="continue_original_btn", 
-                                          type="primary", 
+                                          type=original_button_type, 
                                           use_container_width=True)
             
             # Apply basic enhancements and display the improved version
@@ -74,22 +84,40 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
                 with col2:
                     st.markdown("**Improved Image**")
                     st.image(improved_rgb, use_column_width=True)
+                    
+                    # Determine button style based on current selection
+                    improved_button_type = "secondary"
+                    if st.session_state["selected_image_mode"] == "improved":
+                        improved_button_type = "primary"
+                    
                     # Botón "Use Improved Image" debajo de la imagen mejorada
                     use_improved = st.button("Use Improved Image", 
                                            key="use_improved_btn", 
-                                           type="secondary",
+                                           type=improved_button_type,
                                            use_container_width=True)
                 
                 # Store the user's choice for later use
                 if use_improved:
                     st.session_state["use_improved_image"] = True
+                    st.session_state["selected_image_mode"] = "improved"
                     # We'll use this modified image for facial detection
                     st.session_state.current_image = improved_rgb
-                    st.info("Using improved image for analysis.")
+                    
+                    # Display a success message with the selected image
+                    st.success("Using improved image for analysis.")
                 elif continue_original:
                     st.session_state["use_improved_image"] = False
+                    st.session_state["selected_image_mode"] = "original"
                     st.session_state.current_image = img
+                    
+                    # Display a success message with the selected image
                     st.info("Using original image for analysis.")
+                    
+                # Show a message about the currently selected image
+                if st.session_state["selected_image_mode"] == "improved":
+                    st.markdown("<div style='background-color: #1E3A8A; padding: 10px; border-radius: 5px; color: white;'><center>Using improved image for analysis.</center></div>", unsafe_allow_html=True)
+                else:
+                    st.markdown("<div style='background-color: #1E3A8A; padding: 10px; border-radius: 5px; color: white;'><center>Using original image for analysis.</center></div>", unsafe_allow_html=True)
                 
                 # List the improvements made
                 st.markdown("**Improvements applied:**")