Update app.py

app.py

@@ -344,62 +344,57 @@ def cart():
         else:
             coupons = []
 
-        # Inside your /cart route:
-
-        category = request.args.get('category', 'All')  # Get the category (default is 'All')
-        section = request.args.get('section', 'Biryanis')  # Get the section (default is 'Biryanis')
-
+        # Initialize suggestions as an empty list
         suggestions = []
 
-        # Determine the category filter (Veg/Non veg/All)
-        if category == 'Veg':
-            category_filter = "Veg"
-        elif category == 'Non veg':
-            category_filter = "Non veg"
-        else:  # If category is 'All'
-            category_filter = "All"
-
-        # Define section-to-complementary section mapping
-        complementary_sections = {
-            'Breads': ['Curries', 'Biryanis', 'Starters'],
-            'Biryanis': ['Curries', 'Starters', 'Desserts'],
-            'Curries': ['Rice', 'Breads', 'Starters'],
-            'Starters': ['Biryanis', 'Curries', 'Desserts'],
-            'Desserts': ['Biryanis', 'Curries', 'Soft Drinks'],
-            'Soft Drinks': ['Starters', 'Biryanis', 'Curries']
-        }
-
-        # Get the complementary sections for the selected section
-        suggested_sections = complementary_sections.get(section, [])
-
-        # Fetch suggestions from the complementary sections
-        try:
-            # Loop through complementary sections and fetch the relevant items
-            for suggested_section in suggested_sections:
-                # If category is "All", we can fetch both Veg and Non veg items from the complementary section
-                if category_filter == "All":
-                    query = f"""
-                        SELECT Name, Price__c, Image1__c
-                        FROM Menu_Item__c
-                        WHERE Section__c = '{suggested_section}' 
-                        AND (Veg_NonVeg__c = 'Veg' OR Veg_NonVeg__c = 'Non veg')
-                        LIMIT 4
-                    """
-                else:
-                    # Otherwise, fetch only Veg or Non veg items depending on the selected category
-                    query = f"""
-                        SELECT Name, Price__c, Image1__c
-                        FROM Menu_Item__c
-                        WHERE Section__c = '{suggested_section}' 
-                        AND Veg_NonVeg__c = '{category_filter}'
-                        LIMIT 4
-                    """
-                suggestion_result = sf.query(query)
-                suggestions.extend(suggestion_result.get("records", []))  # Add all suggestions from each section
-
-        except Exception as e:
-            print(f"Error fetching suggestions: {e}")
-
+        # If there are items in the cart, fetch suggestions
+        if cart_items:
+            # Get the category and section of the first item in the cart (You can choose which item you want to base suggestions on)
+            first_item = cart_items[0]
+            item_category = first_item.get('Category__c', 'All')  # Default to 'All' if not found
+            item_section = first_item.get('Section__c', 'Biryanis')  # Default to 'Biryanis' if not found
+
+            # Define section-to-complementary section mapping
+            complementary_sections = {
+                'Breads': ['Curries', 'Biryanis', 'Starters'],
+                'Biryanis': ['Curries', 'Starters', 'Desserts'],
+                'Curries': ['Rice', 'Breads', 'Starters'],
+                'Starters': ['Biryanis', 'Curries', 'Desserts'],
+                'Desserts': ['Biryanis', 'Curries', 'Soft Drinks'],
+                'Soft Drinks': ['Starters', 'Biryanis', 'Curries']
+            }
+
+            # Get the complementary sections for the selected section
+            suggested_sections = complementary_sections.get(item_section, [])
+
+            # Fetch suggestions from the complementary sections
+            try:
+                for suggested_section in suggested_sections:
+                    if item_category == "All":
+                        query = f"""
+                            SELECT Name, Price__c, Image1__c
+                            FROM Menu_Item__c
+                            WHERE Section__c = '{suggested_section}' 
+                            AND (Veg_NonVeg__c = 'Veg' OR Veg_NonVeg__c = 'Non veg')
+                            LIMIT 4
+                        """
+                    else:
+                        query = f"""
+                            SELECT Name, Price__c, Image1__c
+                            FROM Menu_Item__c
+                            WHERE Section__c = '{suggested_section}' 
+                            AND Veg_NonVeg__c = '{item_category}'
+                            LIMIT 4
+                        """
+                    suggestion_result = sf.query(query)
+                    suggestions.extend(suggestion_result.get("records", []))  # Add suggestions from each section
+
+                # Limit the number of suggestions to 4
+                if len(suggestions) > 4:
+                    suggestions = suggestions[:4]
+
+            except Exception as e:
+                print(f"Error fetching suggestions: {e}")
 
         return render_template(
             "cart.html",