Update app.py

app.py

@@ -35,7 +35,6 @@ def classify_and_track(images, budget):
     total_cost = 0
     receipt_lines = []
 
-    # Receipt Header with some emojis and stylized spacing
     receipt_lines.append("        🛒  SCANCART SUPERMARKET  🛒")
     receipt_lines.append("          📍 Online Smart Checkout")
     receipt_lines.append("      ==============================")
@@ -53,7 +52,6 @@ def classify_and_track(images, budget):
                 total_cost += price
                 results.append((img, f"{predicted_class} - ₱{price:.2f}"))
 
-                # Limit item name length and add cool formatting
                 item_name = (predicted_class[:22] + '...') if len(predicted_class) > 25 else predicted_class
                 receipt_lines.append(f"      {item_name:<25} ₱{price:>6.2f}")
         else:
@@ -63,61 +61,20 @@ def classify_and_track(images, budget):
     receipt_lines.append(f"      TOTAL:{'':>20}₱{total_cost:>7.2f}")
     receipt_lines.append(f"      BUDGET:{'':>19}₱{budget:>7.2f}")
     receipt_lines.append("      ==============================")
-    
-    # Fun dynamic message depending on budget
     if total_cost <= budget:
-        receipt_lines.append("      ✅ Awesome! You're within budget! 🎉")
+        receipt_lines.append("      ✅ You're within budget! 🎉")
         receipt_lines.append("      🛍️ Happy Shopping!")
     else:
-        receipt_lines.append("      ❌ Uh-oh! You're over budget! 😬")
-        receipt_lines.append("      ⚠️ Try removing some items.")
+        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 soon for smarter buys! 🧾")
+    receipt_lines.append("       THANK YOU FOR SHOPPING WITH US!")
+    receipt_lines.append("      Come back for smarter buys! 🧾")
 
     receipt_text = "\n".join(receipt_lines)
     return results, receipt_text
 
-# Custom Theme: Rose + Amber + Stone + Google Fonts
-custom_theme = gr.themes.Base(
-    primary_hue="rose",
-    secondary_hue="amber",
-    neutral_hue="stone",
-    font=[gr.themes.GoogleFont("Montserrat")]
-).set(
-    body_background_fill="#fff0f0",
-    body_text_color="#4a4a4a",
-    input_background_fill="#fff5f5",
-    button_primary_background_fill="#ff4d6d",
-    button_primary_text_color="#fff",
-    button_secondary_background_fill="#ffb347",
-    button_secondary_text_color="#5a2a00"
-)
-
-# Custom CSS for hover effect and scroll for gallery
-custom_css = """
-.gr-gallery-item img {
-    border-radius: 15px;
-    transition: transform 0.3s ease;
-}
-.gr-gallery-item:hover img {
-    transform: scale(1.1);
-    box-shadow: 0 8px 20px rgba(255,77,109,0.6);
-    cursor: pointer;
-}
-.gr-textbox textarea {
-    font-family: 'Helvetica', Courier, monospace;
-    background-color: #fff8f8;
-    border: 2px solid #ff4d6d;
-    border-radius: 10px;
-    padding: 12px;
-    font-size: 14px;
-    line-height: 1.4;
-}
-"""
-
-# Gradio Interface
+# 🎛️ Gradio Interface
 with gr.Blocks(theme=custom_theme) as iface:
     gr.Markdown(
         """
@@ -127,30 +84,31 @@ with gr.Blocks(theme=custom_theme) as iface:
         """
     )
 
+
+
     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 to Cart")
-            budget_input = gr.Number(label="💸 Set Your Budget (₱)", value=500.0, precision=2)
-            classify_btn = gr.Button("🧾 Generate Receipt & Checkout")
-            budget_input.tooltip = "Set your shopping budget here to get feedback on spending!"
+            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)
+            classify_btn = gr.Button("Generate Receipt 🧾")
 
         with gr.Column(scale=2):
-            gallery_output = gr.Gallery(label="🛍️ Cart Preview", columns=3, height="auto", object_fit="contain")
-            receipt_output = gr.Textbox(label="🧾 Receipt", lines=20, show_copy_button=True, interactive=False)
+            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)
 
     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]
     )
 
     classify_btn.click(
-        fn=classify_and_track,
+        classify_and_track,
         inputs=[image_list_state, budget_input],
         outputs=[gallery_output, receipt_output]
     )
 
-iface.launch()
+demo.launch()