app.py

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

@app.route("/")
def home():
    return "Welcome to BiryaniHub!"

@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")

@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 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:
         # 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 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__":
    # Use the PORT environment variable provided by Hugging Face
    port = int(os.environ.get("PORT", 7860))  # Default to 7860
    app.run(host="0.0.0.0", port=port, debug=False)