Update app.py

app.py

@@ -4,60 +4,104 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-# Function to fetch menu items from Salesforce
+# Fetch menu items from Salesforce
 def fetch_menu():
-    try:
-        query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
-        menu_items = sf.query(query)['records']
-
-        # Format the menu items into a dictionary grouped by section
-        formatted_menu = {}
-        for item in menu_items:
-            section = item['Section__c'] or "Miscellaneous"
-            if section not in formatted_menu:
-                formatted_menu[section] = []
-
-            formatted_menu[section].append({
-                "name": item['Name'],
-                "price": item['Price__c'],
-                "description": item['Description__c'],
-                "image1": item['Image1__c'],
-                "image2": item['Image2__c'],
-                "veg_nonveg": item['Veg_NonVeg__c']
-            })
-
-        return formatted_menu
-    except Exception as e:
-        return f"Error fetching menu from Salesforce: {str(e)}"
-
-# Function to display menu based on user preference
-def display_menu(preference):
-    menu = fetch_menu()
-    if isinstance(menu, str):  # Error handling
-        return menu
-
-    html_output = ""
-    for section, items in menu.items():
-        html_output += f"<h2>{section}</h2>"
-        for item in items:
-            if preference == "All" or item['veg_nonveg'] == preference:
-                html_output += f"<div style='border: 1px solid #ddd; padding: 10px; margin-bottom: 10px;'>"
-                html_output += f"<h3>{item['name']}</h3>"
-                html_output += f"<p>{item['description']}</p>"
-                html_output += f"<p>Price: ${item['price']}</p>"
-                html_output += f"<img src='{item['image1']}' alt='{item['name']}' style='width: 100px; height: 100px;'>"
-                html_output += f"</div>"
-    return html_output
-
-# Gradio App
+    query = """
+    SELECT Name, Description__c, Price__c, Veg_NonVeg__c, Section__c, Image1__c
+    FROM Menu_Item__c
+    """
+    return sf.query_all(query)['records']
+
+# Fetch Add-ons from Salesforce
+def fetch_add_ons():
+    query = """
+    SELECT Name, Price__c, Description__c, Menu_Item__c
+    FROM Add_Ons__c
+    WHERE Available_In_Menu__c = TRUE
+    """
+    return sf.query_all(query)['records']
+
+# Function to generate HTML for the menu with popups
+def generate_menu_html(menu, add_ons):
+    menu_html = ""
+    add_ons_by_item = {}
+
+    # Organize add-ons by menu item
+    for add_on in add_ons:
+        item_id = add_on['Menu_Item__c']
+        if item_id not in add_ons_by_item:
+            add_ons_by_item[item_id] = []
+        add_ons_by_item[item_id].append(add_on)
+
+    # Generate HTML for menu items
+    sections = {}
+    for item in menu:
+        section = item['Section__c'] or 'Others'
+        if section not in sections:
+            sections[section] = []
+
+        item_html = f"""
+        <div class='menu-item'>
+            <img src='{item['Image1__c']}' alt='{item['Name']}' class='menu-item-image'>
+            <h3>{item['Name']}</h3>
+            <p>{item['Description__c']}</p>
+            <p><strong>Price: ${item['Price__c']}</strong></p>
+            <button onclick=\"openPopup('{item['Name']}', '{item['Price__c']}', {add_ons_by_item.get(item['Id'], [])})\">Customize & Add</button>
+        </div>
+        """
+        sections[section].append(item_html)
+
+    # Generate sections
+    for section, items in sections.items():
+        menu_html += f"<h2>{section}</h2><div class='menu-section'>{''.join(items)}</div>"
+
+    return menu_html
+
+# Frontend
+menu_items = fetch_menu()
+add_ons = fetch_add_ons()
+menu_html = generate_menu_html(menu_items, add_ons)
+
+# Gradio app
 with gr.Blocks() as app:
-    with gr.Row():
-        gr.HTML("<h1 style='text-align: center;'>Welcome to Biryani Hub</h1>")
+    gr.HTML("<h1>Welcome to Biryani Hub</h1>")
+    gr.HTML("<div id='menu-container'>" + menu_html + "</div>")
+
+    gr.HTML("""
+    <div id='popup' style='display:none;'>
+        <h2 id='popup-title'></h2>
+        <p id='popup-price'></p>
+        <div id='popup-add-ons'></div>
+        <button onclick="closePopup()">Close</button>
+    </div>
+    """)
+
+    gr.HTML("""
+    <script>
+    function openPopup(itemName, price, addOns) {
+        document.getElementById('popup-title').innerText = itemName;
+        document.getElementById('popup-price').innerText = `Price: $${price}`;
+
+        let addOnsContainer = document.getElementById('popup-add-ons');
+        addOnsContainer.innerHTML = '';
+
+        if (addOns.length > 0) {
+            addOns.forEach(addOn => {
+                let addOnDiv = document.createElement('div');
+                addOnDiv.innerHTML = `<input type='checkbox' value='${addOn.Name}' data-price='${addOn.Price__c}'> ${addOn.Name} ($${addOn.Price__c}) - ${addOn.Description__c}`;
+                addOnsContainer.appendChild(addOnDiv);
+            });
+        } else {
+            addOnsContainer.innerHTML = '<p>No Add-ons available for this item.</p>';
+        }
 
-    with gr.Row():
-        preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
-        menu_display = gr.HTML()
+        document.getElementById('popup').style.display = 'block';
+    }
 
-    preference.change(display_menu, inputs=preference, outputs=menu_display)
+    function closePopup() {
+        document.getElementById('popup').style.display = 'none';
+    }
+    </script>
+    """)
 
-    app.launch()
+app.launch()