Update app.py

app.py

@@ -64,89 +64,74 @@ def display_dishes(category):
     html_content += "</div>"
     return html_content
 
+def add_to_cart(item_name, item_price):
+    cart_items.append({"name": item_name, "price": item_price})
+    return f"{item_name} added to cart!"
+
+# Display cart items
+def display_cart():
+    if not cart_items:
+        return "<p>Your cart is empty.</p>"
+    html_content = "<ul>"
+    for item in cart_items:
+        html_content += f"<li>{item['name']} - ${item['price']}</li>"
+    html_content += "</ul>"
+    return html_content
+
 # Main Gradio App
 with gr.Blocks() as demo:
     gr.HTML("<h1 style='text-align: center; color: #333;'>🍛 Biryani Hub Menu 🍛</h1>")
 
-    # CSS for small buttons
-    gr.HTML("""
-        <style>
-            #small-btn {
-                padding: 5px 10px;
-                font-size: 14px;
-                border-radius: 5px;
-                margin: 5px;
-            }
-        </style>
-    """)
-
-    # Buttons for category selection
+    # Header with category buttons and cart icon
     with gr.Row():
-        btn_all = gr.Button("ALL", elem_id="small-btn")
-        btn_vegan = gr.Button("VEGAN", elem_id="small-btn")
-        btn_halal = gr.Button("HALAL", elem_id="small-btn")
+        with gr.Column(scale=6, min_width=300):
+            gr.HTML("""
+                <div style="text-align: left;">
+                    <button id="btn-all" onclick="filterDishes('ALL')" style="padding: 5px 10px; margin: 5px;">ALL</button>
+                    <button id="btn-vegan" onclick="filterDishes('VEGAN')" style="padding: 5px 10px; margin: 5px;">VEGAN</button>
+                    <button id="btn-halal" onclick="filterDishes('HALAL')" style="padding: 5px 10px; margin: 5px;">HALAL</button>
+                </div>
+            """)
+        with gr.Column(scale=1, min_width=50):
+            gr.HTML("""
+                <div style="text-align: right;">
+                    <button onclick="showCart()" style="background: none; border: none; cursor: pointer;">
+                        🛒 Cart
+                    </button>
+                </div>
+            """)
 
     # Dish display
     dish_display = gr.HTML(value=display_dishes("ALL"))
 
-    # 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;">
-        <span onclick="closePopup()" style="position: absolute; top: 10px; right: 10px; cursor: pointer; font-size: 20px;">&times;</span>
-
-        <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;">
-            <label style="margin-right: 10px;"><input type="checkbox" name="extras" value="Extra Raitha"> Extra Raitha</label>
-            <label style="margin-right: 10px;"><input type="checkbox" name="extras" value="Extra Salan"> Extra Salan</label>
-        </div>
+    # Cart display
+    cart_display = gr.HTML(value="<p>Your cart is empty.</p>")
+
+    # JavaScript functions
+    gr.HTML("""
+        <script>
+            function filterDishes(category) {
+                fetch(`/filter_dishes?category=${category}`)
+                    .then(response => response.json())
+                    .then(data => {
+                        document.querySelector('#dish-display').innerHTML = data.html;
+                    });
+            }
+
+            function addToCart(name, price) {
+                fetch(`/add_to_cart?item_name=${name}&item_price=${price}`)
+                    .then(response => response.text())
+                    .then(alert);
+            }
+
+            function showCart() {
+                fetch(`/display_cart`)
+                    .then(response => response.text())
+                    .then(data => {
+                        document.querySelector('#cart-display').innerHTML = data;
+                    });
+            }
+        </script>
+    """)
 
-        <button onclick="addToCart()" 
-                style="margin-top: 20px; padding: 10px 20px; background-color: #007bff; color: white; border: none; 
-                       border-radius: 5px; cursor: pointer;">Add to Cart</button>
-    </div>
-    <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 closePopup() {{
-            document.getElementById('popup').style.display = 'none';
-        }}
-        function addToCart() {{
-            const extras = Array.from(document.querySelectorAll('input[name=extras]:checked'))
-                                .map(extra => extra.value)
-                                .join(", ");
-            alert("Item added to cart with extras: " + extras);
-            closePopup();
-        }}
-    </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()
+demo.launch()