Modify arrayLista and JSON

app.py

@@ -49,26 +49,30 @@ torch.hub.download_url_to_file('https://i.pinimg.com/564x/3b/2f/d4/3b2fd4b6881b6
 model = torch.hub.load('ultralytics/yolov5', 'custom', path='best.pt', force_reload=True, autoshape=True)  # local model  o google colab
 #model = torch.hub.load('path/to/yolov5', 'custom', path='/content/yolov56.pt', source='local')  # local repo
 
-def listJSON(a,b,c,d):
-      strlist =[] 
-      strlist.append(a)
-      strlist.append(b)
-      strlist.append(c) 
+def listJSON(a,b,c,d,e,f):
+      #for n in 
       if d =='Pelicano\nSp':
-        strlist.append('Pelicano')
+        d = 'Pelicano'
+      if f!='Pelicano':
+        strlista = '"detail":[{"quantity":"'+str(c)+'","description":"'+str(d)+'"}]'
       else:
-        strlist.append(d)
-      json_string = json.dumps(strlist)
+        strlista = '"detail":[{"quantity":"'+str(c)+'","description":"'+str(d)+'"},{"quantity":"'+str(e)+'","description":"'+str(f)+'"}]'
+      strlist = '{"image":"'+str(a)+'","size":"'+str(b)+'",'+strlista+'}'
+      json_string = json.loads(strlist)
       return json_string
 
-def arrayLista(a,b):
+def arrayLista(a,b,c,d):
       strlist =[]
+      strlist2 =[]
       strlist.append(a)
       if b =='Pelicano\nSp':
         strlist.append('Pelicano')
       else:
         strlist.append(b)
-      strlista = [strlist]
+      if d=='Pelicano':
+        strlist2.append(c)
+        strlist2.append(d)
+      strlista = [strlist,strlist2]
       df = pd.DataFrame(strlista,columns=['Cantidad','Especie'])
       return df
 
@@ -85,8 +89,8 @@ def yolo(size, iou, conf, im):
         results2 = model(im)  # inference
         results2.render()  # updates results.imgs with boxes and labels
         results3 = str(results2)
-        lista = listJSON(results3[0:9], results3[11:18] ,results3[19:20],results3[21:32])
-        lista2 = arrayLista(results3[19:20],results3[21:32])
+        lista = listJSON(results3[0:9], results3[11:18] ,results3[19:20],results3[21:32], results3[35:36], results3[37:45])
+        lista2 = arrayLista(results3[19:20],results3[21:32], results3[35:36], results3[37:45])
         return Image.fromarray(results2.ims[0]), lista2, lista
     except Exception as e:
         logging.error(e, exc_info=True)