import bcrypt
import gradio as gr
from simple_salesforce import Salesforce
# Salesforce Connection
sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
# Function to Hash Password
def hash_password(password):
return bcrypt.hashpw(password.encode('utf-8'), bcrypt.gensalt()).decode('utf-8')
# Function to Verify Password
def verify_password(plain_password, hashed_password):
return bcrypt.checkpw(plain_password.encode('utf-8'), hashed_password.encode('utf-8'))
# Signup function
def signup(name, email, phone, password):
try:
email = email.strip()
query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
result = sf.query(query)
if len(result['records']) > 0:
return "Email already exists! Please use a different email."
hashed_password = hash_password(password)
sf.Customer_Login__c.create({
'Name': name.strip(),
'Email__c': email,
'Phone_Number__c': phone.strip(),
'Password__c': hashed_password
})
return "Signup successful! You can now login."
except Exception as e:
return f"Error during signup: {str(e)}"
# Login function
def login(email, password):
try:
email = email.strip()
query = f"SELECT Name, Password__c FROM Customer_Login__c WHERE Email__c = '{email}'"
result = sf.query(query)
if len(result['records']) == 0:
return "Invalid email or password.", None
user = result['records'][0]
stored_password = user['Password__c']
if verify_password(password.strip(), stored_password):
return "Login successful!", user['Name']
else:
return "Invalid email or password.", None
except Exception as e:
return f"Error during login: {str(e)}", None
# Function to load menu data
def load_menu_from_salesforce():
try:
query = "SELECT Name, Price__c, Description__c, Image1__c, Image2__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
result = sf.query(query)
return result['records']
except Exception as e:
return []
# Function to load add-ons data
def load_add_ons_from_salesforce():
try:
query = "SELECT Name, Price__c FROM Add_Ons__c"
result = sf.query(query)
return result['records']
except Exception as e:
return []
# Function to filter menu items
def filter_menu(preference):
menu_data = load_menu_from_salesforce()
filtered_data = {}
for item in menu_data:
if "Section__c" not in item or "Veg_NonVeg__c" not in item:
continue
if item["Section__c"] not in filtered_data:
filtered_data[item["Section__c"]] = []
if preference == "All" or (preference == "Veg" and item["Veg_NonVeg__c"] in ["Veg", "Both"]) or (preference == "Non-Veg" and item["Veg_NonVeg__c"] in ["Non veg", "Both"]):
filtered_data[item["Section__c"].strip()].append(item)
html_content = '
'
for section, items in filtered_data.items():
html_content += f"
{section}
"
html_content += '
'
for item in items:
html_content += f"""
{item['Name']}
${item['Price__c']}
{item['Description__c']}
"""
html_content += '
'
html_content += '
'
if not any(filtered_data.values()):
return "
No items match your filter.
"
return html_content
# Generate final order summary
def generate_order_summary(cart):
if not cart:
return "
Your cart is empty.
"
summary_html = "
Order Summary
"
total_cost = 0
for item in cart:
item_cost = item['price'] * item['quantity']
add_ons_summary = "
" + "".join([f"
{addon['name']} x {addon['quantity']} (+${addon['price'] * addon['quantity']:.2f})
" for addon in item['extras']]) + "
"
total_item_cost = item_cost + sum(addon['price'] * addon['quantity'] for addon in item['extras'])
summary_html += f"""
{item['name']}
Quantity: {item['quantity']}
Special Instructions: {item['instructions'] or 'None'}