fix square support

app.py

@@ -28,6 +28,10 @@ def process_images(image,no_mesh_draw=False,square_shape=False,progress=gr.Progr
     progress(0, desc="Start Mediapipe")
     
     boxes,mp_image,face_landmarker_result = mp_box.mediapipe_to_box(image)
+    for i in range(len(boxes)):
+        boxes[i] = mp_box.to_int_box(boxes[i])
+        
+
     if no_mesh_draw:
          annotated_image = image
     else:
@@ -39,22 +43,17 @@ def process_images(image,no_mesh_draw=False,square_shape=False,progress=gr.Progr
     }
     index = 1
     
-    print(boxes)
     if square_shape:
         xy_boxes = boxes[3:]
     else:
         xy_boxes = boxes[:3]
 
-    print(len(xy_boxes))
     for box in xy_boxes:
        label=f"type-{index}"
-       print(box)
-       print(mp_box.xywh_to_xyxy(box))
        annotation_boxes.append([mp_box.xywh_to_xyxy(box),label])
        
        
        jsons[label] = boxes[index-1]
-       print(index)
        index+=1
     
     annotation_boxes.append(([0,0,1,1],"None"))
@@ -124,12 +123,12 @@ with gr.Blocks(css=css, elem_id="demo-container") as demo:
                     
             
 
-    btn.click(fn=process_images, inputs=[image,no_mesh_draw], outputs =[image_out,text_out], api_name='infer')
+    btn.click(fn=process_images, inputs=[image,no_mesh_draw,square_shape], outputs =[image_out,text_out], api_name='infer')
     gr.Examples(
                 examples =["examples/00004200.jpg"],
                 inputs=[image]
     )
     gr.HTML(read_file("demo_footer.html"))
 
-    if __name__ == "__main__":
-        demo.launch()
+if __name__ == "__main__":
+    demo.launch()

mp_box.py

@@ -5,10 +5,15 @@ from mediapipe.framework.formats import landmark_pb2
 from mediapipe import solutions
 import numpy as np
 
+# heavy changed in gradio app
+
 # for X,Y,W,H to x1,y1,x2,y2(Left-top,right-bottom style)
 def xywh_to_xyxy(box):
   return [box[0],box[1],box[0]+box[2],box[1]+box[3]]
 
+def to_int_box(box):
+  return [int(box[0]),int(box[1]),int(box[2]),int(box[3])]
+
 def convert_to_box(face_landmarks_list,indices,w=1024,h=1024):
   x1=w
   y1=h