fixing the issue with bboxes being returned

app/main.py

@@ -24,13 +24,36 @@ class RawResponse(BaseModel):
     result: Any
 
 def serialize_line(line):
-    return {
-        "id": line.id,
-        "bbox": line.bbox,
-        "baseline": line.baseline if hasattr(line, 'baseline') else None,
-        "polygon": line.polygon if hasattr(line, 'polygon') else None,
-        "tags": line.tags,
+    # Create a dictionary with all available attributes
+    line_dict = vars(line)
+    
+    # If 'bbox' is not available but 'polygon' is, calculate bbox from polygon
+    if 'bbox' not in line_dict and 'polygon' in line_dict and line_dict['polygon'] is not None:
+        x_coords, y_coords = zip(*line_dict['polygon'])
+        bbox = [min(x_coords), min(y_coords), max(x_coords), max(y_coords)]
+        line_dict['bbox'] = bbox
+    
+    # Convert numpy arrays to lists for JSON serialization
+    for key, value in line_dict.items():
+        if isinstance(value, np.ndarray):
+            line_dict[key] = value.tolist()
+    
+    return line_dict
+
+def serialize_region(region):
+    # Create a dictionary with known attributes
+    region_dict = {
+        "id": getattr(region, 'id', None),
+        "boundary": getattr(region, 'boundary', None),
+        "tags": getattr(region, 'tags', None),
     }
+    
+    # Convert numpy arrays to lists for JSON serialization
+    for key, value in region_dict.items():
+        if isinstance(value, np.ndarray):
+            region_dict[key] = value.tolist()
+    
+    return region_dict
 
 @app.post("/detect_lines", response_model=RawResponse)
 async def detect_lines(file: UploadFile = File(...)):
@@ -40,26 +63,9 @@ async def detect_lines(file: UploadFile = File(...)):
     # Perform baseline and layout analysis (BLLA) segmentation with default model
     baseline_seg = blla.segment(image)
     
-    def serialize_line(line):
-        # Create a dictionary with all available attributes
-        line_dict = vars(line)
-        
-        # If 'bbox' is not available but 'polygon' is, calculate bbox from polygon
-        if 'bbox' not in line_dict and 'polygon' in line_dict and line_dict['polygon'] is not None:
-            x_coords, y_coords = zip(*line_dict['polygon'])
-            bbox = [min(x_coords), min(y_coords), max(x_coords), max(y_coords)]
-            line_dict['bbox'] = bbox
-        
-        # Convert numpy arrays to lists for JSON serialization
-        for key, value in line_dict.items():
-            if isinstance(value, np.ndarray):
-                line_dict[key] = value.tolist()
-        
-        return line_dict
-    
     serialized_seg = {
         "lines": [serialize_line(line) for line in baseline_seg.lines],
-        "regions": [vars(region) for region in baseline_seg.regions],
+        "regions": [serialize_region(region) for region in baseline_seg.regions],
         "type": baseline_seg.type,
         "text_direction": baseline_seg.text_direction,
         "script_detection": baseline_seg.script_detection,