Update app.py

app.py

@@ -6,58 +6,48 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-# Sample Menu Data in JSON Format
-sample_menu = [
-    {
-        "name": "Samosa",
-        "price": 9.0,
-        "description": "Crispy fried pastry filled with spiced potatoes and peas.",
-        "image": "https://via.placeholder.com/200x200",  # Replace with your image URLs
-        "veg_nonveg": "Veg",
-        "section": "Starters",
-    },
-    {
-        "name": "Chicken Biryani",
-        "price": 15.0,
-        "description": "Flavorful rice dish cooked with chicken and spices.",
-        "image": "https://via.placeholder.com/200x200",
-        "veg_nonveg": "Non-Veg",
-        "section": "Biryani",
-    },
-    {
-        "name": "Paneer Butter Masala",
-        "price": 12.0,
-        "description": "Paneer cubes cooked in a creamy tomato-based gravy.",
-        "image": "https://via.placeholder.com/200x200",
-        "veg_nonveg": "Veg",
-        "section": "Curries",
-    }
-]
-
-# Sample Add-Ons
-sample_add_ons = [
-    {"name": "Extra Cheese", "price": 2.0},
-    {"name": "Extra Spicy", "price": 1.5},
-    {"name": "No Onions", "price": 0.5},
-]
-
-# Function to Save Cart Summary to Salesforce
-def save_cart_to_salesforce(cart, total_cost):
+# Function to Hash Password
+def hash_password(password):
+    return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
+
+# Function to Verify Password
+def verify_password(plain_password, hashed_password):
+    return bcrypt.checkpw(plain_password.encode('utf-8'), hashed_password.encode('utf-8'))
+
+# Fetch menu items from Salesforce
+def fetch_menu_items():
     try:
-        if not cart or total_cost <= 0:
-            return "Cart is empty or invalid total cost."
+        query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
+        result = sf.query(query)
+        return result['records']
+    except Exception as e:
+        print(f"Error fetching menu items: {e}")
+        return []
 
-        # Create Order Record
+# Fetch add-ons from Salesforce
+def fetch_add_ons():
+    try:
+        query = "SELECT Name, Price__c FROM Add_Ons__c"
+        result = sf.query(query)
+        return result['records']
+    except Exception as e:
+        print(f"Error fetching add-ons: {e}")
+        return []
+
+# Save cart details to Salesforce
+def save_cart(cart_items, total_cost):
+    try:
+        # Create order record
         order_record = {
             'Name': f"Order {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}",
             'Total_Cost__c': total_cost,
-            'Order_Date__c': datetime.now().isoformat(),
+            'Order_Date__c': datetime.now().isoformat()
         }
         order_result = sf.Order__c.create(order_record)
         order_id = order_result['id']
 
-        # Create Order Items
-        for item in cart:
+        # Save cart items as order items
+        for item in cart_items:
             extras = ", ".join(extra['name'] for extra in item.get('extras', []))
             order_item_record = {
                 'Name': item['name'],
@@ -66,92 +56,80 @@ def save_cart_to_salesforce(cart, total_cost):
                 'Price__c': item['price'],
                 'Extras__c': extras,
                 'Instructions__c': item.get('instructions', ''),
-                'Total_Cost__c': item['totalCost'],
+                'Total_Cost__c': item['total_cost']
             }
             sf.Order_Item__c.create(order_item_record)
 
-        return "Order saved successfully!"
+        return f"Order {order_id} saved successfully!"
     except Exception as e:
-        return f"Error saving order: {str(e)}"
-
-# Function to Load Menu Items
-def load_menu(preference):
-    filtered_menu = [
-        item for item in sample_menu
-        if preference == "All" or item["veg_nonveg"] == preference
-    ]
-
-    html = ""
-    sections = {item["section"] for item in filtered_menu}
-    for section in sections:
-        html += f"<h2>{section}</h2>"
-        for item in filtered_menu:
-            if item["section"] == section:
-                html += f"""
-                <div style="border:1px solid #ddd; padding:10px; margin:10px;">
-                    <img src="{item['image']}" style="width:100px; height:100px;">
-                    <h3>{item['name']}</h3>
-                    <p>{item['description']}</p>
-                    <p>Price: ${item['price']}</p>
-                </div>
-                """
-    return html
-
-# Function to Add Items to Cart
-def add_to_cart(name, price, quantity, extras, instructions):
-    total_cost = (price * quantity) + sum(extra["price"] for extra in extras)
-    cart_item = {
-        "name": name,
-        "price": price,
-        "quantity": quantity,
-        "extras": extras,
-        "instructions": instructions,
-        "totalCost": total_cost,
-    }
-    return cart_item, total_cost
-
-# Gradio Interface
+        print(f"Error saving cart: {e}")
+        return f"Error saving cart: {str(e)}"
+
+# Function to filter menu based on preference
+def filter_menu(preference):
+    menu_items = fetch_menu_items()
+    filtered_items = [item for item in menu_items if preference == "All" or item['Veg_NonVeg__c'] == preference]
+
+    menu_output = {}
+    for item in filtered_items:
+        section = item.get('Section__c', 'Uncategorized')
+        if section not in menu_output:
+            menu_output[section] = []
+        menu_output[section].append(item)
+    return menu_output
+
+# Gradio App
 with gr.Blocks() as app:
     with gr.Row():
         gr.HTML("<h1>Welcome to Biryani Hub</h1>")
-        preference = gr.Radio(["All", "Veg", "Non-Veg"], value="All", label="Filter Preference")
-        menu_display = gr.HTML()
+
+    # Menu and Cart Interface
+    with gr.Row():
+        preference = gr.Radio(choices=["All", "Veg", "Non-Veg"], value="All", label="Filter Preference")
+        menu_display = gr.JSON(label="Menu Items")
+        cart_display = gr.JSON(label="Cart Items")
+        total_cost_display = gr.Number(label="Total Cost", value=0)
 
     with gr.Row():
-        cart_items = gr.JSON(label="Cart Items")
-        total_cost = gr.Textbox(label="Total Cost")
         item_name = gr.Textbox(label="Item Name")
-        item_price = gr.Number(label="Price")
+        item_price = gr.Number(label="Price", value=0)
         item_quantity = gr.Number(label="Quantity", value=1)
-        extras = gr.JSON(value=sample_add_ons, label="Add-Ons")
-        instructions = gr.Textbox(label="Special Instructions")
-        add_item_button = gr.Button("Add to Cart")
+        add_ons_display = gr.JSON(label="Add-Ons")
+        special_instructions = gr.Textbox(label="Special Instructions")
 
     with gr.Row():
-        checkout_button = gr.Button("Checkout")
-        checkout_status = gr.Textbox(label="Status")
-
-    # Load Menu on Filter Change
-    preference.change(
-        lambda pref: load_menu(pref), 
-        inputs=preference, 
-        outputs=menu_display
-    )
-
-    # Add Item to Cart
-    add_item_button.click(
-        lambda name, price, quantity, extras, instructions: add_to_cart(
-            name, price, quantity, extras, instructions
-        ),
-        inputs=[item_name, item_price, item_quantity, extras, instructions],
-        outputs=[cart_items, total_cost],
-    )
-
-    # Checkout and Save to Salesforce
-    checkout_button.click(
-        lambda cart, total: save_cart_to_salesforce(cart, total),
-        inputs=[cart_items, total_cost],
-        outputs=checkout_status,
-    )
+        add_to_cart_button = gr.Button("Add to Cart")
+        save_order_button = gr.Button("Save Order")
+
+    # State
+    cart = gr.State([])  # Cart state to maintain items
+
+    def update_menu(preference):
+        menu_items = filter_menu(preference)
+        return menu_items, fetch_add_ons()
+
+    def add_to_cart(cart, name, price, quantity, add_ons, instructions):
+        if not name or quantity <= 0:
+            return cart, "Invalid item details!"
+        total_cost = price * quantity
+        cart.append({
+            "name": name,
+            "price": price,
+            "quantity": quantity,
+            "extras": add_ons,
+            "instructions": instructions,
+            "total_cost": total_cost
+        })
+        return cart, f"Added {name} to cart!"
+
+    def save_order(cart):
+        total_cost = sum(item['total_cost'] for item in cart)
+        response = save_cart(cart, total_cost)
+        return response
+
+    # Interactivity
+    preference.change(update_menu, inputs=[preference], outputs=[menu_display, add_ons_display])
+    add_to_cart_button.click(add_to_cart, inputs=[cart, item_name, item_price, item_quantity, add_ons_display, special_instructions], outputs=[cart_display, total_cost_display])
+    save_order_button.click(save_order, inputs=[cart], outputs=[total_cost_display])
 
 app.launch()