Update app.py

app.py

@@ -3,14 +3,27 @@ import librosa
 import numpy as np
 import torch
 from transformers import Wav2Vec2Processor, Wav2Vec2Model
-import requests
-import json
+from simple_salesforce import Salesforce
 import os
 from datetime import datetime
 
-# Salesforce API credentials (store securely in environment variables)
-SALESFORCE_API_URL = os.getenv("SALESFORCE_API_URL", "https://your-salesforce-instance.salesforce.com/services/data/v60.0/sobjects/HealthAssessment__c")
-SALESFORCE_TOKEN = os.getenv("SALESFORCE_TOKEN", "your_salesforce_token")
+# Salesforce credentials (store securely in environment variables)
+SF_USERNAME = os.getenv("SF_USERNAME", "smartvoicebot@voice.com")
+SF_PASSWORD = os.getenv("SF_PASSWORD", "voicebot1")
+SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN", "jq4VVHUFti6TmzJDjjegv2h6b")
+SF_INSTANCE_URL = os.getenv("SF_INSTANCE_URL", "https://voicebot-dev-ed.develop.lightning.force.com/lightning/setup/SetupOneHome/home")
+
+# Initialize Salesforce connection
+try:
+    sf = Salesforce(
+        username=SF_USERNAME,
+        password=SF_PASSWORD,
+        security_token=SF_SECURITY_TOKEN,
+        instance_url=SF_INSTANCE_URL
+    )
+except Exception as e:
+    print(f"Failed to connect to Salesforce: {str(e)}")
+    sf = None
 
 # Load Wav2Vec2 model for speech feature extraction
 processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
@@ -45,7 +58,8 @@ def analyze_voice(audio_file):
         feedback += "\n\n**Disclaimer**: This is not a diagnostic tool. Consult a healthcare provider for medical advice."
         
         # Store in Salesforce
-        store_in_salesforce(audio_file, feedback, respiratory_score, mental_health_score)
+        if sf:
+            store_in_salesforce(audio_file, feedback, respiratory_score, mental_health_score)
         
         return feedback
     except Exception as e:
@@ -53,20 +67,16 @@ def analyze_voice(audio_file):
 
 def store_in_salesforce(audio_file, feedback, respiratory_score, mental_health_score):
     """Store analysis results in Salesforce."""
-    headers = {
-        "Authorization": f"Bearer {SALESFORCE_TOKEN}",
-        "Content-Type": "application/json"
-    }
-    data = {
-        "AssessmentDate__c": datetime.utcnow().isoformat(),
-        "Feedback__c": feedback,
-        "RespiratoryScore__c": float(respiratory_score),
-        "MentalHealthScore__c": float(mental_health_score),
-        "AudioFileName__c": os.path.basename(audio_file)
-    }
-    response = requests.post(SALESFORCE_API_URL, headers=headers, json=data)
-    if response.status_code != 201:
-        print(f"Failed to store in Salesforce: {response.text}")
+    try:
+        sf.HealthAssessment__c.create({
+            "AssessmentDate__c": datetime.utcnow().isoformat(),
+            "Feedback__c": feedback,
+            "RespiratoryScore__c": float(respiratory_score),
+            "MentalHealthScore__c": float(mental_health_score),
+            "AudioFileName__c": os.path.basename(audio_file)
+        })
+    except Exception as e:
+        print(f"Failed to store in Salesforce: {str(e)}")
 
 # Gradio interface
 iface = gr.Interface(