Update app.py

app.py

@@ -1,4 +1,6 @@
 import gradio as gr
+from cart import Cart  # Import Cart from cart.py
+
 
 # Menu data
 menu_data = [
@@ -20,15 +22,10 @@ menu_data = [
     {"name": "Chicken Manchurian", "category": "HALAL", "image": "https://huggingface.co/spaces/Rammohan0504/First-Project/resolve/main/pictures/Chicken_manchurian.jpg", "description": "A popular Indo-Chinese dish featuring fried chicken pieces tossed in a tangy and spicy sauce.", "price": 13.99},
 ]
 
-extras = [
-    {"name": "Extra Raitha 4oz", "price": "$1.00"},
-    {"name": "Extra Raitha 8oz", "price": "$2.00"},
-    {"name": "Extra Salan 4oz", "price": "$1.00"},
-    {"name": "Extra Salan 8oz", "price": "$2.00"},
-    {"name": "Extra Onion", "price": "$1.00"},
-    {"name": "Extra Onion & Lemon", "price": "$2.00"},
-    {"name": "Extra Fried Onion 4oz", "price": "$2.00"},
-]
+
+
+# Initialize the cart
+cart = Cart()
 
 # Filter dishes by category
 def filter_dishes(category):
@@ -36,100 +33,90 @@ def filter_dishes(category):
         return menu_data
     return [dish for dish in menu_data if dish["category"] == category]
 
-# Generate dish cards with "Add" button
-def display_dishes(category):
-    filtered_dishes = filter_dishes(category)
-    html_content = "<div style='display: flex; flex-direction: column; align-items: center; width: 100%; height:100%'>"
-    for dish in filtered_dishes:
-        html_content += f"""
-        <div style='margin: 10px; display: flex; align-items: center; justify-content: space-between; 
-                    width: 80%; padding: 10px; border: 1px solid #ddd; border-radius: 10px; 
-                    background-color: #f9f9f9;'>
-            <div style='flex: 1; margin-right: 20px; text-align: left;'>
-                <h4 style='font-size: 18px; color: #444; margin: 0;'>{dish['name']}</h4>
-                <p style='font-size: 14px; color: #666; margin: 5px 0;'>{dish['description']}</p>
-                <p style='font-size: 16px; color: #444; margin: 0;'>Price: {dish['price']}</p>
-            </div>
-            <div style='text-align: center;'>
-                <img src='{dish['image']}' alt='{dish['name']}' 
-                     style='width: 150px; height: 100px; object-fit: cover; border-radius: 10px;'>
-                <button onclick="showPopup('{dish['image']}', '{dish['name']}', '{dish['description']}', '{dish['price']}')" 
-                        style="margin-top: 15px; padding: 10px 20px; background-color: #28a745; color: white; border: none; 
-                               border-radius: 5px; cursor: pointer;">Add</button>
-            </div>
+# Add item to the cart
+def add_to_cart(item_name):
+    item = next((dish for dish in menu_data if dish["name"] == item_name), None)
+    if item:
+        message = cart.add_item(item)
+        return message, cart.view_cart(), f"Total: ${cart.total_price():.2f}"
+    return "Item not found!", cart.view_cart(), f"Total: ${cart.total_price():.2f}"
+
+# Generate pop-up content for a selected dish
+def generate_popup(dish_name):
+    item = next((dish for dish in menu_data if dish["name"] == dish_name), None)
+    if item:
+        popup_html = f"""
+        <div style="text-align: center;">
+            <img src="{item['image']}" alt="{item['name']}" style="width: 300px; height: 200px; object-fit: cover; border-radius: 10px;">
+            <h2 style="color: #444;">{item['name']}</h2>
+            <p style="font-size: 16px; color: #666;">{item['description']}</p>
+            <p style="font-size: 18px; color: #444;">Price: ${item['price']}</p>
         </div>
         """
-    html_content += "</div>"
-    return html_content
+        return popup_html
+    return "Dish not found!"
 
 # Main Gradio App
 with gr.Blocks() as demo:
-    gr.HTML("<h1 style='text-align: center; color: #333;'>🍛 Biryani Hub Menu 🍛</h1>")
+    gr.HTML("<h1 style='text-align: center;'>🍛 Biryani Hub Menu 🍛</h1>")
 
-    # Buttons for category selection
+    # Category buttons
     with gr.Row():
         btn_all = gr.Button("ALL")
         btn_vegan = gr.Button("VEGAN")
         btn_halal = gr.Button("HALAL")
 
-    # Dish display
-    dish_display = gr.HTML(value=display_dishes("ALL"))
+    # Dish display and cart
+    dish_display = gr.HTML()
+    popup_display = gr.HTML()
+    cart_display = gr.HTML("Cart is empty.")
+    cart_total = gr.Textbox(value="Total: $0.00", interactive=False)
 
-    # JavaScript for managing the pop-up
-    gr.HTML(f"""
-        <div id="popup" style="display: none; position: fixed; top: 50%; left: 50%; transform: translate(-50%, -50%); 
-                              background-color: white; padding: 20px; border-radius: 10px; 
-                              box-shadow: 0 4px 8px rgba(0, 0, 0, 0.2); z-index: 1000; text-align: center; width: 500px;">
-            <img id="popup-image" src="" alt="" style="width: 100%; height: auto; border-radius: 10px;">
-            <h2 id="popup-title" style="margin-top: 20px; color: #444;"></h2>
-            <p id="popup-description" style="margin-top: 10px; font-size: 16px; color: #666;"></p>
-            <p id="popup-price" style="margin-top: 10px; font-size: 16px; color: #444;"></p>
-            
-            
-
-            <h4 style="text-align: left; margin-top: 20px;">Choose Extras</h4>
-            <div style="display: flex; flex-wrap: wrap; justify-content: space-between; border: 1px solid #ddd; padding: 10px; border-radius: 5px;">
-    """ + "".join([f'<label style="margin-right: 10px; display: block;"><input type="checkbox" name="extras" value="{extra["name"]}" style="margin-right: 5px;">{extra["name"]} + {extra["price"]}</label>' for extra in extras]) + """
+    # Event to update dishes based on category
+    def display_dishes(category):
+        filtered_dishes = filter_dishes(category)
+        html_content = "<div style='display: flex; flex-wrap: wrap; gap: 20px; justify-content: center;'>"
+        for dish in filtered_dishes:
+            html_content += f"""
+            <div style="width: 200px; border: 1px solid #ddd; border-radius: 10px; padding: 10px; text-align: center; background-color: #f9f9f9;">
+                <img src="{dish['image']}" alt="{dish['name']}" style="width: 100%; height: 150px; object-fit: cover; border-radius: 10px;">
+                <h4 style="margin: 10px 0; color: #444;">{dish['name']}</h4>
+                <button onclick="showPopup('{dish['name']}')" style="padding: 10px 20px; background-color: #28a745; color: white; border: none; border-radius: 5px; cursor: pointer;">View</button>
             </div>
+            """
+        html_content += "</div>"
+        return html_content
 
-            <button onclick="addToCart(document.getElementById('popup-title').textContent, document.querySelector('input[name=spice]:checked').value)" 
-                    style="margin-top: 20px; padding: 10px 20px; background-color: #007bff; color: white; border: none; 
-                           border-radius: 5px; cursor: pointer;">Add to Cart</button>
-            <button onclick="closePopup()" style="margin-top: 20px; padding: 10px 20px; background-color: #28a745; 
-                                                  color: white; border: none; border-radius: 5px; cursor: pointer;">Close</button>
-        </div>
+    # JavaScript for managing the pop-up
+    gr.HTML("""
         <script>
-            function showPopup(image, title, description, price) {
-                document.getElementById('popup-image').src = image;
-                document.getElementById('popup-title').textContent = title;
-                document.getElementById('popup-description').textContent = description;
-                document.getElementById('popup-price').textContent = "Price: " + price;
-
-                // Display the popup
-                document.getElementById('popup').style.display = 'block';
+            function showPopup(dishName) {
+                fetch("/generate_popup", {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify({ name: dishName })
+                })
+                .then(response => response.json())
+                .then(data => {
+                    document.getElementById("popup").innerHTML = data.html;
+                    document.getElementById("popup").style.display = "block";
+                });
             }
             function closePopup() {
-                document.getElementById('popup').style.display = 'none';
-            }
-            function addToCart(dishName, spiceLevel) {
-                const selectedExtras = Array.from(document.querySelectorAll('input[name=extras]:checked')).map(cb => cb.value).join(", ");
-                alert(dishName + " with " + spiceLevel + " spice level and extras: " + selectedExtras + " has been added to the cart!");
+                document.getElementById("popup").style.display = "none";
             }
         </script>
     """)
 
-    # Button click events
-    btn_all.click(
-        lambda: gr.update(value=display_dishes("ALL")),
-        outputs=dish_display
-    )
-    btn_vegan.click(
-        lambda: gr.update(value=display_dishes("VEGAN")),
-        outputs=dish_display
-    )
-    btn_halal.click(
-        lambda: gr.update(value=display_dishes("HALAL")),
-        outputs=dish_display
-    )
-
-demo.launch()
+    # Dish display logic
+    btn_all.click(lambda: display_dishes("ALL"), outputs=dish_display)
+    btn_vegan.click(lambda: display_dishes("VEGAN"), outputs=dish_display)
+    btn_halal.click(lambda: display_dishes("HALAL"), outputs=dish_display)
+
+    # Cart display
+    gr.Button("View Cart").click(lambda: cart.view_cart(), outputs=cart_display)
+
+    # Placeholder for pop-up
+    gr.HTML(id="popup", value="", visible=False)
+
+demo.launch()