Update app.py

app.py

@@ -3,40 +3,42 @@ 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
-import os
-import smtplib
 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 user_details import user_details_blueprint 
-# Initialize Flask app and Salesforce connection
-print("Starting app...")
+from user_details import user_details_blueprint
+from datetime import datetime
+from datetime import datetime
+import pytz  # Library to handle timezone conversions
+import os
+import smtplib
+import random
+import string
+
 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", "xEr0cwgsiatzrzaeFewYrVA1O")  # 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
-print("Session interface configured.")
-app.register_blueprint(user_details_blueprint, url_prefix='/user')
-# Ensure secure session handling for environments like Hugging Face
 app.session_interface = SecureCookieSessionInterface()
-print("Session interface configured.")
-import random
-import string
+
 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.route("/")
 def home():
@@ -57,11 +59,6 @@ def home():
 
     return render_template("index.html")
 
-
-
-from datetime import datetime
-
-
 @app.route("/generate_custom_dish", methods=["POST"])
 def generate_custom_dish():
     try:
@@ -158,12 +155,6 @@ def generate_custom_dish():
     except Exception as e:
         return jsonify({"success": False, "error": str(e)}), 500
 
-
- 
-
-from datetime import datetime
-import pytz  # Library to handle timezone conversions
-
 @app.route("/order-history", methods=["GET"])
 def order_history():
     email = session.get('user_email')  # Get logged-in user's email
@@ -262,68 +253,10 @@ def logout():
     return render_template("redirect_page.html", table_number=table_number)
 
 
-@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")
-# Register Blueprints for each functionality
-app.register_blueprint(menu_blueprint)
- # Register the cart blueprint with URL prefix
 
-app.register_blueprint(order_blueprint)
+
+
+
 
 
 if __name__ == "__main__":