Update app.py

app.py

@@ -4,48 +4,50 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-
-# Function to fetch menu data from Salesforce
-def fetch_menu_from_salesforce(preference):
+# Function to fetch menu items from Salesforce
+def fetch_menu():
     try:
-        # Querying the Salesforce Menu_Item__c object
-        query = "SELECT Name, Description__c, Price__c, Image1__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
-        menu_items = sf.query(query)["records"]
-        
-        # Filter menu based on preference
-        filtered_menu = [
-            item for item in menu_items 
-            if preference == "All" or item["Veg_NonVeg__c"] == preference
-        ]
-
-        # Group items by section
-        sections = {}
-        for item in filtered_menu:
-            section = item["Section__c"]
-            if section not in sections:
-                sections[section] = []
-            sections[section].append(item)
-
-        # Generate HTML for menu display
-        html = ""
-        for section, items in sections.items():
-            html += f"<h2 style='text-align: left;'>{section}</h2>"
-            for item in items:
-                html += f"""
-                <div style="display: flex; align-items: center; border: 1px solid #ddd; border-radius: 8px; padding: 10px; margin-bottom: 10px;">
-                    <div style="flex: 1;">
-                        <h3 style="margin: 0;">{item['Name']}</h3>
-                        <p style="margin: 5px 0;">{item.get('Description__c', 'No description available')}</p>
-                        <p style="margin: 5px 0;"><b>Price:</b> ${item['Price__c']}</p>
-                    </div>
-                    <div>
-                        <img src="{item.get('Image1__c', '')}" alt="{item['Name']}" style="width: 100px; height: 100px; border-radius: 8px; margin-left: 10px;">
-                    </div>
-                </div>
-                """
-        return html if html else "<p>No menu items available.</p>"
+        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"<p>Error fetching menu: {str(e)}</p>"
+        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
 with gr.Blocks() as app:
@@ -56,14 +58,6 @@ with gr.Blocks() as app:
         preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
         menu_display = gr.HTML()
 
-    # Update the menu display based on preference
-    preference.change(
-        lambda pref: fetch_menu_from_salesforce(pref),
-        inputs=preference,
-        outputs=menu_display
-    )
-
-    # Initial Load
-    menu_display.value = fetch_menu_from_salesforce("All")
+    preference.change(display_menu, inputs=preference, outputs=menu_display)
 
-app.launch()
+    app.launch()