Update app.py

app.py

@@ -69,7 +69,7 @@ class DicomAnalyzer:
                 dtype=cv2.CV_8U
             )
             if len(normalized.shape) == 2:
-                normalized = cv2.cvtColor(normalized, cv2.COLOR_GRAY2BGR)
+                normalized = cv2.cvtColor(normalized, cv2.COLOR_GRAY2RGB)
             return normalized
         except Exception as e:
             print(f"Error normalizing image: {str(e)}")
@@ -171,63 +171,69 @@ class DicomAnalyzer:
             return self.display_image, f"Error analyzing ROI: {str(e)}"
 
     def update_display(self):
-        try:
-            if self.original_display is None:
-                return None
-
-            height, width = self.original_display.shape[:2]
-            new_height = int(height * self.zoom_factor)
-            new_width = int(width * self.zoom_factor)
-
-            # Create zoomed image
-            zoomed = cv2.resize(self.original_display, (new_width, new_height), 
-                              interpolation=cv2.INTER_CUBIC)
-
-            # Draw marks
-            for x, y, diameter in self.marks:
-                zoomed_x = int(x * self.zoom_factor)
-                zoomed_y = int(y * self.zoom_factor)
-                zoomed_diameter = int(diameter * self.zoom_factor)
-                
-                # Draw main circle
-                cv2.circle(zoomed,
-                          (zoomed_x, zoomed_y),
-                          zoomed_diameter // 2,
-                          (0, 255, 255),  # BGR: Yellow
-                          1,
-                          lineType=cv2.LINE_AA)
-                
-                # Draw dots on circle
-                num_points = 8
-                for i in range(num_points):
-                    angle = 2 * np.pi * i / num_points
-                    point_x = int(zoomed_x + (zoomed_diameter/2) * np.cos(angle))
-                    point_y = int(zoomed_y + (zoomed_diameter/2) * np.sin(angle))
-                    cv2.circle(zoomed,
-                              (point_x, point_y),
-                              1,
-                              (0, 255, 255),  # BGR: Yellow
-                              -1,
-                              lineType=cv2.LINE_AA)
-
-            # Calculate pan limits
-            self.max_pan_x = max(0, new_width - width)
-            self.max_pan_y = max(0, new_height - height)
+    try:
+        if self.original_display is None:
+            return None
+
+        height, width = self.original_display.shape[:2]
+        new_height = int(height * self.zoom_factor)
+        new_width = int(width * self.zoom_factor)
+
+        # Create zoomed image
+        zoomed = cv2.resize(self.original_display, (new_width, new_height), 
+                          interpolation=cv2.INTER_CUBIC)
+
+        # Convert to BGR for drawing
+        zoomed_bgr = cv2.cvtColor(zoomed, cv2.COLOR_RGB2BGR)
+
+        # Draw marks
+        for x, y, diameter in self.marks:
+            zoomed_x = int(x * self.zoom_factor)
+            zoomed_y = int(y * self.zoom_factor)
+            zoomed_diameter = int(diameter * self.zoom_factor)
             
-            # Ensure pan values are within bounds
-            self.pan_x = min(max(0, self.pan_x), self.max_pan_x)
-            self.pan_y = min(max(0, self.pan_y), self.max_pan_y)
+            # Draw main circle in BGR color space
+            cv2.circle(zoomed_bgr,
+                      (zoomed_x, zoomed_y),
+                      zoomed_diameter // 2,
+                      (0, 255, 255),  # BGR: Yellow
+                      1,
+                      lineType=cv2.LINE_AA)
             
-            # Extract visible portion
-            visible = zoomed[
-                int(self.pan_y):int(self.pan_y + height),
-                int(self.pan_x):int(self.pan_x + width)
-            ]
+            # Draw dots on circle
+            num_points = 8
+            for i in range(num_points):
+                angle = 2 * np.pi * i / num_points
+                point_x = int(zoomed_x + (zoomed_diameter/2) * np.cos(angle))
+                point_y = int(zoomed_y + (zoomed_diameter/2) * np.sin(angle))
+                cv2.circle(zoomed_bgr,
+                          (point_x, point_y),
+                          1,
+                          (0, 255, 255),  # BGR: Yellow
+                          -1,
+                          lineType=cv2.LINE_AA)
 
-            return visible
-        except Exception as e:
-            print(f"Error updating display: {str(e)}")
-            return self.original_display
+        # Convert back to RGB for display
+        zoomed = cv2.cvtColor(zoomed_bgr, cv2.COLOR_BGR2RGB)
+
+        # Calculate pan limits
+        self.max_pan_x = max(0, new_width - width)
+        self.max_pan_y = max(0, new_height - height)
+        
+        # Ensure pan values are within bounds
+        self.pan_x = min(max(0, self.pan_x), self.max_pan_x)
+        self.pan_y = min(max(0, self.pan_y), self.max_pan_y)
+        
+        # Extract visible portion
+        visible = zoomed[
+            int(self.pan_y):int(self.pan_y + height),
+            int(self.pan_x):int(self.pan_x + width)
+        ]
+
+        return visible
+    except Exception as e:
+        print(f"Error updating display: {str(e)}")
+        return self.original_display
 
     def format_results(self):
         if not self.results: