Update app.py

app.py

@@ -4,7 +4,6 @@ from PIL import Image, ImageDraw, ImageFont
 import json
 from paddleocr import PaddleOCR
 import gradio as gr
-import os
 
 # Initialize PaddleOCR
 ocr = PaddleOCR(use_angle_cls=True, lang='en')
@@ -27,8 +26,11 @@ def draw_boxes_on_image(image, data):
     for item in data:
         bounding_box, (text, confidence) = item
 
-        # Convert bounding box to integer
-        box = np.array(bounding_box).astype(int)
+        # Ensure bounding_box is a list of lists
+        if isinstance(bounding_box[0], list):
+            box = np.array(bounding_box).astype(int)
+        else:
+            box = np.array([bounding_box]).astype(int)
 
         # Draw the bounding box
         draw.line([tuple(box[0]), tuple(box[1])], fill="green", width=2)
@@ -50,6 +52,11 @@ def save_results_to_json(ocr_results):
         for word_info in line:
             bounding_box = word_info[0]
             text, confidence = word_info[1]
+
+            # Ensure bounding_box is a list of lists
+            if not isinstance(bounding_box[0], list):
+                bounding_box = [bounding_box]
+
             results.append({
                 "bounding_box": [list(map(float, coord)) for coord in bounding_box],
                 "text": text,