Update app.py

app.py

@@ -12,6 +12,7 @@ assets = joblib.load("deployment_assets.joblib")
 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',
@@ -25,11 +26,13 @@ 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 image to list
 def add_image(new_img, image_list):
     if new_img:
         image_list.append(new_img)
     return image_list, image_list
 
+# Classify and create receipt
 def classify_and_track(images, budget):
     results = []
     total_cost = 0
@@ -74,7 +77,7 @@ def classify_and_track(images, budget):
     receipt_text = "\n".join(receipt_lines)
     return results, receipt_text
 
-# Custom Theme
+# 🎨 Custom Theme
 custom_theme = gr.themes.Base(
     primary_hue="rose",
     secondary_hue="amber",
@@ -84,12 +87,12 @@ custom_theme = gr.themes.Base(
     body_background_fill="#ffe5e5"
 )
 
-# Gradio Interface
+# 🎛️ Gradio Interface
 with gr.Blocks(theme=custom_theme) as demo:
     gr.Markdown(
         """
-        # 🧾 **ScanCart**
-        Upload an image, and ScanCart finds the items, checks the prices, builds your cart, and lets you know if it fits your budget.  
+        #  **ScanCart 🧾**
+        * Upload an image, and ScanCart finds the items, checks the prices, builds your cart, and lets you know if it fits your budget.  
         Like having a shopping assistant with scanner vision. 🛒📸 
         """
     )
@@ -108,11 +111,15 @@ with gr.Blocks(theme=custom_theme) as demo:
     image_list_state = gr.State([])
 
     add_btn.click(
-        add_image,
+        fn=add_image,
         inputs=[image_input, image_list_state],
         outputs=[image_list_state, gallery_output]
     )
 
     classify_btn.click(
-        classify_and_track,
-        inputs=[image_list_state]()
+        fn=classify_and_track,
+        inputs=[image_list_state, budget_input],
+        outputs=[gallery_output, receipt_output]
+    )
+
+demo.launch()