Update app.py

app.py

@@ -1,110 +1,63 @@
-from flask import Flask, render_template, request, jsonify, redirect, url_for
-from simple_salesforce import Salesforce
+from flask import Flask, render_template, request, redirect, url_for, jsonify
+from models.salesforce import fetch_menu_items, place_order_in_salesforce
+from models.cart import add_to_cart, view_cart, clear_cart
+from models.user import login_user, signup_user
 
-# Initialize Flask App
 app = Flask(__name__)
 
-# Salesforce Connection
-sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
-
-# Global Cart
-cart = []
-
-# Home Page (Login)
 @app.route("/", methods=["GET", "POST"])
 def login():
     if request.method == "POST":
-        email = request.form.get("email").strip()
-        password = request.form.get("password").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 render_template("login.html", error="Invalid email or password.")
-
-        user = result['records'][0]
-        stored_password = user['Password__c']
-
-        if password == stored_password:  # Simplified password verification for demo
+        email = request.form.get("email")
+        password = request.form.get("password")
+        success, message = login_user(email, password)
+        if success:
             return redirect(url_for("menu"))
-        else:
-            return render_template("login.html", error="Invalid email or password.")
+        return render_template("login.html", error=message)
     return render_template("login.html")
 
-# Signup Page
 @app.route("/signup", methods=["GET", "POST"])
 def signup():
     if request.method == "POST":
-        name = request.form.get("name").strip()
-        email = request.form.get("email").strip()
-        phone = request.form.get("phone").strip()
-        password = request.form.get("password").strip()
-
-        query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}'"
-        result = sf.query(query)
-
-        if len(result['records']) > 0:
-            return render_template("signup.html", error="Email already exists!")
-
-        sf.Customer_Login__c.create({
-            'Name': name,
-            'Email__c': email,
-            'Phone_Number__c': phone,
-            'Password__c': password  # Store securely in production
-        })
-        return redirect(url_for("login"))
+        name = request.form.get("name")
+        email = request.form.get("email")
+        phone = request.form.get("phone")
+        password = request.form.get("password")
+        success, message = signup_user(name, email, phone, password)
+        if success:
+            return redirect(url_for("login"))
+        return render_template("signup.html", error=message)
     return render_template("signup.html")
 
-# Menu Page
 @app.route("/menu")
 def menu():
-    query = "SELECT Name, Price__c, Description__c, Image1__c, Veg_NonVeg__c, Section__c FROM Menu_Item__c"
-    result = sf.query(query)
-    menu_items = result['records']
+    menu_items = fetch_menu_items()
     return render_template("menu.html", menu_items=menu_items)
 
-# Add to Cart
 @app.route("/add_to_cart", methods=["POST"])
-def add_to_cart():
-    global cart
+def add_to_cart_route():
     data = request.json
-    item_name = data.get("name")
-    item_price = data.get("price")
-
-    # Add item to cart
-    cart.append({"Name": item_name, "Price": item_price, "Quantity": 1})
-    return jsonify({"message": f"{item_name} added to cart!"})
+    cart = add_to_cart(data["name"], data["price"])
+    return jsonify({"message": "Item added to cart!", "cart": cart})
 
-# View Cart
 @app.route("/cart")
-def view_cart():
-    global cart
-    total = sum(float(item["Price"]) * item["Quantity"] for item in cart)
+def cart_page():
+    cart, total = view_cart()
     return render_template("cart.html", cart=cart, total=total)
 
-# Place Order
 @app.route("/place_order", methods=["POST"])
 def place_order():
-    global cart
-    email = request.form.get("email").strip()
-
+    email = request.form.get("email")
+    cart, total = view_cart()
     if not cart:
         return render_template("cart.html", error="Cart is empty!", cart=cart, total=0)
-
     order_details = "\n".join([f"{item['Name']} - ${item['Price']} x {item['Quantity']}" for item in cart])
-    total = sum(float(item["Price"]) * item["Quantity"] for item in cart)
-
     try:
-        sf.Order__c.create({
-            'Customer_Email__c': email,
-            'Order_Items__c': order_details,
-            'Total_Amount__c': total
-        })
-        cart.clear()
-        return render_template("cart.html", success=f"Order placed successfully! Total: ${total}", cart=cart, total=0)
+        place_order_in_salesforce(email, order_details, total)
+        clear_cart()
+        return render_template("cart.html", success="Order placed successfully!", cart=[], total=0)
     except Exception as e:
-        return render_template("cart.html", error=f"Error placing order: {str(e)}", cart=cart, total=total)
+        return render_template("cart.html", error=f"Error: {str(e)}", cart=cart, total=total)
 
 if __name__ == "__main__":
     app.run(debug=True)