Update app.py

app.py

@@ -29,15 +29,20 @@ def object_classify(img):
   num_boxes = len(bboxes)
   index = 0
 
-  # Draw bounding box for each result
+  # Draw bounding box for each result and count the price
   for i in bboxes:
+    
     if i['label'] == 'apple':
       price_total += 25 #pesos? dunno
     elif i['label'] == 'bottle':
       price_total += 15
     elif i['label'] == 'broccoli':
       price_total += 100
-    
+    elif i['label'] == 'orange':
+      price_total += 20
+    elif i['label'] == 'banana':
+      price_total += 50
+      
     box = i['box']
     #Draw the bounding box
     output_image = draw_bounding_box(img, i['score'],i['label'],
@@ -52,6 +57,6 @@ TITLE = 'Object Detection for Effective Self-Checkout in Grocery Shopping [Work
 DESCRIPTION = 'A deep learning application using DETR model to reimagine self-checkout stores.'
 EXAMPLES = ['ex1.jpg']
 interface=gr.Interface(object_classify,
-                       gr.inputs.Image(type = 'pil'),outputs = [gr.outputs.Image(), gr.outputs.Textbox(label='Total Price: ')],
+                       gr.inputs.Image(type = 'pil'),outputs = [gr.outputs.Image(), gr.outputs.Textbox(label='Total Price: '), gr.outputs.Textbox(label='Total items in cart: ')],
                        examples = EXAMPLES,title = TITLE, description=DESCRIPTION, allow_flagging="never")
 interface.launch()