added total result

app.py

@@ -22,7 +22,7 @@ class BinPacking:
         self.width = width
         self.height = height
         self.pgrp = None
-        print(image.shape)
+        self.total = None
         self.image = cv2.resize(image, (int(image.shape[1]*imageScale), int(image.shape[0]*imageScale)))
         if self.image.shape[2] == 4:
             x,y = np.where(self.image[:,:,3]==0)
@@ -57,7 +57,7 @@ class BinPacking:
 
         max_item = int(self.width*self.height/item.area)
         self.pgrp = nest([item,]*max_item, self.box)
-
+        self.total = len(self.pgrp[0])
         return self
 
     def visualize(self):
@@ -110,7 +110,7 @@ def process(width:int, height:int, scale:float, image: np.ndarray):
 
     packer.pack()
     figure = packer.visualize()
-    return figure
+    return [figure, packer.total]
 
 
 if __name__ == "__main__":
@@ -132,7 +132,8 @@ if __name__ == "__main__":
                 fit = gr.Button("Submit")
                 
             plot_output = gr.Plot()
+            total = gr.Number(label="Total")
 
-        fit.click(fn=process, inputs=[width, height, scale, image], outputs=plot_output)
+        fit.click(fn=process, inputs=[width, height, scale, image], outputs=[plot_output, total])
 
     demo.launch()