Update app.py

app.py

@@ -4,74 +4,99 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
+# Initialize cart
+cart = {}
+
 # Fetch menu items from Salesforce
 def fetch_menu_items():
     query = "SELECT Id, Name, Price__c, Description__c, Image1__c FROM Menu_Item__c"
-    records = sf.query(query)["records"]
-    return records
-
-# Fetch cart items (local data structure for demonstration)
-cart_items = []
-
-# Add item to cart
-def add_to_cart(item_id, name, price):
-    for item in cart_items:
-        if item["id"] == item_id:
-            item["quantity"] += 1
-            return f"{name} added to the cart!", cart_items
-    cart_items.append({"id": item_id, "name": name, "price": price, "quantity": 1})
-    return f"{name} added to the cart!", cart_items
-
-# Display cart items
-def view_cart():
-    cart_html = ""
-    total_items = 0
-    total_price = 0
-    for item in cart_items:
-        cart_html += f"<p>{item['name']} - ₹{item['price']} x {item['quantity']}</p>"
-        total_items += item["quantity"]
-        total_price += item["price"] * item["quantity"]
-    cart_html += f"<h3>Total: ₹{total_price} ({total_items} items)</h3>"
-    return cart_html
+    menu_items = sf.query(query)["records"]
+    return menu_items
 
-# Generate menu page HTML
-def generate_menu_page(menu_items):
-    menu_html = ""
+# Generate menu HTML with add-to-cart functionality
+def generate_menu_html(menu_items):
+    html = ""
     for item in menu_items:
-        menu_html += f"""
-        <div style="display: flex; align-items: center; margin-bottom: 20px;">
-            <img src="{item['Image1__c']}" style="width: 100px; height: 100px; margin-right: 20px;">
+        html += f"""
+        <div style='border: 1px solid #ccc; padding: 10px; margin: 10px; display: flex; align-items: center;'>
+            <img src='{item.get('Image1__c', '')}' style='width: 100px; height: 100px; margin-right: 10px;' />
             <div>
-                <h4>{item['Name']}</h4>
+                <h3>{item['Name']}</h3>
+                <p>{item.get('Description__c', '')}</p>
                 <p>₹{item['Price__c']}</p>
-                <button onclick="add_to_cart('{item['Id']}', '{item['Name']}', {item['Price__c']})">Add</button>
+                <div style='display: flex; align-items: center;'>
+                    <button onclick="decreaseQuantity('{item['Id']}')">-</button>
+                    <span id="quantity-{item['Id']}" style="margin: 0 10px;">0</span>
+                    <button onclick="increaseQuantity('{item['Id']}')">+</button>
+                </div>
             </div>
         </div>
         """
-    return menu_html
+    return html
 
-# Fetch menu items for display
-menu_items = fetch_menu_items()
-menu_html = generate_menu_page(menu_items)
+# Update cart when items are added
+def update_cart(item_id, quantity):
+    if item_id in cart:
+        cart[item_id]["quantity"] = quantity
+    else:
+        menu_item = sf.Menu_Item__c.get(item_id)
+        cart[item_id] = {
+            "name": menu_item["Name"],
+            "price": menu_item["Price__c"],
+            "quantity": quantity
+        }
+    return view_cart()
 
-# Gradio interface for menu page
-def menu_interface():
-    return menu_html, gr.update(visible=False)
+# View cart details dynamically
+def view_cart():
+    cart_html = "<h3>Cart Details</h3><ul>"
+    total_price = 0
+    for item_id, details in cart.items():
+        item_total = details["price"] * details["quantity"]
+        total_price += item_total
+        cart_html += f"<li>{details['name']} x {details['quantity']} - ₹{item_total}</li>"
+    cart_html += f"</ul><p><strong>Total: ₹{total_price}</strong></p>"
+    return cart_html
+
+# Place order and sync with Salesforce
+def place_order():
+    # Create Order in Salesforce
+    order = sf.Order.create({
+        "Status": "Draft",
+        "TotalAmount": sum(details["price"] * details["quantity"] for details in cart.values())
+    })
+    order_id = order["id"]
+
+    # Add Order Items
+    for item_id, details in cart.items():
+        sf.OrderItem.create({
+            "OrderId": order_id,
+            "Product2Id": item_id,
+            "UnitPrice": details["price"],
+            "Quantity": details["quantity"]
+        })
 
-def cart_interface():
-    return view_cart(), gr.update(visible=True)
+    # Clear the cart
+    cart.clear()
+    return "Order placed successfully!"
+
+# Set up Gradio interface
+menu_items = fetch_menu_items()
+menu_html = generate_menu_html(menu_items)
 
-# Gradio setup
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("# Welcome to Biryani Hub")
-
+    
+    with gr.Row():
+        menu_display = gr.HTML(value=menu_html)
+    
     with gr.Row():
-        menu_display = gr.HTML(value=menu_html)  # Removed `interactive=False`
-        cart_button = gr.Button("View Cart", visible=True)
-        cart_display = gr.HTML(value=view_cart(), visible=False)
+        cart_display = gr.HTML(value=view_cart())
+        place_order_button = gr.Button("Place Order")
     
-    cart_button.click(cart_interface, inputs=[], outputs=[menu_display, cart_display])
+    menu_display.change(update_cart, inputs=["item_id", "quantity"], outputs=cart_display)
+    place_order_button.click(place_order, inputs=[], outputs=cart_display)
 
 # Launch the app
-demo.launch()
+demo.launch()