Update app.py

app.py

@@ -4,102 +4,104 @@ from simple_salesforce import Salesforce
 # Salesforce Connection
 sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
 
-# Initialize cart
+# Local cart to manage food items
 cart = {}
 
-# Fetch menu items from Salesforce
 def fetch_menu_items():
+    """Fetch menu items from Salesforce."""
     query = "SELECT Id, Name, Price__c, Description__c, Image1__c FROM Menu_Item__c"
-    menu_items = sf.query(query)["records"]
-    return menu_items
+    result = sf.query(query)
+    return result['records']
 
-# Generate menu HTML with add-to-cart functionality
 def generate_menu_html(menu_items):
-    html = ""
+    """Generate HTML for menu items."""
+    html = "<div>"
     for item in menu_items:
         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 style='display: flex; align-items: center; justify-content: space-between; margin-bottom: 10px;'>
             <div>
+                <img src='{item['Image1__c']}' alt='{item['Name']}' style='width: 100px; height: 100px; object-fit: cover; margin-right: 10px;'>
                 <h3>{item['Name']}</h3>
-                <p>{item.get('Description__c', '')}</p>
                 <p>₹{item['Price__c']}</p>
-                <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>
+                <button onclick=\"addToCart('{item['Id']}', '{item['Name']}', {item['Price__c']})\">Add to Cart</button>
             </div>
         </div>
         """
+    html += "</div>"
     return html
 
-# Update cart when items are added
-def update_cart(item_id, quantity):
+def add_to_cart(item_id, name, price):
+    """Add an item to the cart."""
     if item_id in cart:
-        cart[item_id]["quantity"] = quantity
+        cart[item_id]['quantity'] += 1
     else:
-        menu_item = sf.Menu_Item__c.get(item_id)
         cart[item_id] = {
-            "name": menu_item["Name"],
-            "price": menu_item["Price__c"],
-            "quantity": quantity
+            'name': name,
+            'price': price,
+            'quantity': 1
         }
-    return view_cart()
+    return update_cart_display()
 
-# View cart details dynamically
-def view_cart():
-    cart_html = "<h3>Cart Details</h3><ul>"
+def update_cart_display():
+    """Update the cart display."""
+    if not cart:
+        return "Cart is empty."
+
+    cart_html = "<div>"
     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>"
+        total_price += details['price'] * details['quantity']
+        cart_html += f"""
+        <div style='display: flex; justify-content: space-between; align-items: center;'>
+            <span>{details['name']} (x{details['quantity']})</span>
+            <span>₹{details['price'] * details['quantity']}</span>
+        </div>
+        """
+    cart_html += f"<div style='margin-top: 10px;'>Total: ₹{total_price}</div>"
+    cart_html += "</div>"
     return cart_html
 
-# Place order and sync with Salesforce
-def place_order():
-    # Create Order in Salesforce
+def proceed_to_order():
+    """Place the order in Salesforce."""
+    if not cart:
+        return "Cart is empty. Cannot place an order."
+
+    # Create an Order record
     order = sf.Order.create({
-        "Status": "Draft",
-        "TotalAmount": sum(details["price"] * details["quantity"] for details in cart.values())
+        'Status': 'Draft',
+        'EffectiveDate': '2023-01-01',  # Replace with dynamic date if needed
+        'AccountId': 'your_account_id'  # Replace with actual Account ID
     })
-    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"]
+            'OrderId': order['id'],
+            'Quantity': details['quantity'],
+            'UnitPrice': details['price'],
+            'PricebookEntryId': 'your_pricebook_entry_id'  # Replace accordingly
         })
 
-    # Clear the cart
+    # Clear the cart after placing the order
     cart.clear()
     return "Order placed successfully!"
 
-# Fetch menu and initialize UI components
+# Fetch menu items and generate HTML
 menu_items = fetch_menu_items()
 menu_html = generate_menu_html(menu_items)
 
 with gr.Blocks() as demo:
-    with gr.Row():
-        gr.Markdown("# Welcome to Biryani Hub")
-    
+    gr.Markdown("# Welcome to Biryani Hub")
+
     with gr.Row():
         menu_display = gr.HTML(value=menu_html)
-    
+        cart_display = gr.HTML(value="Cart is empty.")
+
     with gr.Row():
-        cart_display = gr.HTML(value=view_cart())
-        item_id_input = gr.Textbox(placeholder="Item ID")
-        quantity_input = gr.Number(label="Quantity", value=1, interactive=True)
-        add_to_cart_button = gr.Button("Add to Cart")
-        place_order_button = gr.Button("Place Order")
-    
-    add_to_cart_button.click(update_cart, inputs=[item_id_input, quantity_input], outputs=cart_display)
-    place_order_button.click(place_order, inputs=[], outputs=cart_display)
-
-# Launch the app
-demo.launch()
+        add_button = gr.Button("Add to Cart")
+        proceed_button = gr.Button("Proceed to Order")
+
+    add_button.click(add_to_cart, inputs=[], outputs=cart_display)
+    proceed_button.click(proceed_to_order, inputs=[], outputs=cart_display)
+
+demo.launch()