Update app.py

app.py

@@ -1,9 +1,7 @@
 import torch
 from flask import Flask, render_template, request, jsonify
-from flask_session import Session
 import json
 import os
-import re
 from transformers import pipeline
 from gtts import gTTS
 from pydub import AudioSegment
@@ -13,8 +11,6 @@ import time
 from waitress import serve
 from simple_salesforce import Salesforce
 import requests
-import logging
-
 
 app = Flask(__name__)
 
@@ -47,6 +43,19 @@ 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:
@@ -62,7 +71,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
 
-# Connect to Salesforce
+# Salesforce connection details
 try:
     print("Attempting to connect to Salesforce...")
     sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
@@ -70,80 +79,82 @@ try:
 except Exception as e:
     print(f"Failed to connect to Salesforce: {str(e)}")
 
-# ✅ LOGIN ENDPOINT (Validates User)
-@app.route('/login', methods=['POST'])
-def login():
-    data = request.json
-    email = data.get('email').strip().lower()  # Ensure no leading/trailing spaces
-    phone_number = data.get('phone_number').strip()
+# 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")
 
-    # Sanitize phone number by removing non-numeric characters
-    phone_number = re.sub(r'\D', '', phone_number)
+        # 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)
 
-    # Check if email or phone number is missing
-    if not email or not phone_number:
-        return jsonify({'error': 'Missing email or phone number'}), 400
+        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
 
-    try:
-        # Log the query for debugging
-        query = f"SELECT Id, Name FROM Customer_Login__c WHERE LOWER(Email__c) = '{email}' AND Phone_Number__c = '{phone_number}' LIMIT 1"
-        logging.info(f"Executing query: {query}")
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860, debug=True)
 
-        # Query Salesforce to check if user exists
-        result = sf.query(query)
 
-        if result['totalSize'] == 0:
-            return jsonify({'error': 'Invalid email or phone number. User not found'}), 401
 
-        user_data = result['records'][0]
-        return jsonify({'success': True, 'message': 'Login successful', 'user_id': user_data['Id'], 'name': user_data['Name']}), 200
+# Initialize Flask app
+app = Flask(__name__)
 
-    except requests.exceptions.RequestException as req_error:
-        logging.error(f"Salesforce connection error: {str(req_error)}")
-        return jsonify({'error': f'Salesforce connection error: {str(req_error)}'}), 500
-    except Exception as e:
-        logging.error(f"Unexpected error: {str(e)}")
-        return jsonify({'error': f'Unexpected error: {str(e)}'}), 500
+# Set the secret key to handle sessions securely
+app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")  # Replace with a secure key
+
+# 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
+
+# Initialize the session
+Session(app)
 
-# ✅ 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()
+# Set up logging
+logging.basicConfig(level=logging.INFO)
 
-    if not name or not email or not phone:
-        return jsonify({'error': 'Missing data'}), 400
+@app.route("/")
+def index():
+    # Serve the HTML page for the voice-based login
+    return render_template("index.html")
 
+@app.route("/capture_email_and_mobile", methods=["POST"])
+def capture_email_and_mobile():
     try:
-        # 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
+        # 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
 
     except Exception as e:
-        return jsonify({'error': str(e)}), 500
+        logging.error(f"Error: {str(e)}")
+        return jsonify({"error": "Something went wrong while processing."}), 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:
@@ -173,5 +184,4 @@ def transcribe():
 
 # Start Production Server
 if __name__ == "__main__":
-    serve(app, host="0.0.0.0", port=7860)
-    
+    serve(app, host="0.0.0.0", port=7860)