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 from datetime import timedelta from email.mime.multipart import MIMEMultipart from email.mime.text import MIMEText from menu import menu_blueprint # Make sure this import is correct from cart import cart_blueprint # Same for other blueprints from order import order_blueprint # Same for user blueprint from orderhistory import orderhistory_blueprint from user_details import user_details_blueprint from customdish import customdish_blueprint from combined_summary import combined_summary_blueprint from datetime import datetime from datetime import datetime import pytz # Library to handle timezone conversions import os import smtplib import random import string from flask import Flask, render_template, request, jsonify, send_from_directory from simple_salesforce import Salesforce from dotenv import load_dotenv import os import logging import uuid from datetime import datetime app = Flask(__name__) logging.basicConfig(level=logging.DEBUG) logger = logging.getLogger(__name__) # Add debug logs in Salesforce connection setup sf = get_salesforce_connection() # Set the secret key to handle sessions securely app.secret_key = os.getenv("SECRET_KEY", "PWEmya351XHeWQy0ZHbIvYm3") # Replace with a secure key app.config["SESSION_TYPE"] = "filesystem" # Storing sessions in filesystem app.config["SESSION_COOKIE_SECURE"] = True # Enabling secure cookies (ensure your app is served over HTTPS) app.config["SESSION_COOKIE_SAMESITE"] = "None" # Cross-site cookies allowed # Initialize the session Session(app) # Correctly initialize the Session object app.session_interface = SecureCookieSessionInterface() app.register_blueprint(cart_blueprint, url_prefix='/cart') app.register_blueprint(user_details_blueprint, url_prefix='/user') app.register_blueprint(menu_blueprint) app.register_blueprint(order_blueprint) app.register_blueprint(orderhistory_blueprint, url_prefix='/orderhistory') app.register_blueprint(customdish_blueprint, url_prefix='/customdish') app.register_blueprint(combined_summary_blueprint, url_prefix='/combined_summary') @app.route("/") def home(): # Fetch user details from URL parameters user_email = request.args.get("email") user_name = request.args.get("name") table_number = request.args.get("table") # Capture table number if user_email and user_name: session["user_email"] = user_email session["user_name"] = user_name session["table_number"] = table_number # Store table number in session print(f"User logged in: {user_email} - {user_name} - Table: {table_number}") # Ensure session is saved before redirecting session.modified = True return redirect(url_for("menu.menu")) # Redirect to menu directly return render_template("index.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 from Salesforce 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'] # ✅ Always store or update session email if 'user_email' not in session or session['user_email'] != email: session['user_email'] = email session['user_name'] = user.get("Name", "") print(f"✅ Session email updated: {session['user_email']}") reward_points = user.get("Reward_Points__c") or 0 # Coupon generation logic (if reward points >= 500) if reward_points >= 500: new_coupon_code = generate_coupon_code() coupon_query = sf.query(f"SELECT Id, Coupon_Code__c FROM Referral_Coupon__c WHERE Referral_Email__c = '{email}'") if coupon_query["records"]: coupon_record = coupon_query["records"][0] referral_coupon_id = coupon_record["Id"] existing_coupons = coupon_record.get("Coupon_Code__c", "") updated_coupons = f"{existing_coupons}\n{new_coupon_code}".strip() sf.Referral_Coupon__c.update(referral_coupon_id, {"Coupon_Code__c": updated_coupons}) else: sf.Referral_Coupon__c.create({ "Referral_Email__c": email, "Name": user.get("Name", ""), "Coupon_Code__c": new_coupon_code }) new_reward_points = reward_points - 500 sf.Customer_Login__c.update(user['Id'], {"Reward_Points__c": new_reward_points}) return redirect(url_for("menu.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("/logout") def logout(): # Retrieve table number before clearing session table_number = session.get('table_number', '') # Clear session variables session.pop('name', None) session.pop('email', None) session.pop('rewardPoints', None) session.pop('coupon', None) # Pass table number to redirect page return render_template("redirect_page.html", table_number=table_number) @app.route('/customdish') def customdish(): # Retrieve the user_name from the session user_name = session.get('user_name', None) # Pass user_name to the template if it exists, otherwise pass None return render_template('customdish.html', user_name=user_name) @app.route('/get_ingredients', methods=['POST']) def get_ingredients(): global sf if not sf: sf = get_salesforce_connection() if not sf: return jsonify({"error": "Unable to connect to Salesforce"}), 500 data = request.json dietary_preference = data.get('dietary_preference', 'both').lower() try: category_map = { 'vegetarian': 'Veg', 'non-vegetarian': 'Non-Veg', 'chicken': 'Non-Veg', 'beef': 'Non-Veg', 'lamb': 'Non-Veg', 'both': 'both' } category = category_map.get(dietary_preference, 'both') soql = f"SELECT Name, Image_URL__c, Category__c FROM Sector_Detail__c WHERE Category__c = '{category}'" soql += " LIMIT 200" logger.debug(f"Executing SOQL query for Sector_Detail__c: {soql}") result = sf.query(soql) ingredients = [ { "name": record['Name'], "image_url": record.get('Image_URL__c', ''), "category": record.get('Category__c', '') } for record in result['records'] if 'Name' in record ] logger.debug(f"Fetched {len(ingredients)} ingredients from Sector_Detail__c") return jsonify({"ingredients": ingredients}) except Exception as e: logger.error(f"Failed to fetch ingredients: {str(e)}") return jsonify({"error": f"Failed to fetch ingredients from Salesforce: {str(e)}"}), 500 @app.route('/get_menu_items', methods=['POST']) def get_menu_items(): global sf if not sf: sf = get_salesforce_connection() if not sf: return jsonify({"error": "Unable to connect to Salesforce"}), 500 data = request.json ingredient_names = data.get('ingredient_names', '') category = data.get('category', '') try: soql = "SELECT Name, Description__c, Image1__c, Image2__c, Price__c, Section__c, Veg_NonVeg__c, Total_Ordered__c FROM Menu_Item__c" conditions = [] if ingredient_names: words = ingredient_names.split() name_conditions = [f"Name LIKE '%{word}%'" for word in words] conditions.append(f"({' OR '.join(name_conditions)})") if category: if category.lower() == 'vegetarian': conditions.append("Veg_NonVeg__c = 'Vegetarian'") elif category.lower() == 'non-vegetarian': conditions.append("Veg_NonVeg__c = 'Non-Vegetarian'") if conditions: soql += " WHERE " + " AND ".join(conditions) soql += " LIMIT 200" logger.debug(f"Executing SOQL query for Menu_Item__c: {soql}") result = sf.query(soql) menu_items = [ { "name": record['Name'], "description": record.get('Description__c', 'No description available'), "image_url": record.get('Image1__c', '') or record.get('Image2__c', ''), "price": record.get('Price__c', 0.0), "section": record.get('Section__c', ''), "veg_nonveg": record.get('Veg_NonVeg__c', ''), "total_ordered": record.get('Total_Ordered__c', 0) } for record in result['records'] if 'Name' in record ] logger.debug(f"Fetched {len(menu_items)} menu items") return jsonify({"menu_items": menu_items}) except Exception as e: logger.error(f"Failed to fetch menu items: {str(e)}") return jsonify({"error": f"Failed to fetch menu items from Salesforce: {str(e)}"}), 500 @app.route('/submit_customization_ingredients', methods=['POST']) def submit_customization_ingredients(): global sf if not sf: sf = get_salesforce_connection() if not sf: return jsonify({"error": "Unable to connect to Salesforce"}), 500 data = request.json items = data.get('items', []) menu_item = data.get('menu_item', {}) ingredients = data.get('ingredients', []) instructions = data.get('instructions', '') # Get the session email (this should already be set during login) customer_email = session.get('user_email') if not customer_email: return jsonify({"error": "User email not found in session"}), 400 try: if items: # Cart submission for item in items: ingredient_names = ', '.join(i['name'] for i in item.get('ingredients', [])) if item.get('ingredients') else '' base_price = item.get('price', 0.0) quantity = 1 addons_price = 0 total_price = (base_price * quantity) + addons_price # Check if item already exists in the cart for the user based on Customer_Email__c and Item Name soql = f"SELECT Id, Quantity__c, Add_Ons__c, Instructions__c, Price__c FROM Cart_Item__c WHERE Customer_Email__c = '{customer_email}' AND Name = '{item['name']}' LIMIT 1" existing_item = sf.query(soql) if existing_item['records']: # If the item exists, update it item_id = existing_item['records'][0]['Id'] existing_quantity = existing_item['records'][0]['Quantity__c'] existing_addons = existing_item['records'][0]['Add_Ons__c'] existing_instructions = existing_item['records'][0]['Instructions__c'] existing_price = existing_item['records'][0]['Price__c'] # Update quantity, addons, and instructions new_quantity = existing_quantity + quantity new_addons = (existing_addons or '') + (', ' + ingredient_names if ingredient_names else '') new_instructions = (existing_instructions or '') + ('; ' + instructions if instructions else '') # Calculate the new total price updated_price = (base_price * new_quantity) + addons_price # Update the existing item in Salesforce sf.Cart_Item__c.update(item_id, { 'Quantity__c': new_quantity, 'Add_Ons__c': new_addons, 'Instructions__c': new_instructions, 'Price__c': updated_price }) logger.debug(f"Updated {item['name']} in Cart_Item__c") else: # If the item doesn't exist, create a new one sf.Cart_Item__c.create({ 'Name': item['name'], 'Base_Price__c': base_price, 'Quantity__c': quantity, 'Add_Ons__c': ingredient_names, 'Add_Ons_Price__c': addons_price, 'Price__c': total_price, 'Image1__c': item.get('image_url', ''), 'Instructions__c': item.get('instructions', ''), 'Category__c': item.get('veg_nonveg', ''), 'Section__c': item.get('section', ''), 'Customer_Email__c': customer_email # Store session email }) logger.debug(f"Created new Cart_Item__c for {item['name']}") return jsonify({"success": True, "message": f"Processed {len(items)} items"}) elif menu_item: # Single item customization ingredient_names = ', '.join(i['name'] for i in ingredients) if ingredients else '' base_price = menu_item.get('price', 0.0) quantity = 1 addons_price = 0 total_price = (base_price * quantity) + addons_price # Check if the menu item already exists in the cart for the user soql = f"SELECT Id, Quantity__c, Add_Ons__c, Instructions__c, Price__c FROM Cart_Item__c WHERE Customer_Email__c = '{customer_email}' AND Name = '{menu_item['name']}' LIMIT 1" existing_item = sf.query(soql) if existing_item['records']: # If the item exists, update it item_id = existing_item['records'][0]['Id'] existing_quantity = existing_item['records'][0]['Quantity__c'] existing_addons = existing_item['records'][0]['Add_Ons__c'] existing_instructions = existing_item['records'][0]['Instructions__c'] existing_price = existing_item['records'][0]['Price__c'] # Update quantity, addons, and instructions new_quantity = existing_quantity + quantity new_addons = (existing_addons or '') + (', ' + ingredient_names if ingredient_names else '') new_instructions = (existing_instructions or '') + ('; ' + instructions if instructions else '') # Calculate the new total price updated_price = (base_price * new_quantity) + addons_price # Update the existing item in Salesforce sf.Cart_Item__c.update(item_id, { 'Quantity__c': new_quantity, 'Add_Ons__c': new_addons, 'Instructions__c': new_instructions, 'Price__c': updated_price }) logger.debug(f"Updated customization for {menu_item['name']} in Cart_Item__c") else: # If the item doesn't exist, create a new one sf.Cart_Item__c.create({ 'Name': menu_item['name'], 'Base_Price__c': base_price, 'Quantity__c': quantity, 'Add_Ons__c': ingredient_names, 'Add_Ons_Price__c': addons_price, 'Price__c': total_price, 'Image1__c': menu_item.get('image_url', ''), 'Instructions__c': instructions, 'Category__c': menu_item.get('veg_nonveg', ''), 'Section__c': menu_item.get('section', ''), 'Customer_Email__c': customer_email # Store session email }) logger.debug(f"Created new Cart_Item__c for {menu_item['name']}") return jsonify({"success": True, "message": "Customization submitted"}) else: return jsonify({"error": "No items or menu item provided"}), 400 except Exception as e: logger.error(f"Failed to submit: {str(e)}") return jsonify({"error": f"Failed to submit: {str(e)}"}), 500 if __name__ == "__main__": app.run(debug=True, host="0.0.0.0", port=7860)