Update app.py

app.py

@@ -1,124 +1,267 @@
-import gradio as gr
-from models.salesforce import fetch_menu_items, place_order_in_salesforce
-from models.cart import add_to_cart, view_cart, clear_cart
-from models.user import login_user, signup_user
-
-# Global user state
-current_user = {"email": None, "name": None}
-
-# Login Interface
-def login(email, password):
-    success, message = login_user(email, password)
-    if success:
-        current_user["email"] = email
-        current_user["name"] = message
-        return "Login successful! Redirecting to menu..."
-    return f"Login failed: {message}"
-
-# Signup Interface
-def signup(name, email, phone, password):
-    success, message = signup_user(name, email, phone, password)
-    if success:
-        return "Signup successful! Please login now."
-    return f"Signup failed: {message}"
-
-# Load Menu Items
-def load_menu(preference):
-    items = fetch_menu_items()
-    print(f"Fetched menu items: {items}")  # Debugging output
-    filtered_items = [
-        item for item in items
-        if preference == "All"
-        or (preference == "Veg" and item["Veg_NonVeg__c"] == "Veg")
-        or (preference == "Non-Veg" and item["Veg_NonVeg__c"] == "Non-Veg")
-    ]
-    print(f"Filtered menu items: {filtered_items}")  # Debugging output
-    return filtered_items
-    
-def display_menu(preference):
-    menu_items = fetch_menu_items()
-    # Filter based on preference
-    if preference == "Veg":
-        filtered_items = [item for item in menu_items if item["Veg_NonVeg"] == "Veg"]
-    elif preference == "Non-Veg":
-        filtered_items = [item for item in menu_items if item["Veg_NonVeg"] == "Non veg"]
-    else:
-        filtered_items = menu_items
-
-    # Prepare for display in Gradio
-    return [[item["Name"], item["Price"], item["Description"]] for item in filtered_items]
+from flask import Flask, render_template, request, jsonify, redirect, url_for, session
+import random
+from salesforce import get_salesforce_connection
 
-    
+# Initialize Flask app and Salesforce connection
+app = Flask(__name__)
+sf = get_salesforce_connection()
+
+# Set the secret key to handle sessions securely
+app.secret_key = 'sSSjyhInIsUohKpG8sHzty2q'  # Replace with a secure key
 
-# Add to Cart
-def add_to_cart_interface(item_name, price):
-    add_to_cart(item_name, price)
-    return f"Added {item_name} to cart!"
+@app.route("/")
+def home():
+    return render_template("index.html")
 
-# View Cart
-def view_cart_interface():
-    cart, total = view_cart()
-    return cart, total
+@app.route("/signup", methods=["GET", "POST"])
+def signup():
+    if request.method == "POST":
+        name = request.form.get("name")
+        phone = request.form.get("phone")
+        email = request.form.get("email")
+        password = request.form.get("password")
+        try:
+            sf.Customer_Login__c.create({
+                "Name": name,
+                "Phone_Number__c": phone,
+                "Email__c": email,
+                "Password__c": password
+            })
+            return redirect(url_for("login"))
+        except Exception as e:
+            return render_template("signup.html", error=f"Error: {str(e)}")
+    return render_template("signup.html")
 
-# Place Order
-def place_order_interface():
-    email = current_user["email"]
+@app.route("/login", methods=["GET", "POST"])
+def login():
+    if request.method == "POST":
+        email = request.form.get("email")
+        password = request.form.get("password")
+        try:
+            query = f"SELECT Id, Name, Email__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
+            result = sf.query(query)
+            if result["records"]:
+                session['user_id'] = result["records"][0]['Id']
+                session['user_email'] = email
+                return redirect(url_for("menu"))
+            else:
+                return render_template("login.html", error="Invalid credentials!")
+        except Exception as e:
+            return render_template("login.html", error=f"Error: {str(e)}")
+    return render_template("login.html")
+
+@app.route("/menu", methods=["GET", "POST"])
+def menu():
+    selected_category = request.args.get("category", "All")
+    user_id = session.get('user_id')
+    if not user_id:
+        return redirect(url_for('login'))
+    try:
+        query = "SELECT Name, Price__c, Image1__c, Category__c, Description__c FROM Menu_Item__c"
+        result = sf.query(query)
+        food_items = result['records'] if 'records' in result else []
+        categories = {item['Category__c'] for item in food_items if 'Category__c' in item}
+        if selected_category != "All":
+            food_items = [item for item in food_items if item.get("Category__c") == selected_category]
+    except Exception as e:
+        food_items = []
+        categories = []
+        print(f"Error fetching data: {e}")
+    return render_template("menu.html", food_items=food_items, categories=categories, selected_category=selected_category)
+
+@app.route("/cart", methods=["GET"])
+def cart():
+    email = session.get('user_email') # Get logged-in user's email
     if not email:
-        return "Error: Please login to place an order."
-    cart, total = view_cart()
-    if not cart:
-        return "Error: Cart is empty!"
-    order_details = "\n".join([f"{item['Name']} - ${item['Price']} x {item['Quantity']}" for item in cart])
+        return redirect(url_for("login")) # Redirect to login if not logged in
+    try:
+        result = sf.query(f"""
+            SELECT Name, Price__c, Quantity__c, Add_Ons__c, Image1__c
+            FROM Cart_Item__c
+            WHERE Customer_Email__c = '{email}'
+        """)
+        cart_items = result.get("records", [])
+        subtotal = sum(item['Quantity__c'] * item['Price__c'] for item in cart_items)
+    except Exception as e:
+        print(f"Error fetching cart items: {e}")
+        cart_items = []
+        subtotal = 0
+        
+    return render_template("cart.html", cart_items=cart_items, subtotal=subtotal)
+
+
+
+@app.route('/cart/add', methods=['POST'])
+def add_to_cart():
+    data = request.json
+    item_name = data.get('itemName').strip()
+    item_price = data.get('itemPrice')
+    item_image = data.get('itemImage')
+    addons = data.get('addons', [])
+    customer_email = session.get('user_email')  # Get logged-in user's email from session
+
+    if not item_name or not item_price:
+        return jsonify({"success": False, "error": "Item name and price are required."})
+
     try:
-        place_order_in_salesforce(email, order_details, total)
-        clear_cart()
-        return f"Order placed successfully! Total: ${total}"
+         # Check if the item already exists in the cart for this custpmer
+         query= f""" SELECT Id, Quantity__c FROM Cart_Item__c
+            WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
+        """
+         result = sf.query(query)
+         if result['totalSize'] > 0:
+            # Update quantity if the item exists
+            cart_item = result['records'][0]
+            sf.Cart_Item__c.update(cart_item['Id'], {
+                "Quantity__c": cart_item['Quantity__c'] + 1
+            })
+         else:
+             # Add a new item if it doesn't exist
+             sf.Cart_Item__c.create({
+                 "Name": item_name,
+            "Price__c": item_price,
+            "Quantity__c": 1,
+            "Add_Ons__c": ";".join(addons) if addons else None,
+            "Image1__c": item_image,
+            "Customer_Email__c": customer_email,  # Associate with the logged-in user
+                 
+             })
+             return jsonify({"success": True, "message": "Item added/updated successfully."})
     except Exception as e:
-        return f"Error placing order: {str(e)}"
-
-# Gradio Interfaces
-with gr.Blocks() as app:
-    preference_radio = gr.Radio(["All", "Veg", "Non-Veg"], label="Filter Menu", value="All")
-    menu_data_table = gr.Dataframe(headers=["Name", "Price", "Description"], datatype=["str", "number", "str"])
-
-    # Add a button to trigger menu display
-    menu_button = gr.Button("Show Menu")
-
-    # Connect display logic
-    menu_button.click(
-        display_menu,  # Function to call
-        inputs=preference_radio,  # Input component
-        outputs=menu_data_table  # Output component
-    )
-
-
-    with gr.Tab("Signup"):
-        name = gr.Textbox(label="Name", placeholder="Enter your name")
-        signup_email = gr.Textbox(label="Email", placeholder="Enter your email")
-        phone = gr.Textbox(label="Phone", placeholder="Enter your phone number")
-        signup_password = gr.Textbox(label="Password", type="password", placeholder="Create a password")
-        signup_button = gr.Button("Signup")
-        signup_output = gr.Textbox(label="Signup Status")
-        signup_button.click(signup, inputs=[name, signup_email, phone, signup_password], outputs=signup_output)
-
-    with gr.Tab("Menu"):
-        preference = gr.Radio(["All", "Veg", "Non-Veg"], label="Filter Menu", value="All")
-        menu_output = gr.Dataframe(headers=["Name", "Description", "Price"])
-        preference.change(lambda p: load_menu(p), inputs=[preference], outputs=menu_output)
-        item_name = gr.Textbox(label="Item Name")
-        item_price = gr.Textbox(label="Price")
-        add_button = gr.Button("Add to Cart")
-        cart_status = gr.Textbox(label="Cart Status")
-        add_button.click(add_to_cart_interface, inputs=[item_name, item_price], outputs=cart_status)
-
-    with gr.Tab("Cart"):
-        view_button = gr.Button("View Cart")
-        cart_output = gr.Dataframe(headers=["Name", "Price", "Quantity"])
-        total_output = gr.Textbox(label="Total Amount")
-        view_button.click(view_cart_interface, outputs=[cart_output, total_output])
-
-        place_order_button = gr.Button("Place Order")
-        order_status = gr.Textbox(label="Order Status")
-        place_order_button.click(place_order_interface, outputs=order_status)
-
-app.launch()
+        return jsonify({"success": False, "error": str(e)})
+
+@app.route("/cart/add_item", methods=["POST"])
+def add_item_to_cart():
+    data = request.json  # Extract JSON data from the request
+    email = data.get('email')  # Customer email
+    item_name = data.get('item_name')  # Item name
+    quantity = data.get('quantity', 0)  # Quantity to add (default is 1) // default value is 1
+
+    try:
+        # Check if the item already exists in the cart for this customer
+        cart_items = sf.query(
+            f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Item_Name__c = '{item_name}'"
+        )['records']
+
+        if cart_items:
+            # If the item already exists, update its quantity
+            cart_item = cart_items[0]
+            new_quantity = cart_item['Quantity__c'] + quantity
+            sf.Cart_Item__c.update(cart_item['Id'], {"Quantity__c": new_quantity})
+            return jsonify({"success": True, "message": "Item quantity updated successfully."})
+        else:
+            # If the item does not exist, add it to the cart
+            sf.Cart_Item__c.create({
+                "Customer_Email__c": email,
+                "Item_Name__c": item_name,
+                "Quantity__c": quantity
+            })
+
+        return jsonify({"success": True, "message": "Item added/updated successfully."})
+    except Exception as e:
+        return jsonify({"success": False, "error": str(e)}), 500
+
+
+@app.route('/cart/remove/<item_name>', methods=['POST'])
+def remove_cart_item(item_name):
+    try:
+        customer_email = session.get('user_email')
+        if not customer_email:
+            return jsonify({'success': False, 'message': 'User email not found. Please log in again.'}), 400
+        query = f"""
+            SELECT Id FROM Cart_Item__c 
+            WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
+        """
+        result = sf.query(query)
+        if result['totalSize'] == 0:
+            return jsonify({'success': False, 'message': 'Item not found in cart.'}), 400
+        cart_item_id = result['records'][0]['Id']
+        sf.Cart_Item__c.delete(cart_item_id)
+        return jsonify({'success': True, 'message': f"'{item_name}' removed successfully!"}), 200
+    except Exception as e:
+        print(f"Error: {str(e)}")
+        return jsonify({'success': False, 'message': f"An error occurred: {str(e)}"}), 500
+
+@app.route('/api/addons', methods=['GET'])
+def get_addons():
+    item_name = request.args.get('item_name')
+    if not item_name:
+        return jsonify({"success": False, "error": "Item name is required."})
+
+    try:
+        query = f"SELECT Name, Price__c FROM Add_Ons__c WHERE Menu_Item__r.Name = '{item_name}'"
+        addons = sf.query(query)['records']
+        return jsonify({"success": True, "addons": addons})
+    except Exception as e:
+        print(f"Error fetching add-ons: {e}")
+        return jsonify({"success": False, "error": "Unable to fetch add-ons. Please try again later."})
+
+@app.route("/cart/update_quantity", methods=["POST"])
+def update_quantity():
+    data = request.json  # Extract JSON data from the request
+    email = data.get('email')  # Customer email
+    item_name = data.get('item_name')  # Item name (Cart Item Name in Salesforce)
+    quantity = data.get('quantity')  # New quantity
+
+    # Validate inputs
+    if not email or not item_name:
+        return jsonify({"success": False, "error": "Email and item name are required."}), 400
+
+    try:
+        # Query the cart item using the correct field names
+        cart_items = sf.query(
+            f"SELECT Id, Quantity__c FROM Cart_Item__c WHERE Customer_Email__c = '{email}' AND Name__c = '{item_name}'"
+        )['records']
+
+        if not cart_items:
+            return jsonify({"success": False, "error": "Cart item not found."}), 404
+
+        # Get the first matching record ID
+        cart_item_id = cart_items[0]['Id']
+
+        # Update the quantity in Salesforce
+        sf.Cart_Item__c.update(cart_item_id, {"Quantity__c": quantity})
+
+        return jsonify({"success": True, "new_quantity": quantity})
+    except Exception as e:
+        return jsonify({"success": False, "error": str(e)}), 500
+
+    
+
+@app.route("/checkout", methods=["POST"])
+def checkout():
+    email = session.get('user_email')
+    user_id = session.get('user_id')
+    if not email or not user_id:
+        return jsonify({"success": False, "message": "User not logged in"})
+    try:
+        result = sf.query(f"""
+            SELECT Id, Name, Price__c, Quantity__c, Add_Ons__c
+            FROM Cart_Item__c
+            WHERE Customer_Email__c = '{email}'
+        """)
+        cart_items = result["records"]
+        if not cart_items:
+            return jsonify({"success": False, "message": "Cart is empty"})
+        total_price = sum(item['Quantity__c'] * item['Price__c'] for item in cart_items)
+        order_data = {
+            "Customer_Name__c": user_id,
+            "Customer_Email__c": email,
+            "Total_Amount__c": total_price,
+            "Order_Status__c": "Pending",
+            "Order_Items__c": "\n".join(
+                [f"{item['Name']} (Qty: {item['Quantity__c']})" for item in cart_items]
+            ),
+            "Add_Ons__c": "\n".join(
+                [f"{item['Add_Ons__c']}" if item['Add_Ons__c'] else "None" for item in cart_items]
+            ),
+        }
+        sf.Order__c.create(order_data)
+        for item in cart_items:
+            sf.Cart_Item__c.delete(item["Id"])
+        return jsonify({"success": True, "message": "Order placed successfully!"})
+    except Exception as e:
+        return jsonify({"success": False, "error": str(e)})
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=8000)