Update app.py

app.py

@@ -26,13 +26,17 @@ prices = [55.20, 31.75, 90.00, 63.50, 50.00, 35.00, 150.00, 15.00, 25.00, 30.00,
 
 pricelist = dict(zip(items, prices))
 
-# Add uploaded image to state list
+# Functions
 def add_image(new_img, image_list):
     if new_img:
         image_list.append(new_img)
     return image_list, image_list
 
-# Classify uploaded images and generate receipt
+def remove_last_image(image_list):
+    if image_list:
+        image_list.pop()
+    return image_list, image_list
+
 def classify_and_track(images, budget):
     results = []
     total_cost = 0
@@ -60,16 +64,17 @@ def classify_and_track(images, budget):
         else:
             results.append((None, "Invalid image"))
 
+    remaining = budget - total_cost
     receipt_lines.append("      ==============================")
     receipt_lines.append(f"      TOTAL:{'':>20}₱{total_cost:>7.2f}")
     receipt_lines.append(f"      BUDGET:{'':>19}₱{budget:>7.2f}")
+    receipt_lines.append(f"      REMAINING:{'':>16}₱{remaining:>7.2f}")
     receipt_lines.append("      ==============================")
     if total_cost <= budget:
         receipt_lines.append("      ✅ You're within budget! 🎉")
         receipt_lines.append("      🛍️ Happy Shopping!")
     else:
         receipt_lines.append("      ❌ Over budget! Try removing items 😬")
-
     receipt_lines.append("      ==============================")
     receipt_lines.append("       THANK YOU FOR SHOPPING WITH US!")
     receipt_lines.append("      Come back for smarter buys! 🧾")
@@ -91,20 +96,21 @@ custom_theme = gr.themes.Base(
 with gr.Blocks(theme=custom_theme) as iface:
     gr.Markdown(
         """
-        <div style="text-align:center">
-            <h1>🧾 ScanCart 🧾</h1>
-            <p>Upload images of your grocery items, and ScanCart will identify them,<br>
-            calculate the total cost, and check if it fits within your budget.<br><br>
+        <div style="text-align:center;">
+            <h1 style="font-size: 3em;">ScanCart 🧾</h1>
+            <p style="font-size: 1.4em;">Upload images of your grocery items, and ScanCart will identify them, calculate the total cost, and check if it fits within your budget.<br>
             It's like having a smart shopping assistant! 🛒📸</p>
         </div>
-        """
+        """,
+        elem_id="title"
     )
 
     with gr.Row():
         with gr.Column(scale=1):
             image_input = gr.Image(type="pil", label="Upload Item Image", height=224)
             add_btn = gr.Button("Add Image ➕")
-            budget_input = gr.Number(label="Your Budget (₱)", value=500, precision=2)
+            remove_btn = gr.Button("Remove Last Item 🗑️")
+            budget_input = gr.Number(label="Your Budget (₱)", value=500.0, precision=2)
             classify_btn = gr.Button("Generate Receipt 🧾")
 
         with gr.Column(scale=2):
@@ -114,13 +120,19 @@ with gr.Blocks(theme=custom_theme) as iface:
     image_list_state = gr.State([])
 
     add_btn.click(
-        fn=add_image,
+        add_image,
         inputs=[image_input, image_list_state],
         outputs=[image_list_state, gallery_output]
     )
 
+    remove_btn.click(
+        remove_last_image,
+        inputs=image_list_state,
+        outputs=[image_list_state, gallery_output]
+    )
+
     classify_btn.click(
-        fn=classify_and_track,
+        classify_and_track,
         inputs=[image_list_state, budget_input],
         outputs=[gallery_output, receipt_output]
     )