Create diagram_extraction.py

diagram_extraction.py

@@ -0,0 +1,11 @@
+import cv2
+
+def extract_diagram(image):
+    """Extracts the diagram within the largest contour (border)."""
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    edges = cv2.Canny(gray, 50, 150)
+    contours, _ = cv2.findContours(edges, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
+
+    largest_contour = max(contours, key=cv2.contourArea)
+    x, y, w, h = cv2.boundingRect(largest_contour)
+    return image[y:y+h, x:x+w]