Update app.py

app.py

@@ -1,9 +1,7 @@
-
 import gradio as gr
 import pandas as pd
 
 # ---------- Inventory ----------
-# ID -> (Name, Price per KG)
 INVENTORY = {
     1: ("Apples", 250),
     2: ("Cherry", 650),
@@ -17,15 +15,10 @@ def inventory_markdown():
                         for pid, (name, price) in INVENTORY.items()])
     return f"### Inventory\n{items}"
 
-# ---------- Cart helpers ----------
 def summarize_cart(cart):
-    """Return (DataFrame, total_cost) for the cart.
-    Cart structure: list of dicts with keys: id, name, unit_price, qty
-    """
     if not cart:
         return pd.DataFrame(columns=["Index","Name","Quantity","Price Per KG"]), 0
-    rows = []
-    total = 0
+    rows, total = [], 0
     for idx, item in enumerate(cart):
         rows.append({
             "Index": idx,
@@ -34,28 +27,25 @@ def summarize_cart(cart):
             "Price Per KG": f"${item['unit_price']:.2f}",
         })
         total += item["qty"] * item["unit_price"]
-    df = pd.DataFrame(rows)
-    return df, total
+    return pd.DataFrame(rows), total
 
 def add_item_to_cart(pid, cart):
     add_msg = ""
     if pid is None:
         return *summarize_cart(cart), add_msg, ""
     if pid not in INVENTORY:
-        add_msg = f":warning: Invalid ID {pid}. Please use one of: {list(INVENTORY.keys())}"
+        add_msg = f":warning: Invalid ID {pid}. Use: {list(INVENTORY.keys())}"
         return *summarize_cart(cart), add_msg, ""
     name, price = INVENTORY[pid]
-    # Check if already in cart -> increment quantity
     for item in cart:
         if item["id"] == pid:
             item["qty"] += 1
             df, total = summarize_cart(cart)
-            add_msg = f">>>>>>>>>> The item '{name}' has been added to the cart. <<<<<<<<<<"
+            add_msg = f">>>>>>>>>>> The item '{name}' has been added to the cart. <<<<<<<<<<"
             return df, total, add_msg, ""
-    # Otherwise append new
     cart.append({"id": pid, "name": name, "unit_price": float(price), "qty": 1})
     df, total = summarize_cart(cart)
-    add_msg = f">>>>>>>>>> The item '{name}' has been added to the cart. <<<<<<<<<<"
+    add_msg = f">>>>>>>>>>> The item '{name}' has been added to the cart. <<<<<<<<<<"
     return df, total, add_msg, ""
 
 def remove_item_from_cart(index, cart):
@@ -75,8 +65,6 @@ def remove_item_from_cart(index, cart):
 
 with gr.Blocks(title="Fruit Shopping Cart System") as demo:
     gr.Markdown("## Fruit Shopping Cart System")
-
-    # App state: the cart
     cart_state = gr.State([])
 
     with gr.Row():
@@ -91,16 +79,14 @@ with gr.Blocks(title="Fruit Shopping Cart System") as demo:
 
     with gr.Row():
         with gr.Column(scale=3):
-            gr.Markdown("### Your Cart\n_Use 0-based index for removal_ :shopping_trolley:")
-            cart_df = gr.Dataframe(headers=["Index","Name","Quantity","Price Per KG"],
-                                   interactive=False,
-                                   wrap=True,
-                                   height=220)
+            gr.Markdown("### Your Cart\n_Use 0-based index for removal_")
+            empty_df = pd.DataFrame(columns=["Index","Name","Quantity","Price Per KG"])
+            cart_df = gr.Dataframe(value=empty_df, interactive=False)
             add_msg_out = gr.Textbox(label="Add Item Message", interactive=False)
             rm_msg_out = gr.Textbox(label="Remove Item Message", interactive=False)
             total_cost_out = gr.Textbox(label="Total Cost", interactive=False)
         with gr.Column(scale=2):
-            inv_md = gr.Markdown(inventory_markdown())
+            gr.Markdown(inventory_markdown())
             gr.Markdown("""
 ### Business Requirements
 1. **Add Products to the Shopping Cart**: Add by ID from the inventory; quantities aggregate automatically.  
@@ -118,17 +104,10 @@ with gr.Blocks(title="Fruit Shopping Cart System") as demo:
         total_str = f"$ {total:,.2f}"
         return df, add_msg, rm_msg, total_str, cart
 
-    add_btn.click(
-        handle_add,
-        inputs=[pid_in, cart_state],
-        outputs=[cart_df, add_msg_out, rm_msg_out, total_cost_out, cart_state]
-    )
-
-    rm_btn.click(
-        handle_remove,
-        inputs=[rm_index_in, cart_state],
-        outputs=[cart_df, add_msg_out, rm_msg_out, total_cost_out, cart_state]
-    )
+    add_btn.click(handle_add, [pid_in, cart_state],
+                  [cart_df, add_msg_out, rm_msg_out, total_cost_out, cart_state])
+    rm_btn.click(handle_remove, [rm_index_in, cart_state],
+                 [cart_df, add_msg_out, rm_msg_out, total_cost_out, cart_state])
 
 if __name__ == "__main__":
     demo.launch()