Update app.py

app.py

@@ -18,9 +18,6 @@ app = Flask(__name__)
 # Configure Flask session
 app.secret_key = os.getenv("SECRET_KEY", "sSSjyhInIsUohKpG8sHzty2q")
 app.config["SESSION_TYPE"] = "filesystem"
-app.config["SESSION_COOKIE_NAME"] = "my_session"
-app.config["SESSION_COOKIE_SECURE"] = True
-app.config["SESSION_COOKIE_SAMESITE"] = "None"
 Session(app)
 
 # Set up logging
@@ -28,18 +25,17 @@ logging.basicConfig(level=logging.INFO)
 
 # Connect to Salesforce
 try:
-    print("Attempting to connect to Salesforce...")
     sf = Salesforce(username='diggavalli98@gmail.com', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
-    print("Connected to Salesforce successfully!")
+    print("✅ Connected to Salesforce successfully!")
 except Exception as e:
-    print(f"Failed to connect to Salesforce: {str(e)}")
+    print(f"❌ Failed to connect to Salesforce: {str(e)}")
 
 # Whisper ASR Configuration
 device = "cuda" if torch.cuda.is_available() else "cpu"
 config = AutoConfig.from_pretrained("openai/whisper-small")
 config.update({"timeout": 60})
 
-# Voice prompts for registration
+# Voice prompts
 prompts = {
     "welcome": "Welcome to Biryani Hub.",
     "ask_name": "Tell me your name.",
@@ -47,13 +43,12 @@ prompts = {
     "thank_you": "Thank you for registration."
 }
 
+# Function to generate voice prompts
 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}")
-        print("Retrying after 5 seconds...")
         time.sleep(5)
         generate_audio_prompt(text, filename)
 
@@ -62,12 +57,9 @@ for key, text in prompts.items():
 
 # Function to convert audio to WAV format
 def convert_to_wav(input_path, output_path):
-    try:
-        audio = AudioSegment.from_file(input_path)
-        audio = audio.set_frame_rate(16000).set_channels(1)
-        audio.export(output_path, format="wav")
-    except Exception as e:
-        raise Exception(f"Audio conversion failed: {str(e)}")
+    audio = AudioSegment.from_file(input_path)
+    audio = audio.set_frame_rate(16000).set_channels(1)
+    audio.export(output_path, format="wav")
 
 # Function to check if audio contains actual speech
 def is_silent_audio(audio_path):
@@ -79,47 +71,39 @@ def is_silent_audio(audio_path):
 def index():
     return render_template("index.html")
 
-# ✅ LOGIN ENDPOINT (Validates Existing Users)
+# ✅ LOGIN ENDPOINT
 @app.route('/login', methods=['POST'])
 def login():
     data = request.json
-    email = data.get('email').strip().lower()
-    phone_number = data.get('phone_number').strip()
+    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
 
     try:
-        print(f"🔍 Checking login for Email: {email}, Phone: {phone_number}")
-
         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)
 
         if result['totalSize'] == 0:
-            print("❌ No matching records found!")
             return jsonify({'error': 'Invalid email or phone number. User not found'}), 401
 
         user_data = result['records'][0]
         session['user_id'] = user_data['Id']
         session['name'] = user_data['Name']
-        print("✅ User found:", user_data)
 
         return jsonify({'success': True, 'message': 'Login successful', 'user_id': user_data['Id'], 'name': user_data['Name']}), 200
 
-    except requests.exceptions.RequestException as req_error:
-        print("🔴 Salesforce Connection Error:", req_error)
-        return jsonify({'error': f'Salesforce connection error: {str(req_error)}'}), 500
     except Exception as e:
-        print("🚨 Unexpected Error:", e)
         return jsonify({'error': f'Unexpected error: {str(e)}'}), 500
 
-# ✅ REGISTRATION ENDPOINT (Creates New User)
+# ✅ REGISTRATION ENDPOINT
 @app.route("/register", methods=["POST"])
 def register():
     data = request.json
-    name = data.get('name')
-    email = data.get('email').strip().lower()
-    phone = data.get('phone').strip()
+    name = data.get('name', '').strip()
+    email = data.get('email', '').strip().lower()
+    phone = data.get('phone', '').strip()
 
     if not name or not email or not phone:
         return jsonify({'error': 'Missing data'}), 400
@@ -145,7 +129,7 @@ def register():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-# ✅ TRANSCRIPTION ENDPOINT (Converts Speech to Text)
+# ✅ TRANSCRIPTION ENDPOINT
 @app.route("/transcribe", methods=["POST"])
 def transcribe():
     if "audio" not in request.files:
@@ -173,4 +157,3 @@ def transcribe():
 # Start Production Server
 if __name__ == "__main__":
     serve(app, host="0.0.0.0", port=7860)
-