Update app.py

app.py

@@ -76,32 +76,52 @@ def order():
 def final():
     return render_template("final.html")
 
-@app.route("/register", methods=["POST"])
+@app.route("/register", methods=["GET", "POST"])
 def register():
-    data = request.json
-    name = data.get('name')
-    email = data.get('email')
-    phone = data.get('phone')
-    password = data.get('password')
-
-    if not name or not email or not phone or not password:
-        return jsonify({'error': 'Missing required fields'}), 400
-
-    try:
-        customer_register = sf.Customer_Login__c.create({
-            'Name': name,
-            'Email__c': email,
-            'Phone_Number__c': phone,
-            'Password__c': password
-        })
-
-        if customer_register.get('id'):
-            return jsonify({'success': True, 'id': customer_register['id']}), 200
-        else:
-            return jsonify({'error': 'Failed to create record'}), 500
-
-    except Exception as e:
-        return jsonify({'error': str(e)}), 500
+    if request.method == "POST":
+        data = request.form
+        name = data.get('name')
+        email = data.get('email')
+        phone = data.get('phone')
+        password = data.get('password')
+
+        if not name or not email or not phone or not password:
+            return jsonify({'error': 'Missing required fields'}), 400
+
+        try:
+            customer_register = sf.Customer_Login__c.create({
+                'Name': name,
+                'Email__c': email,
+                'Phone_Number__c': phone,
+                'Password__c': password
+            })
+            return render_template("register_success.html", success=True)
+        except Exception as e:
+            return render_template("register.html", error=str(e))
+    
+    return render_template("register.html")
+
+@app.route("/login", methods=["GET", "POST"])
+def login():
+    if request.method == "POST":
+        data = request.form
+        email = data.get('email')
+        password = data.get('password')
+
+        if not email or not password:
+            return render_template("login.html", error="Missing email or password")
+
+        try:
+            query = f"SELECT Id FROM Customer_Login__c WHERE Email__c = '{email}' AND Password__c = '{password}'"
+            result = sf.query(query)
+            if result['totalSize'] > 0:
+                return render_template("dashboard.html", success=True)
+            else:
+                return render_template("login.html", error="Invalid credentials")
+        except Exception as e:
+            return render_template("login.html", error=str(e))
+    
+    return render_template("login.html")
 
 if __name__ == "__main__":
     serve(app, host="0.0.0.0", port=7860)