Update app.py

app.py

@@ -43,19 +43,6 @@ def generate_audio_prompt(text, filename):
 for key, text in prompts.items():
     generate_audio_prompt(text, f"{key}.mp3")
 
-# Symbol mapping for proper recognition
-SYMBOL_MAPPING = {
-    "at the rate": "@",
-    "at": "@",
-    "dot": ".",
-    "underscore": "_",
-    "hash": "#",
-    "plus": "+",
-    "dash": "-",
-    "comma": ",",
-    "space": " "
-}
-
 # Function to convert audio to WAV format
 def convert_to_wav(input_path, output_path):
     try:
@@ -71,7 +58,7 @@ def is_silent_audio(audio_path):
     nonsilent_parts = detect_nonsilent(audio, min_silence_len=500, silence_thresh=audio.dBFS-16)
     return len(nonsilent_parts) == 0
 
-# Salesforce connection details
+# Connect to Salesforce
 try:
     print("Attempting to connect to Salesforce...")
     sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
@@ -79,82 +66,71 @@ try:
 except Exception as e:
     print(f"Failed to connect to Salesforce: {str(e)}")
 
-# Function to handle login & registration in Salesforce
-@app.route("/validate_login", methods=["POST"])
-def validate_login():
-    try:
-        # Get the email and mobile number from the request
-        data = request.json
-        email = data.get("email")
-        mobile = data.get("mobile")
-
-        # Salesforce query to check if the email and mobile exist
-        query = f"SELECT Id, Name FROM Customer_Login__c WHERE Email__c = '{email}' AND Phone_Number__c = '{mobile}'"
-        result = sf.query(query)
-
-        if result['totalSize'] > 0:
-            return jsonify({'success': True, 'message': 'User authenticated successfully.'}), 200
-        else:
-            return jsonify({'success': False, 'error': 'Invalid email or mobile number.'}), 400
-    except Exception as e:
-        logging.error(f"Error: {str(e)}")
-        return jsonify({'error': 'Something went wrong. Please try again later.'}), 500
-
-if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
-
+# ✅ LOGIN ENDPOINT (Validates User)
+@app.route('/login', methods=['POST'])
+def login():
+    data = request.json
+    email = data.get('email').strip().lower()
+    phone_number = data.get('phone_number').strip()
 
+    if not email or not phone_number:
+        return jsonify({'error': 'Missing email or phone number'}), 400
 
-# Initialize Flask app
-app = Flask(__name__)
+    try:
+        # Check if user exists in Salesforce
+        query = f"SELECT Id, Name FROM Customer_Login__c WHERE LOWER(Email__c) = '{email}' AND Phone_Number__c = '{phone_number}' LIMIT 1"
+        result = sf.query(query)
 
-# Set the secret key to handle sessions securely
-app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")  # Replace with a secure key
+        if result['totalSize'] == 0:
+            return jsonify({'error': 'Invalid email or phone number. User not found'}), 401
 
-# Configure the session type
-app.config["SESSION_TYPE"] = "filesystem"  # Use filesystem for session storage
-app.config["SESSION_COOKIE_NAME"] = "my_session"  # Optional: Change session cookie name
-app.config["SESSION_COOKIE_SECURE"] = True  # Ensure cookies are sent over HTTPS
-app.config["SESSION_COOKIE_SAMESITE"] = "None"  # Allow cross-site cookies
+        user_data = result['records'][0]
+        return jsonify({'success': True, 'message': 'Login successful', 'user_id': user_data['Id'], 'name': user_data['Name']}), 200
 
-# Initialize the session
-Session(app)
+    except requests.exceptions.RequestException as req_error:
+        return jsonify({'error': f'Salesforce connection error: {str(req_error)}'}), 500
+    except Exception as e:
+        return jsonify({'error': f'Unexpected error: {str(e)}'}), 500
 
-# Set up logging
-logging.basicConfig(level=logging.INFO)
+# ✅ REGISTRATION ENDPOINT (Creates New User)
+@app.route("/submit", methods=["POST"])
+def submit():
+    data = request.json
+    name = data.get('name')
+    email = data.get('email').strip().lower()
+    phone = data.get('phone').strip()
 
-@app.route("/")
-def index():
-    # Serve the HTML page for the voice-based login
-    return render_template("index.html")
+    if not name or not email or not phone:
+        return jsonify({'error': 'Missing data'}), 400
 
-@app.route("/capture_email_and_mobile", methods=["POST"])
-def capture_email_and_mobile():
     try:
-        # Get the voice captured email and mobile number from the request
-        data = request.json
-        email = data.get("email")
-        mobile = data.get("mobile")
-
-        # Validate the captured email and mobile number
-        if not email or not mobile:
-            return jsonify({"error": "Email or mobile number is missing."}), 400
-
-        # Log the captured data for now (you can replace it with actual processing logic)
-        logging.info(f"Captured Email: {email}, Mobile: {mobile}")
-        
-        # For simplicity, we'll assume the capture was successful.
-        return jsonify({"success": True, "message": "Email and mobile captured successfully."}), 200
+        # Check if user already exists
+        query = f"SELECT Id FROM Customer_Login__c WHERE LOWER(Email__c) = '{email}' AND Phone_Number__c = '{phone}' LIMIT 1"
+        existing_user = sf.query(query)
+
+        if existing_user['totalSize'] > 0:
+            return jsonify({'error': 'User already exists'}), 409  # Conflict
+
+        # Create new user
+        customer_login = sf.Customer_Login__c.create({
+            'Name': name,
+            'Email__c': email,
+            'Phone_Number__c': phone
+        })
+
+        if customer_login.get('id'):
+            return jsonify({'success': True, 'user_id': customer_login['id']}), 200
+        else:
+            return jsonify({'error': 'Failed to create record'}), 500
 
     except Exception as e:
-        logging.error(f"Error: {str(e)}")
-        return jsonify({"error": "Something went wrong while processing."}), 500
-
+        return jsonify({'error': str(e)}), 500
 
 @app.route("/")
 def index():
     return render_template("index.html")
 
+# ✅ TRANSCRIPTION ENDPOINT (Converts Speech to Text)
 @app.route("/transcribe", methods=["POST"])
 def transcribe():
     if "audio" not in request.files: