Update app.py

app.py

@@ -33,6 +33,9 @@ cart = {}
 
 # Function to render the menu
 def display_menu():
+    """
+    Generate the menu as HTML with functional Add buttons tied to Gradio's backend.
+    """
     menu_html = ""
     for item in menu_data:
         if item["name"] in cart:
@@ -60,13 +63,18 @@ def display_menu():
                     <h4 style="margin: 0;">{item['name']} - ${item['price']}</h4>
                     <p style="font-size: 12px; color: gray;">{item['description']}</p>
                 </div>
-                <button id="add-{item['name']}" style="background-color: #28a745; color: white; border: none; padding: 8px 12px; border-radius: 5px; cursor: pointer; margin-left: auto;">Add</button>
+                <button name="add-{item['name']}" style="background-color: #28a745; color: white; border: none; padding: 8px 12px; border-radius: 5px; cursor: pointer; margin-left: auto;">Add</button>
             </div>
             """
     return menu_html
 
+
 # Function to add items to the cart
+# Add to Cart Logic
 def add_to_cart(item_name):
+    """
+    Add an item to the cart and return the updated menu and cart summary.
+    """
     if item_name in cart:
         cart[item_name]["quantity"] += 1
     else:
@@ -78,7 +86,8 @@ def add_to_cart(item_name):
                     "quantity": 1
                 }
                 break
-    return update_cart_summary()
+    return display_menu(), update_cart_summary()
+
 
 # Function to update the cart summary
 def update_cart_summary():
@@ -115,16 +124,19 @@ def display_cart():
 with gr.Blocks() as demo:
     gr.Markdown("# Dynamic Food Menu")
     
-    with gr.Tab("Menu"):
-        menu_html = gr.HTML(value=display_menu())
-        cart_summary = gr.HTML(value="No items added to cart.")
+    menu_html = gr.HTML(value=display_menu())
+    cart_summary = gr.HTML(value="No items added to cart.")
 
-        # Add to Cart buttons
-        item_name_input = gr.Textbox(visible=False)
-        add_button = gr.Button("Add to Cart", visible=False)
-        add_button.click(add_to_cart, inputs=item_name_input, outputs=[menu_html, cart_summary])
+    # Hidden input for item name
+    item_name_input = gr.Textbox(visible=False)
 
-    with gr.Tab("Cart"):
-        cart_html = gr.HTML(value=display_cart())
+    # Add Button Logic
+    add_button = gr.Button("Add to Cart", visible=False)
+    add_button.click(fn=add_to_cart, inputs=item_name_input, outputs=[menu_html, cart_summary])
+
+    # Display Menu and Cart Summary
+    menu_html.render()
+    cart_summary.render()
 
 demo.launch()
+