Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import requests
 import json
 from datetime import datetime, date
-from uuid import uuid4  # Added to resolve uuid4 error
+from uuid import uuid4
 import logging
 import pandas as pd
 import os
@@ -145,11 +145,13 @@ class SalesforceClient:
             logging.info(f"Created {object_name} record: {record_id}")
             return record_id
         except requests.exceptions.HTTPError as e:
-            logging.error(f"HTTP error creating {object_name} record: {e.response.status_code} - {e.response.text}")
-            return None
+            error_message = f"HTTP error creating {object_name} record: {e.response.status_code} - {e.response.text}"
+            logging.error(error_message)
+            return error_message
         except requests.exceptions.RequestException as e:
-            logging.error(f"Failed to create {object_name} record: {str(e)}")
-            return None
+            error_message = f"Failed to create {object_name} record: {str(e)}"
+            logging.error(error_message)
+            return error_message
 
     def query_records(self, query):
         try:
@@ -243,14 +245,16 @@ def register_patient(phone, email, language, consent_status):
             "Language__c": language,
             "ConsentGiven__c": consent_status
         }
-        patient_id = sf_client.create_record("Patient__c", patient_data)
-        if patient_id:
-            logging.info(f"Registered patient: {patient_id}")
-            return f"Patient registered successfully! ID: {patient_id}"
+        result = sf_client.create_record("Patient__c", patient_data)
+        if isinstance(result, str):  # Error message returned
+            return result
+        if result:
+            logging.info(f"Registered patient: {result}")
+            return f"Patient registered successfully! ID: {result}"
         return "Failed to register patient. Check Salesforce logs."
     except Exception as e:
         logging.error(f"Error registering patient: {str(e)}")
-        return f"Error: {str(e)}"
+        return f"Error registering patient: {str(e)}"
 
 def submit_consent(patient_phone, consent_method):
     if not patient_phone or not consent_method:
@@ -259,8 +263,10 @@ def submit_consent(patient_phone, consent_method):
         patients = sf_client.query_records(f"SELECT Id, Name FROM Patient__c WHERE Phone__c = '{patient_phone}'")
         if patients:
             consent_data = {"Method__c": consent_method, "GivenOn__c": date.today().isoformat()}
-            consent_id = sf_client.create_record("Consent__c", consent_data)
-            if consent_id:
+            result = sf_client.create_record("Consent__c", consent_data)
+            if isinstance(result, str):  # Error message returned
+                return result
+            if result:
                 logging.info(f"Consent recorded for patient: {patients[0]['Name']}")
                 return "Consent recorded successfully!"
             return "Failed to record consent."
@@ -283,8 +289,10 @@ def schedule_followup(patient_phone, message_template, follow_up_date):
             "Patient__c": patients[0]["Id"], "FollowUpDate__c": follow_up_date.isoformat(),
             "MessageTemplate__c": message_template, "Status__c": "Scheduled"
         }
-        followup_id = sf_client.create_record("FollowUpPlan__c", followup_data)
-        if not followup_id:
+        result = sf_client.create_record("FollowUpPlan__c", followup_data)
+        if isinstance(result, str):  # Error message returned
+            return result
+        if not result:
             return "Failed to schedule follow-up."
         send_result = twilio_client.send_message(patient_phone, message_template)
         if send_result is True:
@@ -328,8 +336,10 @@ def escalate_case(patient_id, response_text):
             "RelatedPatient__c": patient_id, "Priority__c": "High",
             "Description__c": f"High-risk response: {response_text}"
         }
-        case_id = sf_client.create_record("Case__c", case_data)
-        if case_id:
+        result = sf_client.create_record("Case__c", case_data)
+        if isinstance(result, str):  # Error message returned
+            return result
+        if result:
             logging.info(f"Case created for patient: {patient_id}")
             return "Case created successfully!"
         return "Failed to create case."
@@ -347,8 +357,10 @@ def schedule_appointment(patient_email, event_type_slug="30min"):
         event_time = calendly_client.create_event(patient_email, event_type_uuid)
         if event_time:
             appointment_data = {"DateTime__c": event_time, "Status__c": "Scheduled"}
-            appointment_id = sf_client.create_record("Appointment__c", appointment_data)
-            if appointment_id:
+            result = sf_client.create_record("Appointment__c", appointment_data)
+            if isinstance(result, str):  # Error message returned
+                return result
+            if result:
                 logging.info(f"Appointment scheduled for {patient_email}")
                 return f"Appointment scheduled for {event_time}!"
             return "Failed to create appointment record."
@@ -365,8 +377,10 @@ def submit_survey(patient_phone, question, answer):
         if not patients:
             return "Patient not found."
         survey_data = {"Question__c": question, "Answer__c": answer}
-        survey_id = sf_client.create_record("Survey__c", survey_data)
-        if not survey_id:
+        result = sf_client.create_record("Survey__c", survey_data)
+        if isinstance(result, str):  # Error message returned
+            return result
+        if not result:
             return "Failed to submit survey."
         analysis = hf_client.analyze_response(answer, patients[0]["Language__c"])
         if analysis:
@@ -375,8 +389,10 @@ def submit_survey(patient_phone, question, answer):
                 "RiskScore__c": analysis["risk_level"], "Severity__c": analysis["severity_score"],
                 "Sentiment__c": analysis["sentiment_score"]
             }
-            symptom_id = sf_client.create_record("SymptomLog__c", symptom_data)
-            if symptom_id and analysis["risk_level"] == "High":
+            result = sf_client.create_record("SymptomLog__c", symptom_data)
+            if isinstance(result, str):  # Error message returned
+                return result
+            if result and analysis["risk_level"] == "High":
                 case_data = {
                     "RelatedPatient__c": patients[0]["Id"], "Priority__c": "High",
                     "Description__c": f"High-risk survey response: {answer}"