Update app.py

app.py

@@ -83,6 +83,7 @@ menu_data = [
 # The rest of the code remains unchanged
 
 
+
 # Spice levels and extras
 spice_levels = ["American Mild", "American Medium", "American Spicy", 
                 "Indian Mild", "Indian Medium", "Indian Spicy", "Indian Very Spicy"]
@@ -152,37 +153,52 @@ with gr.Blocks() as demo:
             <p id="popup-price" style="margin-top: 10px; font-size: 16px; color: #444;"></p>
             
             <h4 style="text-align: left; margin-top: 20px;">Choose a spice level</h4>
-            <div style="display: flex; flex-wrap: wrap; justify-content: space-between; border: 1px solid #ddd; padding: 10px; border-radius: 5px;">
+            <div id="spice-levels-container" 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="radio" name="spice" value="{spice}" style="margin-right: 5px;">{spice}</label>' for spice in spice_levels]) + """
             </div>
 
             <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;">
+            <div id="extras-container" 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]) + """
             </div>
 
-            <button onclick="addToCart(document.getElementById('popup-title').textContent, document.querySelector('input[name=spice]:checked').value)" 
+            <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>
             <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>
         <script>
+            let selectedSpice = null;
+            let selectedExtras = [];
+
             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;
 
+                // Reset selections
+                selectedSpice = null;
+                selectedExtras = [];
+                const spiceInputs = document.querySelectorAll('input[name=spice]');
+                spiceInputs.forEach(input => input.checked = false);
+                const extraInputs = document.querySelectorAll('input[name=extras]');
+                extraInputs.forEach(input => input.checked = false);
+
                 // Display the popup
                 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!");
+
+            function addToCart() {
+                selectedSpice = document.querySelector('input[name=spice]:checked')?.value || "No spice selected";
+                selectedExtras = Array.from(document.querySelectorAll('input[name=extras]:checked')).map(input => input.value);
+                alert(`Dish added with spice: ${selectedSpice} and extras: ${selectedExtras.join(", ")}`);
+                closePopup();
             }
         </script>
     """)
@@ -201,4 +217,4 @@ with gr.Blocks() as demo:
         outputs=dish_display
     )
 
-demo.launch()
+demo.launch()