app deploy

app.py

@@ -96,7 +96,6 @@ path  = [['image_0.jpg'], ['image_1.jpg'], ['image_2.jpg']]
 video_path = [['video.mp4']]
 
 def show_preds_image(image_path):
-    global angle
     image = cv2.imread(image_path)
     #resize image (optional)
     img_res_toshow = cv2.resize(image, None, fx= 0.5, fy= 0.5, interpolation= cv2.INTER_LINEAR)
@@ -105,6 +104,7 @@ def show_preds_image(image_path):
     dim=(width,height)
 
     outputs = model.predict(source=img_res_toshow)
+    angle_list = []
     for object in range(len(outputs[0].masks.masks)):
         #obtain BW image
         bw=(outputs[0].masks.masks[object].cpu().numpy() * 255).astype("uint8")
@@ -124,8 +124,12 @@ def show_preds_image(image_path):
             cv2.drawContours(img, contours, i, (0, 0, 255), 2)
             
             # Find the orientation of each shape
-            angle = getOrientation(c, img)
-
+            angle_deg = getOrientation(c, img)
+            angle_list.append(angle_deg)
+    text = 'The orientation of the objects are: '
+    for angle in angle_list:
+        text += str(angle) + ' '
+    text += 'degrees'
     results = outputs[0].cpu().numpy()
     for i, det in enumerate(results.boxes.xyxy):
         cv2.rectangle(
@@ -136,13 +140,14 @@ def show_preds_image(image_path):
             thickness=2,
             lineType=cv2.LINE_AA
         )
-    return cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    return cv2.cvtColor(img, cv2.COLOR_BGR2RGB), text
 
 inputs_image = [
     gr.components.Image(type="filepath", label="Input Image"),
 ]
 outputs_image = [
     gr.components.Image(type="numpy", label="Output Image"),
+    gr.outputs.Textbox(label="Orientation Angle")
 ]
 interface_image = gr.Interface(
     fn=show_preds_image,
@@ -153,7 +158,6 @@ interface_image = gr.Interface(
     cache_examples=False,
 )
 
-
 def show_preds_video(video_path):
     cap = cv2.VideoCapture(video_path)
     while(cap.isOpened()):