Update app.py

app.py

@@ -1,5 +1,7 @@
 import gradio as gr
-from cart_tab import add_to_cart, display_cart  # Import shared cart functions
+
+# Global variables for cart and menu data
+cart = []
 
 menu_data = [
     {"name": "Veg Burger", "category": "VEGAN",
@@ -16,7 +18,20 @@ menu_data = [
      "price": "$10.99"}
 ]
 
+# Function to add an item to the cart
+def add_to_cart(item_name):
+    cart.append({"item_name": item_name})
+    return f"{item_name} added to cart!"
 
+# Function to display the current cart
+def display_cart():
+    if not cart:
+        return "Your cart is empty."
+    cart_content = "<h3>Your Cart:</h3><ul>"
+    for item in cart:
+        cart_content += f"<li>{item['item_name']}</li>"
+    cart_content += "</ul>"
+    return cart_content
 
 # Function to generate dish cards
 def display_dishes(category):
@@ -55,17 +70,19 @@ popup_js = """
             <button id="add-to-cart-button" style="padding: 10px 20px; background-color: #28a745; color: white; border: none; border-radius: 5px; cursor: pointer;">
                 Add to Cart
             </button>
-            <button onclick="closePopup()" style="padding: 10px 20px; background-color: #dc3545; color: white; border: none; border-radius: 5px; cursor: pointer; margin-left: 10px;">
-                Close
-            </button>
         `;
         popup.style.display = "block";
         overlay.style.display = "block";
         
         document.getElementById("add-to-cart-button").onclick = async function() {
-            const response = await google.colab.kernel.invokeFunction('add_to_cart', [name], {});
-            alert(response.data);
-            closePopup();
+            const response = await fetch('/run/add_to_cart', {
+                method: 'POST',
+                body: JSON.stringify({ name }),
+                headers: { 'Content-Type': 'application/json' }
+            });
+            const result = await response.json();
+            alert(result.data);  // Show success message
+            closePopup();  // Automatically close popup
         };
     }
 
@@ -123,4 +140,7 @@ with gr.Blocks() as demo:
                 outputs=cart_display
             )
 
+# Register the add_to_cart function
+demo.api("/add_to_cart", add_to_cart)
+
 demo.launch()