Update app.py

app.py

@@ -5,8 +5,9 @@ from transformers import pipeline
 from gtts import gTTS
 from pydub import AudioSegment
 from pydub.silence import detect_nonsilent
+from salesforce import get_salesforce_connection  # Import the Salesforce connection function
+import re
 from waitress import serve
-from simple_salesforce import Salesforce
 
 app = Flask(__name__)
 
@@ -58,17 +59,25 @@ def is_silent_audio(audio_path):
     nonsilent_parts = detect_nonsilent(audio, min_silence_len=500, silence_thresh=audio.dBFS-16)  # Reduced silence duration
     return len(nonsilent_parts) == 0  # If no speech detected
 
-# Salesforce connection details
-sf = Salesforce(username='[email protected]', password='Sati@1020', security_token='sSSjyhInIsUohKpG8sHzty2q')
+# Extract name, email, and phone number from transcribed text
+def extract_name_email_phone(text):
+    # Regex for basic email and phone number
+    email = re.search(r'\S+@\S+', text)
+    phone = re.search(r'\+?\d{10,15}', text)  # Consider different formats for phone numbers
 
-# Debugging - Confirming Salesforce connection
-print("Connected to Salesforce, checking user info...")
-print(sf.UserInfo)  # This will give you the user info if the connection is successful
+    name = text.split(' ')[0]  # Simplified assumption that name is the first word
+    email = email.group(0) if email else "[email protected]"
+    phone = phone.group(0) if phone else "0000000000"
+
+    return name, email, phone
+
+# Get Salesforce connection using salesforce.py
+sf = get_salesforce_connection()
 
 # Function to create Salesforce record
 def create_salesforce_record(name, email, phone_number):
     try:
-        # Attempt to create a record in Salesforce
+        # Create the record in Salesforce
         customer_login = sf.Customer_Login__c.create({
             'Name': name,
             'Email__c': email,
@@ -76,21 +85,15 @@ def create_salesforce_record(name, email, phone_number):
         })
         
         # Log the response from Salesforce
-        print(f"Salesforce response: {customer_login}")
-        
         if customer_login.get('id'):
             print(f"Record created successfully with ID: {customer_login['id']}")
+            return customer_login
         else:
-            print("No ID returned, record creation may have failed.")
-        
-        return customer_login
+            print("Record creation failed: No ID returned")
+            return {"error": "Record creation failed: No ID returned"}
     except Exception as e:
-        # Catch and log any exceptions during record creation
-        error_message = str(e)
-        print(f"Error creating Salesforce record: {error_message}")
-        return {"error": f"Failed to create record in Salesforce: {error_message}"}
-
-
+        print(f"Error creating Salesforce record: {str(e)}")
+        return {"error": f"Failed to create record in Salesforce: {str(e)}"}
 
 @app.route("/")
 def index():
@@ -118,11 +121,8 @@ def transcribe():
         result = asr_model(output_audio_path, generate_kwargs={"language": "en"})
         transcribed_text = result["text"].strip().capitalize()
 
-        # Now, let's split the transcribed text into name, email, and phone number (basic example)
-        parts = transcribed_text.split()  # This is a simplistic approach; you may need a better parsing mechanism
-        name = parts[0]  # Assuming first word is the name
-        email = parts[1] if '@' in parts[1] else "[email protected]"  # Very basic email extraction
-        phone_number = parts[2] if len(parts) > 2 else "0000000000"  # Basic phone number assumption
+        # Extract name, email, and phone number from the transcribed text
+        name, email, phone_number = extract_name_email_phone(transcribed_text)
 
         # Create record in Salesforce
         salesforce_response = create_salesforce_record(name, email, phone_number)
@@ -139,7 +139,6 @@ def transcribe():
         print(f"Error in transcribing or processing: {str(e)}")
         return jsonify({"error": f"Speech recognition error: {str(e)}"}), 500
 
-
 # Start Production Server
 if __name__ == "__main__":
     serve(app, host="0.0.0.0", port=7860)