Simplificado el comportamiento de botones y reducido mensajes de confirmación para una experiencia más coherente

utils/preprocessing_ui.py

@@ -41,31 +41,16 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
         # 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"
+            st.session_state["use_improved_image"] = True
         
         # 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
             col1, col2 = st.columns(2)
             
-            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=original_button_type, 
-                                          use_container_width=True)
-            
-            # Apply basic enhancements and display the improved version
+            # Apply basic enhancements and prepare the improved version
             try:
-                # Enhance the image - in a real application, this would use AI-based enhancements
-                # For demonstration, we'll reduce brightness and contrast
+                # Standard enhancement parameters
                 brightness_factor = 0.7  # reduce brightness by 30%
                 contrast_factor = 0.7  # reduce contrast by 30%
                 
@@ -81,43 +66,39 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
                 # Convert back to RGB for display
                 improved_rgb = cv2.cvtColor(improved_bgr, cv2.COLOR_BGR2RGB)
                 
+                # Display images
+                with col1:
+                    st.markdown("**Original Image**")
+                    st.image(img, use_column_width=True)
+                
                 with col2:
                     st.markdown("**Improved Image**")
                     st.image(improved_rgb, use_column_width=True)
-                    
+                
+                # Use a single row for buttons with current selection visually indicated
+                col1, col2 = st.columns(2)
+                
+                with col1:
+                    # Determine button style based on current selection
+                    original_type = "primary" if st.session_state["selected_image_mode"] == "original" else "secondary"
+                    if st.button("Continue with Original", key="continue_original_btn", type=original_type, use_container_width=True):
+                        st.session_state["selected_image_mode"] = "original"
+                        st.session_state["use_improved_image"] = False
+                        st.session_state.current_image = img
+                        st.experimental_rerun()
+                
+                with col2:
                     # 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=improved_button_type,
-                                           use_container_width=True)
+                    improved_type = "primary" if st.session_state["selected_image_mode"] == "improved" else "secondary"
+                    if st.button("Use Improved Image", key="use_improved_btn", type=improved_type, use_container_width=True):
+                        st.session_state["selected_image_mode"] = "improved"
+                        st.session_state["use_improved_image"] = True
+                        st.session_state.current_image = improved_rgb
+                        st.experimental_rerun()
                 
-                # 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
-                    
-                    # 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)
+                # Single status message showing the current selection
+                message = f"Using {'improved' if st.session_state['selected_image_mode'] == 'improved' else 'original'} image for analysis."
+                st.info(message)
                 
                 # List the improvements made
                 st.markdown("**Improvements applied:**")
@@ -144,8 +125,8 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
                 error_trace = traceback.format_exc()
                 logger.error(f"Error in preprocessing UI: {str(e)}\n{error_trace}")
                 
-                # Show error to user
-                st.error(f"Error during image enhancement: {str(e)}")
+                # Show error to user - simple message
+                st.error(f"Error processing image: {str(e)}")
                 result["success"] = False
                 result["message"] = f"Error: {str(e)}"
         
@@ -156,8 +137,8 @@ def show_preprocessing_ui(image_service, img: np.ndarray) -> Dict[str, Any]:
         error_trace = traceback.format_exc()
         logger.error(f"Error in preprocessing UI: {str(e)}\n{error_trace}")
         
-        # Show error to user
-        st.error(f"Error preparing image preprocessing: {str(e)}")
+        # Show error to user - simple message
+        st.error(f"Error processing image")
         
         # Return error information
         return {