Update app.py

app.py

@@ -19,11 +19,13 @@ def extract_coordinates(string):
     numbers = list(map(int, numbers))
 
     # Ensuring there are exactly 4 numbers
-    if len(numbers) != 4:
+    if len(numbers) >= 4: #if len(numbers) != 4:
         numbers = numbers[-4:]
 
-    # Extracting coordinates
-    x1, y1, x2, y2 = numbers
+        # Extracting coordinates
+        x1, y1, x2, y2 = numbers
+    else: 
+        return []
 
     return [x1, y1, x2, y2]
 
@@ -99,12 +101,15 @@ with col2:
                 
                 # Get the coordinates from the output text and denormalize them
                 coordinates = extract_coordinates(output_text)
-                coordinates = unnormalize_box(coordinates, unnormalized_image.width, unnormalized_image.height)
-                
-                draw = ImageDraw.Draw(unnormalized_image)
-                draw.rectangle(coordinates, outline="red")
-                
-                st.image(unnormalized_image, caption="Output Image")
+                if coordinates is not None: 
+                    coordinates = unnormalize_box(coordinates, unnormalized_image.width, unnormalized_image.height)
+                    
+                    draw = ImageDraw.Draw(unnormalized_image)
+                    draw.rectangle(coordinates, outline="red")
+                    
+                    st.image(unnormalized_image, caption="Output Image")
+                else: 
+                    st.write("Cannot obtain Bounding Box coordinates: " + output_text)
     elif testButton and selected_file == "None":
         st.write("Please upload and select a document (/an image).")