make better error handling

app.py

@@ -65,8 +65,15 @@ async def image_step1(image_file: UploadFile = File(...),background_image: Optio
         image = Image.open(io.BytesIO(contents_img))
         
 
-        output_step1=SegmenterBackground().Back_step1(image,input_to_type_of_filters,int(blur_radius))
-        produced_image=output_step1[0]
+        output_step1 =SegmenterBackground().Back_step1(image,input_to_type_of_filters,int(blur_radius))
+
+        if (output_step1[-1] == 0):
+            return {
+                "Error": output_step1[0],
+                "status": output_step1[-1]
+            }
+        
+        produced_image = output_step1[0]
         
         '''
         # Save the processed image to a temporary file
@@ -83,7 +90,8 @@ async def image_step1(image_file: UploadFile = File(...),background_image: Optio
         # Returning both text and the base64 image
         return {
                 "message": output_step1[1],
-                "image_base64": encoded_img
+                "image_base64": encoded_img,
+                "status": output_step1[-1]
         }
         
 

server.py

@@ -98,6 +98,11 @@ class SegmenterBackground():
         
         masks = [results[0].masks.data[i].cpu().numpy() for i in range(len(results[0].masks.data))]
 
+        if len(masks)==0:
+            return ("No thing in image") ,0 # 0 error happened ,if so it's 2 returns ,first is message
+        
+        #----------> other methods
+        
         ##### put masks on image
         outpt = self.putMaskImage(raw_image,masks,background_image,blur_radius)
         
@@ -123,7 +128,7 @@ class SegmenterBackground():
 
             label_counter.append(current_label)
 
-        return outpt.resize(org_size), label_counter
+        return outpt.resize(org_size), label_counter , 1
 
 
     def Back_step2(self,raw_image:Image,background_image:Image,things_replace:list,blur_radius=23):