Update app.py

app.py

@@ -1,6 +1,6 @@
 
 import torch
-from flask import Flask, render_template, request, jsonify
+from flask import Flask, render_template, request, jsonify, redirect, session
 import json
 import os
 from transformers import pipeline
@@ -13,115 +13,89 @@ from waitress import serve
 from simple_salesforce import Salesforce
 import requests
 
+# Initialize Flask app
 app = Flask(__name__)
+app.secret_key = os.urandom(24)  # For session handling
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
-config = AutoConfig.from_pretrained("openai/whisper-small")
-config.update({"timeout": 60})
-
-def generate_audio_prompt(text, filename):
-    try:
-        tts = gTTS(text)
-        tts.save(os.path.join("static", filename))
-    except gtts.tts.gTTSError as e:
-        print(f"Error: {e}")
-        time.sleep(5)
-        generate_audio_prompt(text, filename)
-
-prompts = {
-    "welcome": "Welcome to Biryani Hub.",
-    "ask_name": "Tell me your name.",
-    "ask_email": "Please provide your email address.",
-    "thank_you": "Thank you for registration."
-}
-
-for key, text in prompts.items():
-    generate_audio_prompt(text, f"{key}.mp3")
-
+# ✅ Salesforce Connection Setup
 try:
     print("Attempting to connect to Salesforce...")
-    sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
-    print("Connected to Salesforce successfully!")
+    sf = Salesforce(username='diggavalli98@gmail.com', 
+                    password='Sati@1020', 
+                    security_token='sSSjyhInIsUohKpG8sHzty2q')
+    print("✅ Connected to Salesforce successfully!")
 except Exception as e:
-    print(f"Failed to connect to Salesforce: {str(e)}")
-
-@app.route("/")
-def index():
-    return render_template("index.html")
-
-@app.route("/dashboard")
-def dashboard():
-    return render_template("dashboard.html")
-
-@app.route("/menu")
-def menu():
-    menu_items = [
-        {"id": 1, "name": "Chicken Biryani", "price": 250},
-        {"id": 2, "name": "Mutton Biryani", "price": 350},
-        {"id": 3, "name": "Veg Biryani", "price": 200},
-        {"id": 4, "name": "Paneer Butter Masala", "price": 180},
-        {"id": 5, "name": "Butter Naan", "price": 50}
-    ]
-    return render_template("menu_page.html", menu_items=menu_items)
-
-@app.route("/cart")
-def cart():
-    return render_template("cart.html")
+    print(f"❌ Failed to connect to Salesforce: {str(e)}")
 
-@app.route("/order")
-def order():
-    return render_template("order.html")
+# ✅ REGISTER API: Create a new Customer in Salesforce
+@app.route("/register", methods=["POST"])
+def register():
+    data = request.json
+    name = data.get("name")
+    email = data.get("email")
+    phone_number = data.get("phone")
 
-@app.route("/final")
-def final():
-    return render_template("final.html")
+    if not name or not email or not phone_number:
+        return jsonify({"error": "Missing required fields"}), 400
 
-@app.route("/register", methods=["GET", "POST"])
-def register():
-    if request.method == "POST":
-        data = request.form
-        name = data.get('name')
-        email = data.get('email')
-        phone = data.get('phone')
-        password = data.get('password')
+    try:
+        new_customer = sf.Customer_Login__c.create({
+            'Name': name,
+            'Email__c': email,
+            'Phone_Number__c': phone_number
+        })
+        return jsonify({"success": True, "message": "Registration successful", "customer_id": new_customer["id"]})
+    except Exception as e:
+        return jsonify({"error": f"Failed to register: {str(e)}"}), 500
+
+# ✅ LOGIN API: Validate user credentials
+@app.route("/login", methods=["POST"])
+def login():
+    data = request.json
+    email = data.get("email")
+    phone_number = data.get("phone")
 
-        if not name or not email or not phone or not password:
-            return jsonify({'error': 'Missing required fields'}), 400
+    if not email or not phone_number:
+        return jsonify({"error": "Missing email or phone"}), 400
 
-        try:
-            customer_register = sf.Customer_Login__c.create({
-                'Name': name,
-                'Email__c': email,
-                'Phone_Number__c': phone,
-                'Password__c': password
+    try:
+        query = f"SELECT Id, Name FROM Customer_Login__c WHERE Email__c = '{email}' AND Phone_Number__c = '{phone_number}'"
+        result = sf.query(query)
+
+        if result["totalSize"] > 0:
+            user = result["records"][0]
+            session["customer_id"] = user["Id"]  # Store in session
+            return jsonify({"success": True, "message": "Login successful", "customer_id": user["Id"], "name": user["Name"]})
+        else:
+            return jsonify({"error": "Invalid login credentials"}), 401
+    except Exception as e:
+        return jsonify({"error": f"Login failed: {str(e)}"}), 500
+
+# ✅ MENU API: Fetch menu items from Salesforce
+@app.route("/menu", methods=["GET"])
+def get_menu():
+    try:
+        query = "SELECT Name, Price__c, Ingredients__c, Category__c FROM Menu_Item__c"
+        result = sf.query(query)
+
+        menu_items = []
+        for item in result["records"]:
+            menu_items.append({
+                "name": item["Name"],
+                "price": item["Price__c"],
+                "ingredients": item["Ingredients__c"],
+                "category": item["Category__c"]
             })
-            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")
+        return jsonify({"success": True, "menu": menu_items})
+    except Exception as e:
+        return jsonify({"error": f"Failed to fetch menu: {str(e)}"}), 500
 
-        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")
+# ✅ DASHBOARD ROUTE
+@app.route("/dashboard", methods=["GET"])
+def dashboard():
+    return render_template("dashboard.html")
 
+# ✅ START PRODUCTION SERVER
 if __name__ == "__main__":
     serve(app, host="0.0.0.0", port=7860)