Update app.py

app.py

@@ -13,41 +13,27 @@ transform = assets['transform']
 class_names = assets['class_names']
 
 # Item names and prices
-items = [
-    'Bisconni Chocolate Chip Cookies 46.8gm', 'Coca Cola Can 250ml',
-    'Colgate Maximum Cavity Protection 75gm', 'Fanta 500ml',
-    'Fresher Guava Nectar 500ml', 'Fruita Vitals Red Grapes 200ml',
-    'Islamabad Tea 238gm', 'Kolson Slanty Jalapeno 18gm',
-    'Kurkure Chutney Chaska 62gm', 'LU Candi Biscuit 60gm',
-    'LU Oreo Biscuit 19gm', 'LU Prince Biscuit 55.2gm',
-    'Lays Masala 34gm', 'Lays Wavy Mexican Chili 34gm',
-    'Lifebuoy Total Protect Soap 96gm', 'Lipton Yellow Label Tea 95gm',
-    'Meezan Ultra Rich Tea 190gm', 'Peek Freans Sooper Biscuit 13.2gm',
-    'Safeguard Bar Soap Pure White 175gm', 'Shezan Apple 250ml',
-    'Sunsilk Shampoo Soft - Smooth 160ml', 'Super Crisp BBQ 30gm',
-    'Supreme Tea 95gm', 'Tapal Danedar 95gm',
-    'Vaseline Healthy White Lotion 100ml'
-]
-
-prices = [
-    55.20, 31.75, 90.00, 63.50, 50.00, 35.00, 150.00, 15.00, 25.00,
-    30.00, 10.00, 30.00, 20.00, 20.00, 44.50, 100.00, 200.00, 10.00,
-    70.00, 25.00, 120.00, 15.00, 100.00, 100.00, 120.00
-]
+items = ['Bisconni Chocolate Chip Cookies 46.8gm', 'Coca Cola Can 250ml', 'Colgate Maximum Cavity Protection 75gm',
+         'Fanta 500ml', 'Fresher Guava Nectar 500ml', 'Fruita Vitals Red Grapes 200ml', 'Islamabad Tea 238gm',
+         'Kolson Slanty Jalapeno 18gm', 'Kurkure Chutney Chaska 62gm', 'LU Candi Biscuit 60gm', 'LU Oreo Biscuit 19gm',
+         'LU Prince Biscuit 55.2gm', 'Lays Masala 34gm', 'Lays Wavy Mexican Chili 34gm', 'Lifebuoy Total Protect Soap 96gm',
+         'Lipton Yellow Label Tea 95gm', 'Meezan Ultra Rich Tea 190gm', 'Peek Freans Sooper Biscuit 13.2gm',
+         'Safeguard Bar Soap Pure White 175gm', 'Shezan Apple 250ml', 'Sunsilk Shampoo Soft - Smooth 160ml',
+         'Super Crisp BBQ 30gm', 'Supreme Tea 95gm', 'Tapal Danedar 95gm', 'Vaseline Healthy White Lotion 100ml']
 
-pricelist = dict(zip(items, prices))
+prices = [55.20, 31.75, 90.00, 63.50, 50.00, 35.00, 150.00, 15.00, 25.00, 30.00, 10.00, 30.00, 20.00, 20.00, 44.50, 100.00,
+          200.00, 10.00, 70.00, 25.00, 120.00, 15.00, 100.00, 100.00, 120.00]
 
-# Initialize image list
-image_list = []
+pricelist = dict(zip(items, prices))
 
-# Function to add image to the list
-def add_image(new_img):
+# Add uploaded image to state list
+def add_image(new_img, image_list):
     if new_img:
         image_list.append(new_img)
     return image_list, image_list
 
-# Function to classify images and generate receipt
-def classify_and_track(budget):
+# Classify uploaded images and generate receipt
+def classify_and_track(images, budget):
     results = []
     total_cost = 0
     receipt_lines = []
@@ -58,7 +44,7 @@ def classify_and_track(budget):
     receipt_lines.append("      ITEM                     PRICE")
     receipt_lines.append("      ==============================")
 
-    for img in image_list:
+    for img in images:
         if isinstance(img, Image.Image):
             input_tensor = transform(img).unsqueeze(0)
             with torch.no_grad():
@@ -105,9 +91,12 @@ custom_theme = gr.themes.Base(
 with gr.Blocks(theme=custom_theme) as iface:
     gr.Markdown(
         """
-        #  **ScanCart 🧾**
-        * Upload images of your grocery items, and ScanCart will identify them, calculate the total cost, and check if it fits within your budget.  
-        It's like having a smart shopping assistant! 🛒📸 
+        <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>
+            It's like having a smart shopping assistant! 🛒📸</p>
+        </div>
         """
     )
 
@@ -122,16 +111,17 @@ with gr.Blocks(theme=custom_theme) as iface:
             gallery_output = gr.Gallery(label="Cart Preview", columns=3, height="auto", object_fit="contain")
             receipt_output = gr.Textbox(label="Receipt", lines=20, interactive=False, show_copy_button=True)
 
-    # Define interactions
+    image_list_state = gr.State([])
+
     add_btn.click(
         fn=add_image,
-        inputs=[image_input],
-        outputs=[gallery_output, gallery_output]
+        inputs=[image_input, image_list_state],
+        outputs=[image_list_state, gallery_output]
     )
 
     classify_btn.click(
         fn=classify_and_track,
-        inputs=[budget_input],
+        inputs=[image_list_state, budget_input],
         outputs=[gallery_output, receipt_output]
     )