Update json and size

app.py

@@ -53,21 +53,6 @@ region = os.environ['region']
 def removeStr(string): 
   return string.replace(" ", "")
 
-def qtyEspecies(datax, datay, resImg):
-  numLobos = 0
-  numPelicanos = 0
-  dfEspecies = pd.DataFrame(datax)
-  datay = str(datay)
-  for i in range(0, dfEspecies['name'].size):
-    if(dfEspecies['name'][i] == 'Lobo marino'):
-      numLobos= numLobos + 1
-    if(dfEspecies['name'][i] == 'Pelicano'):
-      numPelicanos= numPelicanos + 1
-  strlista = '"detail":[{"quantity":"'+str(numLobos)+'","description":"Lobo marino"},{"quantity":"'+str(numPelicanos)+'","description":"Pelicano"}]'
-  data = '{"image":"'+str(removeStr(datay[0:9]))+'","size":"'+str(removeStr(datay[11:18]))+'",'+strlista+','+resImg+'}'
-  json_data = json.loads(data)
-  return json_data
-
 def arrayLista(a,b,c,d):
   x = re.findall("obo Mar", b)
   y = re.findall("elica", b)
@@ -130,6 +115,21 @@ def upload_file(file_name, bucket=None, object_name=None):
     return False
   return stado
 
+def qtyEspecies(datax, datay, resImg):
+  numLobos = 0
+  numPelicanos = 0
+  dfEspecies = pd.DataFrame(datax)
+  datay = str(datay)
+  for i in range(0, dfEspecies['name'].size):
+    if(dfEspecies['name'][i] == 'Lobo marino'):
+      numLobos= numLobos + 1
+    if(dfEspecies['name'][i] == 'Pelicano'):
+      numPelicanos= numPelicanos + 1
+  strlista = '"detail":[{"quantity":"'+str(numLobos)+'","description":"Lobo marino"},{"quantity":"'+str(numPelicanos)+'","description":"Pelicano"}]'
+  data = '{"image":"'+str(removeStr(datay[0:9]))+'","size":"'+str(removeStr(datay[11:18]))+'",'+strlista+','+resImg+'}'
+  json_data = json.loads(data)
+  return json_data
+
 # Model
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True)  # local model  o google colab
 
@@ -156,9 +156,9 @@ def yolo(size, iou, conf, im):
       pil_im.save(b, 'jpeg')
       im_bytes = b.getvalue()
       fileImg = tempFileJSON(im_bytes)
-      lista=qtyEspecies(results5,results2,fileImg)
+      results6 = qtyEspecies(results5,results2,fileImg)
       lista2 = arrayLista(results3[19:21],results3[22:32], results3[34:36], results3[37:45])
-      return Image.fromarray(results2.ims[0]), lista2, lista
+      return Image.fromarray(results2.ims[0]), lista2, results6
     except Exception as e:
       logging.error(e, exc_info=True)