Update app.py

app.py

@@ -2,52 +2,49 @@ import gradio as gr
 from simple_salesforce import Salesforce
 
 # Salesforce Connection
-sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
+sf = Salesforce(username='your_username', password='your_password', security_token='your_security_token')
 
-# Function to fetch menu items from Salesforce
-def fetch_menu():
+# Function to fetch menu data from Salesforce
+def fetch_menu_from_salesforce(preference):
     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
+        # 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>"
     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
+        return f"<p>Error fetching menu: {str(e)}</p>"
 
 # Gradio App
 with gr.Blocks() as app:
@@ -58,6 +55,14 @@ with gr.Blocks() as app:
         preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], label="Filter Preference", value="All")
         menu_display = gr.HTML()
 
-    preference.change(display_menu, inputs=preference, outputs=menu_display)
+    # 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")
 
-    app.launch()
+app.launch()