Update utils/helpers.py

utils/helpers.py

@@ -235,39 +235,44 @@ def draw_boxes(image: Any, predictions: List[Dict[str, Any]]) -> Any:
     return image
 
 
-def draw_bounding_boxes_for_textract(image, json_data):
-    # Load the image
-    # image: image = Image.open(image)
-    fig, ax = plt.subplots(1)
-    ax.imshow(image)
+def draw_bounding_boxes_for_textract(image: Image.Image, json_data: str) -> Image.Image:
+    """
+    Draws bounding boxes on an image based on the provided JSON data from Textract.
+
+    Args:
+        image_path: The path to the image on which to draw bounding boxes.
+        json_data: The JSON string containing the bounding box data from Textract.
+
+    Returns:
+        A PIL Image object with bounding boxes drawn.
+    """
+    # Load the image from the provided path
+    draw = ImageDraw.Draw(image)
     
     # Parse the JSON data
-    data = json.loads(json_data)
-    
-    # Check if 'body' key exists and load its content
-    if 'body' in data:
-        blocks = json.loads(data['body'])
-    else:
+    try:
+        data = json.loads(json_data)
+        blocks = json.loads(data['body']) if 'body' in data else None
+    except json.JSONDecodeError:
         st.error('Invalid JSON data.')
-        return
+        return image
+    
+    if blocks is None:
+        st.error('No bounding box data found.')
+        return image
     
-    # Iterate through the elements to find bounding boxes
+    # Iterate through the elements to find bounding boxes and draw them
     for item in blocks:
         if 'BlockType' in item and item['BlockType'] in ['LINE', 'WORD']:
             bbox = item['Geometry']['BoundingBox']
             # Extract coordinates and dimensions
             left, top, width, height = bbox['Left'], bbox['Top'], bbox['Width'], bbox['Height']
-            # Convert to image coordinates (may need adjustment based on your image dimensions)
-            rect = patches.Rectangle((left * image.width, top * image.height), width * image.width, height * image.height, linewidth=1, edgecolor='r', facecolor='none')
-            ax.add_patch(rect)
+            # Calculate bounding box coordinates in image space
+            left_top = (left * image.width, top * image.height)
+            right_bottom = ((left + width) * image.width, (top + height) * image.height)
+            # Draw rectangle
+            draw.rectangle([left_top, right_bottom], outline='red', width=2)
     
-    # Hide axes
-    ax.axis('off')
-    # Save the annotated image to a temporary file
-    temp_file = "temp_annotated_image.png"
-    plt.savefig(temp_file, bbox_inches='tight', pad_inches=0)
-    plt.close()
-    
-    # Display the annotated image in Streamlit
-    st.image(temp_file)
+    return image
+