Update app.py

app.py

@@ -5,7 +5,6 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-
 # Function to Hash Password
 def hash_password(password):
     return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
@@ -64,7 +63,7 @@ def login(email, password):
 # Function to load menu items from Salesforce
 def load_menu_from_salesforce():
     try:
-        query = "SELECT Name, Price__c, Description__c, Image1__c, Veg_NonVeg__c FROM Menu_Item__c"
+        query = "SELECT Name, Price__c, Description__c, Image1__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
         result = sf.query(query)
         return result['records']
     except Exception as e:
@@ -73,28 +72,33 @@ def load_menu_from_salesforce():
 # Function to filter menu items based on preference
 def filter_menu(preferences):
     menu_data = load_menu_from_salesforce()
-    filtered_data = []
+    filtered_data = {}
 
     for item in menu_data:
+        if item["Section__c"] not in filtered_data:
+            filtered_data[item["Section__c"]] = []
+
         if "Veg" in preferences and item["Veg_NonVeg__c"] in ["Veg", "Both"]:
-            filtered_data.append(item)
+            filtered_data[item["Section__c"]].append(item)
         elif "Non-Veg" in preferences and item["Veg_NonVeg__c"] in ["Non-Veg", "Both"]:
-            filtered_data.append(item)
-
-    # Generate HTML content
-    html_content = '<div style="display: flex; flex-wrap: wrap; gap: 20px; justify-content: center;">'
-    for item in filtered_data:
-        html_content += f"""
-        <div style="width: 300px; border: 1px solid #ddd; border-radius: 10px; overflow: hidden; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
-            <img src="{item['Image1__c']}" alt="{item['Name']}" style="width: 100%; height: 200px; object-fit: cover;">
-            <div style="padding: 15px;">
-                <h3 style="margin: 0; font-size: 20px; color: #333;">{item['Name']}</h3>
-                <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price__c']}</p>
-                <p style="margin: 10px 0; font-size: 14px; color: #555;">{item['Description__c']}</p>
-                <button style="width: 100%; padding: 10px; background-color: #28a745; color: white; border: none; border-radius: 5px; cursor: pointer;">Add to Cart</button>
+            filtered_data[item["Section__c"]].append(item)
+
+    html_content = '<div style="padding: 20px; max-width: 1200px; margin: auto;">'
+    for section, items in filtered_data.items():
+        html_content += f"<h2 style='text-align: center; color: #333;'>{section}</h2>"
+        html_content += '<div style="display: flex; flex-wrap: wrap; justify-content: center; gap: 20px;">'
+        for item in items:
+            html_content += f"""
+            <div style="width: 300px; border: 1px solid #ddd; border-radius: 10px; overflow: hidden; box-shadow: 0px 4px 8px rgba(0, 0, 0, 0.1);">
+                <img src="{item['Image1__c']}" alt="{item['Name']}" style="width: 100%; height: 200px; object-fit: cover;">
+                <div style="padding: 15px;">
+                    <h3 style="margin: 0; font-size: 20px; color: #333;">{item['Name']}</h3>
+                    <p style="margin: 5px 0; font-size: 16px; color: #888;">${item['Price__c']}</p>
+                    <p style="margin: 10px 0; font-size: 14px; color: #555;">{item['Description__c']}</p>
+                </div>
             </div>
-        </div>
-        """
+            """
+        html_content += '</div>'
     html_content += '</div>'
     return html_content or "<p style='text-align: center;'>No items match your filter.</p>"
 
@@ -102,6 +106,10 @@ def filter_menu(preferences):
 with gr.Blocks() as app:
     login_status = gr.State(value=False)
 
+    # Header
+    with gr.Row():
+        gr.HTML("<h1 style='text-align: center; color: #222;'>Welcome to Biryani Hub</h1>")
+
     # Login Page
     with gr.Row(visible=True) as login_page:
         with gr.Column():
@@ -125,10 +133,14 @@ with gr.Blocks() as app:
     # Menu Page
     with gr.Row(visible=False) as menu_page:
         preferences = gr.CheckboxGroup(
-            choices=["Veg", "Non-Veg"], label="Filter Preference", value=["Veg", "Non-Veg"]
+            choices=["Veg", "Non-Veg"], label="Filter Preference"
         )
         menu_output = gr.HTML()
 
+    # Footer
+    with gr.Row():
+        gr.HTML("<footer style='text-align: center; color: #666; padding: 20px; background-color: #f9f9f9;'>Thank you! Welcome again!</footer>")
+
     # Functions for page transitions and operations
     def handle_login(email, password):
         status, user = login(email, password)