Update app.py

app.py

@@ -868,7 +868,8 @@ def main():
             image = uploaded_image
             col1, col2 = st.columns([1, 2])
             with col1:
-                st.image(image, caption="Uploaded Image", width=300)
+                # Use a consistent width based on the container
+                st.image(image, caption="Uploaded Image", use_column_width=True)
             
             # Continue with Xception model analysis
             if st.session_state.xception_model_loaded:
@@ -918,11 +919,20 @@ def main():
                             
                             # Display face box on image if detected
                             if face_box:
+                                # Create a proper visualization that matches the GradCAM alignment
                                 img_to_show = original_image.copy()
                                 img_draw = np.array(img_to_show)
                                 x, y, w, h = face_box
-                                cv2.rectangle(img_draw, (x, y), (x + w, y + h), (0, 255, 0), 2)
-                                st.image(Image.fromarray(img_draw), caption="Detected Face", width=300)
+                                
+                                # Draw rectangle with thicker line and better color for visibility
+                                cv2.rectangle(img_draw, (x, y), (x + w, y + h), (0, 255, 0), 3)
+                                
+                                # Add a caption inside the image for clarity
+                                font = cv2.FONT_HERSHEY_SIMPLEX
+                                cv2.putText(img_draw, "Detected Face", (x, y-10), font, 0.6, (0, 255, 0), 2)
+                                
+                                # Display full image with box instead of cropped face
+                                st.image(Image.fromarray(img_draw), caption="Face Detection Result", use_column_width=True)
                         
                         # GradCAM visualization with error handling
                         st.subheader("GradCAM Visualization")
@@ -932,8 +942,11 @@ def main():
                             )
                             
                             if comparison:
-                                # Display GradCAM results (controlled size)
-                                st.image(comparison, caption="Original | CAM | Overlay", width=700)
+                                # Add some spacing for better visual separation
+                                st.write("")
+                                
+                                # Display GradCAM results with better sizing and alignment
+                                st.image(comparison, caption="Original | CAM | Overlay", use_column_width=True)
                                 
                                 # Save for later use
                                 st.session_state.comparison_image = comparison