from flask import Flask, render_template, request, jsonify, redirect, url_for, session
from flask_session import Session # Import the Session class
from flask.sessions import SecureCookieSessionInterface # Import the class
from salesforce import get_salesforce_connection
import os
# Initialize Flask app and Salesforce connection
print("Starting app...")
app = Flask(__name__)
print("Flask app initialized.")
# Add debug logs in Salesforce connection setup
sf = get_salesforce_connection()
print("Salesforce connection established.")
# Set the secret key to handle sessions securely
app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q") # Replace with a secure key
# Configure the session type
app.config["SESSION_TYPE"] = "filesystem" # Use filesystem for session storage
#app.config["SESSION_COOKIE_NAME"] = "my_session" # Optional: Change session cookie name
app.config["SESSION_COOKIE_SECURE"] = True # Ensure cookies are sent over HTTPS
app.config["SESSION_COOKIE_SAMESITE"] = "None" # Allow cross-site cookies
# Initialize the session
Session(app) # Correctly initialize the Session object
print("Session interface configured.")
# Ensure secure session handling for environments like Hugging Face
app.session_interface = SecureCookieSessionInterface()
print("Session interface configured.")
import random
import string
def generate_referral_code(length=8):
# Generates a random referral code with uppercase, lowercase letters, and digits
characters = string.ascii_letters + string.digits # A-Z, a-z, 0-9
referral_code = ''.join(random.choice(characters) for _ in range(length))
return referral_code
@app.route('/')
def home():
# Get email and name from the URL query parameters
email = request.args.get('email')
name = request.args.get('name')
# Use email and name as needed (e.g., display them in the page)
return render_template('index.html', email=email, name=name)
from datetime import datetime
def generate_coupon_code(length=10):
"""Generates a random alphanumeric coupon code"""
characters = string.ascii_uppercase + string.digits # A-Z, 0-9
return ''.join(random.choice(characters) for _ in range(length))
import re
import re
@app.route("/order-history", methods=["GET"])
def order_history():
email = session.get('user_email') # Get logged-in user's email
if not email:
return redirect(url_for("login"))
try:
# Fetch past orders for the user
result = sf.query(f"""
SELECT Id, Customer_Name__c, Customer_Email__c, Total_Amount__c,
Order_Details__c, Order_Status__c, Discount__c, Total_Bill__c, CreatedDate
FROM Order__c
WHERE Customer_Email__c = '{email}'
ORDER BY CreatedDate DESC
""")
orders = result.get("records", []) # Fetch all orders
# Strip image URLs from order details and split remaining data by new lines
for order in orders:
order_details = order.get("Order_Details__c", "")
# Remove image URLs using regex
cleaned_details = re.sub(r'http[s]?://\S+', '', order_details)
# Now split the cleaned details by lines and join them with
to create line breaks
cleaned_details = cleaned_details.replace("\n", "
")
# Update the order details with the cleaned and formatted details
order['Order_Details__c'] = cleaned_details
return render_template("order_history.html", orders=orders)
except Exception as e:
print(f"Error fetching order history: {str(e)}")
return render_template("order_history.html", orders=[], error=str(e))
@app.route("/logout")
def logout():
session.clear() # Clear the session
return redirect(url_for("login")) # Redirect to the login page
@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").strip() # Trim spaces
password = request.form.get("password")
referral_code = request.form.get("referral") # Fetch referral code from the form
generated_referral_code = generate_referral_code()
try:
ref = 0 # Default reward points for new user
# **Fix: Fetch all emails and compare in Python (Case-Insensitive)**
email_query = "SELECT Id, Email__c FROM Customer_Login__c"
email_result = sf.query(email_query)
# Convert all stored emails to lowercase and compare with user input
existing_emails = {record["Email__c"].lower() for record in email_result["records"]}
if email.lower() in existing_emails:
return render_template("signup.html", error="Email already in use! Please use a different email.")
# Check if a referral code is entered
if referral_code:
referral_query = f"SELECT Id, Email__c, Name FROM Customer_Login__c WHERE Referral__c = '{referral_code}'"
referral_result = sf.query(referral_query)
if not referral_result['records']:
return render_template("signup.html", error="Invalid referral code!")
# Get referrer's details
referrer = referral_result['records'][0]
referrer_email = referrer.get('Email__c')
referrer_name = referrer.get('Name')
# Generate a new unique coupon code
new_coupon_code = generate_coupon_code()
# Check if referrer already has a record in Referral_Coupon__c
existing_coupon_query = f"SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{referrer_email}'"
existing_coupon_result = sf.query(existing_coupon_query)
if existing_coupon_result['records']:
referral_record = existing_coupon_result['records'][0]
referral_id = referral_record['Id']
existing_coupons = referral_record.get('Coupon_Code__c', '')
updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
# Update the existing record with the new coupon
sf.Referral_Coupon__c.update(referral_id, {
"Coupon_Code__c": updated_coupons
})
else:
# If no record exists, create a new one
sf.Referral_Coupon__c.create({
"Name": referrer_name,
"Referral_Email__c": referrer_email,
"Coupon_Code__c": new_coupon_code
})
# **Fix: Ensure Salesforce enforces unique email constraint**
sf.Customer_Login__c.create({
"Name": name,
"Phone_Number__c": phone,
"Email__c": email,
"Password__c": password,
"Reward_Points__c": ref, # No points added, only coupon is created
"Referral__c": generated_referral_code
})
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")
print(f"Login attempt with email: {email}") # Debug log
try:
# Fetch user details, including Name and Reward_Points__c
query = f"SELECT Id, Name, Email__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c='{email}' AND Password__c='{password}'"
result = sf.query(query)
if result["records"]:
user = result["records"][0]
session['user_id'] = user['Id']
session['user_email'] = email
print(f"Session variables set: user_id={session['user_id']}, user_email={session['user_email']}")
user_name = user.get("Name", "") # Get the user's name
reward_points = user.get("Reward_Points__c") or 0 # Ensures reward_points is always an integer
if reward_points >= 500:
print(f"User {email} has {reward_points} reward points. Generating coupon...")
# Generate a new coupon code
new_coupon_code = generate_coupon_code()
# Check if user already has a record in Referral_Coupon__c
coupon_query = sf.query(f"""
SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
""")
if coupon_query["records"]:
# If record exists, append the new coupon
coupon_record = coupon_query["records"][0]
referral_coupon_id = coupon_record["Id"]
existing_coupons = coupon_record.get("Coupon_Code__c", "")
# Append the new coupon on the next line
updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip()
# Update the Referral_Coupon__c record
sf.Referral_Coupon__c.update(referral_coupon_id, {
"Coupon_Code__c": updated_coupons
})
print(f"Updated existing coupon record for {email}. New Coupon: {new_coupon_code}")
else:
# If no record exists, create a new one with Referral_Name__c
sf.Referral_Coupon__c.create({
"Referral_Email__c": email,
"Name": user_name, # Store user's name in Referral_Coupon__c
"Coupon_Code__c": new_coupon_code
})
print(f"Created new coupon record for {email} with name {user_name}. Coupon: {new_coupon_code}")
# Subtract 500 reward points from user's account
new_reward_points = reward_points - 500
sf.Customer_Login__c.update(user['Id'], {
"Reward_Points__c": new_reward_points
})
print(f"Coupon {new_coupon_code} generated and 500 points deducted. New balance: {new_reward_points}")
return redirect(url_for("menu"))
else:
print("Invalid credentials!")
return render_template("login.html", error="Invalid credentials!")
except Exception as e:
print(f"Error during login: {str(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')
user_email = session.get('user_email') # Fetch the user's email
print(f"Session check in /menu: user_id={user_id}, user_email={user_email}")
if not user_id:
print("Session missing, redirecting to login.")
return redirect(url_for('login'))
try:
# Fetch the user's Referral__c and Reward_Points__c
user_query = f"SELECT Referral__c, Reward_Points__c FROM Customer_Login__c WHERE Email__c = '{user_email}'"
user_result = sf.query(user_query)
if not user_result['records']:
print("User not found!")
return redirect(url_for('login'))
referral_code = user_result['records'][0].get('Referral__c', 'N/A') # Default to 'N/A' if empty
reward_points = user_result['records'][0].get('Reward_Points__c', 0) # Default to 0 if empty
# Query to fetch menu items
menu_query = """
SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c
FROM Menu_Item__c
"""
result = sf.query(menu_query)
food_items = result['records'] if 'records' in result else []
# Filter categories dynamically
categories = {item.get("Veg_NonVeg__c").capitalize() for item in food_items if item.get("Veg_NonVeg__c")}
categories = {"Veg", "Non-Veg"} # Ensure valid categories only
# Filter food items based on the selected category
if selected_category == "Veg":
food_items = [item for item in food_items if item.get("Veg_NonVeg__c") in ["Veg", "both"]]
elif selected_category == "Non-Veg":
food_items = [item for item in food_items if item.get("Veg_NonVeg__c") in ["Non veg", "both"]]
except Exception as e:
print(f"Error fetching menu data: {str(e)}")
food_items = []
categories = {"All", "Veg", "Non-Veg"} # Default categories on error
referral_code = 'N/A'
reward_points = 0
# Render the menu page with the fetched data
return render_template(
"menu.html",
food_items=food_items,
categories=sorted(categories), # Sort categories alphabetically if needed
selected_category=selected_category,
referral_code=referral_code,
reward_points=reward_points
)
@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"))
try:
# Query cart items
result = sf.query(f"""
SELECT Name, Price__c, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Image1__c, Instructions__c
FROM Cart_Item__c
WHERE Customer_Email__c = '{email}'
""")
cart_items = result.get("records", [])
# Subtotal should be the sum of all item prices in the cart
subtotal = sum(item['Price__c'] for item in cart_items)
# Fetch reward points
customer_result = sf.query(f"""
SELECT Reward_Points__c
FROM Customer_Login__c
WHERE Email__c = '{email}'
""")
reward_points = customer_result['records'][0].get('Reward_Points__c', 0) if customer_result['records'] else 0
# Fetch coupons for the user
coupon_result = sf.query(f"""
SELECT Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
""")
# Extract and split coupons into a list
if coupon_result["records"]:
raw_coupons = coupon_result["records"][0].get("Coupon_Code__c", "")
coupons = raw_coupons.split("\n") if raw_coupons else []
else:
coupons = []
return render_template(
"cart.html",
cart_items=cart_items,
subtotal=subtotal,
reward_points=reward_points,
customer_email=email,
coupons=coupons # Send coupons to template
)
except Exception as e:
print(f"Error fetching cart items: {e}")
return render_template("cart.html", cart_items=[], subtotal=0, reward_points=0, coupons=[])
@app.route('/cart/add', methods=['POST'])
def add_to_cart():
data = request.json # Extract JSON payload from frontend
item_name = data.get('itemName').strip() # Item name
item_price = data.get('itemPrice') # Base price of the item
item_image = data.get('itemImage') # Item image
addons = data.get('addons', []) # Add-ons array
instructions = data.get('instructions', '') # Special instructions
customer_email = session.get('user_email') # Get logged-in user's email
if not item_name or not item_price:
return jsonify({"success": False, "error": "Item name and price are required."})
try:
# Query the cart to check if the item already exists
query = f"""
SELECT Id, Quantity__c, Add_Ons__c, Add_Ons_Price__c, Instructions__c FROM Cart_Item__c
WHERE Customer_Email__c = '{customer_email}' AND Name = '{item_name}'
"""
result = sf.query(query)
cart_items = result.get("records", [])
# Calculate the price of the new add-ons
addons_price = sum(addon['price'] for addon in addons) # New add-ons price
new_addons = "; ".join([f"{addon['name']} (${addon['price']})" for addon in addons]) # Format new add-ons
if cart_items:
# If the item already exists in the cart, update it
cart_item_id = cart_items[0]['Id']
existing_quantity = cart_items[0]['Quantity__c']
existing_addons = cart_items[0].get('Add_Ons__c', "None") # Previous add-ons
existing_addons_price = cart_items[0].get('Add_Ons_Price__c', 0) # Previous add-ons price
existing_instructions = cart_items[0].get('Instructions__c', "") # Previous instructions
# Combine the existing and new add-ons
combined_addons = existing_addons if existing_addons != "None" else ""
if new_addons:
combined_addons = f"{combined_addons}; {new_addons}".strip("; ")
# Combine the existing and new instructions
combined_instructions = existing_instructions
if instructions:
combined_instructions = f"{combined_instructions} | {instructions}".strip(" | ")
# Recalculate the total add-ons price
combined_addons_list = combined_addons.split("; ")
combined_addons_price = sum(
float(addon.split("($")[1][:-1]) for addon in combined_addons_list if "($" in addon
)
# Update the item in the cart
sf.Cart_Item__c.update(cart_item_id, {
"Quantity__c": existing_quantity + 1, # Increase quantity by 1
"Add_Ons__c": combined_addons, # Update add-ons list
"Add_Ons_Price__c": combined_addons_price, # Update add-ons price
"Instructions__c": combined_instructions, # Update instructions
"Price__c": (existing_quantity + 1) * item_price + combined_addons_price, # Update total price
})
else:
# If the item does not exist in the cart, create a new one
addons_string = "None"
if addons:
addons_string = new_addons # Use the formatted add-ons string
total_price = item_price + addons_price # Base price + add-ons price
# Create a new cart item
sf.Cart_Item__c.create({
"Name": item_name, # Item name
"Price__c": total_price, # Total price (item + add-ons)
"Base_Price__c": item_price, # Base price without add-ons
"Quantity__c": 1, # Default quantity is 1
"Add_Ons_Price__c": addons_price, # Total add-ons price
"Add_Ons__c": addons_string, # Add-ons with names and prices
"Image1__c": item_image, # Item image URL
"Customer_Email__c": customer_email, # Associated customer's email
"Instructions__c": instructions # Save instructions
})
return jsonify({"success": True, "message": "Item added to cart successfully."})
except Exception as e:
print(f"Error adding item to cart: {str(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', 1) # Quantity to add (default is 1)
addons = data.get('addons', []) # Add-ons for the item (optional)
# Validate inputs
if not email or not item_name:
return jsonify({"success": False, "error": "Email and item name are required."}), 400
try:
# Add a new item to the cart with the provided details
sf.Cart_Item__c.create({
"Customer_Email__c": email, # Associate the cart item with the customer's email
"Item_Name__c": item_name, # Item name
"Quantity__c": quantity, # Quantity to add
"Add_Ons__c": addons_string
})
return jsonify({"success": True, "message": "Item added to cart successfully."})
except Exception as e:
print(f"Error adding item to cart: {str(e)}") # Log the error for debugging
return jsonify({"success": False, "error": str(e)}), 500
@app.route('/cart/remove/', 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') # Fetch the requested item name
if not item_name:
return jsonify({"success": False, "error": "Item name is required."})
try:
# Fetch add-ons related to the item (update query as needed)
query = f"""
SELECT Name, Price__c
FROM Add_Ons__c
"""
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')
item_name = data.get('item_name')
try:
# Convert quantity to an integer
quantity = int(data.get('quantity'))
except (ValueError, TypeError):
return jsonify({"success": False, "error": "Invalid quantity provided."}), 400
# Validate inputs
if not email or not item_name or quantity is None:
return jsonify({"success": False, "error": "Email, item name, and quantity are required."}), 400
try:
# Query the cart item in Salesforce
cart_items = sf.query(
f"SELECT Id, Quantity__c, Price__c, Base_Price__c, Add_Ons_Price__c FROM Cart_Item__c "
f"WHERE Customer_Email__c = '{email}' AND Name = '{item_name}'"
)['records']
if not cart_items:
return jsonify({"success": False, "error": "Cart item not found."}), 404
# Retrieve the first matching record
cart_item_id = cart_items[0]['Id']
base_price = cart_items[0]['Base_Price__c']
addons_price = cart_items[0].get('Add_Ons_Price__c', 0)
# Calculate the new item price
new_item_price = (base_price * quantity) + addons_price
# Update the record in Salesforce
sf.Cart_Item__c.update(cart_item_id, {
"Quantity__c": quantity,
"Price__c": new_item_price, # Update base price
})
# Recalculate the subtotal for all items in the cart
cart_items = sf.query(f"""
SELECT Price__c, Add_Ons_Price__c
FROM Cart_Item__c
WHERE Customer_Email__c = '{email}'
""")['records']
new_subtotal = sum(item['Price__c'] for item in cart_items)
# Return updated item price and subtotal
return jsonify({"success": True, "new_item_price": new_item_price, "subtotal": new_subtotal})
print(f"New item price: {new_item_price}, New subtotal: {new_subtotal}")
return jsonify({"success": True, "new_item_price": new_item_price, "subtotal": new_subtotal})
except Exception as e:
print(f"Error updating quantity: {str(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:
data = request.json
selected_coupon = data.get("selectedCoupon", "").strip()
# Fetch cart items
result = sf.query(f"""
SELECT Id, Name, Price__c, Add_Ons_Price__c, Quantity__c, Add_Ons__c, Instructions__c, Image1__c
FROM Cart_Item__c
WHERE Customer_Email__c = '{email}'
""")
cart_items = result.get("records", [])
if not cart_items:
return jsonify({"success": False, "message": "Cart is empty"})
total_price = sum(item['Price__c'] for item in cart_items)
discount = 0
# Fetch the user's existing coupons
coupon_query = sf.query(f"""
SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'
""")
has_coupons = bool(coupon_query["records"]) # Check if user has any coupons
if selected_coupon:
# Case 3: User selected a valid coupon → Apply discount & remove coupon
discount = total_price * 0.10 # 10% discount
referral_coupon_id = coupon_query["records"][0]["Id"]
existing_coupons = coupon_query["records"][0]["Coupon_Code__c"].split("\n")
# Remove only the selected coupon
updated_coupons = [coupon for coupon in existing_coupons if coupon.strip() != selected_coupon]
# Convert list back to a string with newlines
updated_coupons_str = "\n".join(updated_coupons).strip()
# Update the Referral_Coupon__c record with the remaining coupons
sf.Referral_Coupon__c.update(referral_coupon_id, {
"Coupon_Code__c": updated_coupons_str
})
else:
# Case 1 & Case 2: User has no coupons or has coupons but didn’t select one → Add 10% to reward points
reward_points_to_add = total_price * 0.10
# Fetch current reward points
customer_record = sf.query(f"""
SELECT Id, Reward_Points__c FROM Customer_Login__c
WHERE Email__c = '{email}'
""")
customer = customer_record.get("records", [])[0] if customer_record else None
if customer:
current_reward_points = customer.get("Reward_Points__c") or 0
new_reward_points = current_reward_points + reward_points_to_add
print(f"Updating reward points: Current = {current_reward_points}, Adding = {reward_points_to_add}, New = {new_reward_points}")
# Update reward points in Salesforce
sf.Customer_Login__c.update(customer["Id"], {
"Reward_Points__c": new_reward_points
})
print(f"Successfully updated reward points for {email}")
total_bill = total_price - discount
# ✅ Store **all details** including Add-Ons, Instructions, Price, and Image
order_details = "\n".join(
f"{item['Name']} x{item['Quantity__c']} | Add-Ons: {item.get('Add_Ons__c', 'None')} | "
f"Instructions: {item.get('Instructions__c', 'None')} | "
f"Price: ${item['Price__c']} | Image: {item['Image1__c']}"
for item in cart_items
)
# Store the order details in Order__c
order_data = {
"Customer_Name__c": user_id,
"Customer_Email__c": email,
"Total_Amount__c": total_price,
"Discount__c": discount,
"Total_Bill__c": total_bill,
"Order_Status__c": "Pending",
"Order_Details__c": order_details # ✅ Now includes **all details**
}
sf.Order__c.create(order_data)
# ✅ Delete cart items after order is placed
for item in cart_items:
sf.Cart_Item__c.delete(item["Id"])
return jsonify({"success": True, "message": "Order placed successfully!"})
except Exception as e:
print(f"Error during checkout: {str(e)}")
return jsonify({"success": False, "error": str(e)})
@app.route("/order", methods=["GET"])
def order_summary():
email = session.get('user_email') # Fetch logged-in user's email
if not email:
return redirect(url_for("login"))
try:
# Fetch the most recent order for the user
result = sf.query(f"""
SELECT Id, Customer_Name__c, Customer_Email__c, Total_Amount__c, Order_Details__c, Order_Status__c, Discount__c, Total_Bill__c
FROM Order__c
WHERE Customer_Email__c = '{email}'
ORDER BY CreatedDate DESC
LIMIT 1
""")
order = result.get("records", [])[0] if result.get("records") else None
if not order:
return render_template("order.html", order=None)
return render_template("order.html", order=order)
except Exception as e:
print(f"Error fetching order details: {str(e)}")
return render_template("order.html", order=None, error=str(e))
if __name__ == "__main__":
app.run(debug=False, host="0.0.0.0", port=7860)