Update app.py

app.py

@@ -84,7 +84,6 @@ menu_data = [
 
 
 
-# Spice levels and extras
 spice_levels = ["American Mild", "American Medium", "American Spicy", 
                 "Indian Mild", "Indian Medium", "Indian Spicy", "Indian Very Spicy"]
 extras = [
@@ -178,15 +177,20 @@ with gr.Blocks() as demo:
                 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);
+                // Keep current state
+                selectedSpice = document.querySelector('input[name=spice]:checked')?.value || null;
+                selectedExtras = Array.from(document.querySelectorAll('input[name=extras]:checked')).map(input => input.value);
+
+                // Re-select checkboxes and radio buttons
+                if (selectedSpice) {
+                    document.querySelector(`input[name=spice][value="${selectedSpice}"]`).checked = true;
+                }
+                document.querySelectorAll('input[name=extras]').forEach(input => {
+                    if (selectedExtras.includes(input.value)) {
+                        input.checked = true;
+                    }
+                });
 
-                // Display the popup
                 document.getElementById('popup').style.display = 'block';
             }
 
@@ -217,4 +221,4 @@ with gr.Blocks() as demo:
         outputs=dish_display
     )
 
-demo.launch()
+demo.launch()