Update app.py

app.py

@@ -6,7 +6,7 @@ from uuid import uuid4
 import logging
 import pandas as pd
 import os
-from urllib.parse import quote
+from urllib.parse import quote, urlparse, parse_qs
 
 # Configure logging for audit purposes (FR4, NFR: Security)
 logging.basicConfig(level=logging.INFO, filename='app_log.txt', 
@@ -67,13 +67,13 @@ SALESFORCE_HEADERS = {
     "Content-Type": "application/json"
 }
 
-# Hugging Face Client (No token needed in Hugging Face Spaces)
+# Hugging Face Client
 class HuggingFaceClient:
     def __init__(self):
         self.api_url = "https://api-inference.huggingface.co/models/"
-        self.headers = {}  # Token is implicit in Hugging Face Spaces
+        self.headers = {}
         self.sentiment_model = "distilbert-base-uncased-finetuned-sst-2-english"
-        self.severity_model = "bert-base-uncased"  # Placeholder; use custom model
+        self.severity_model = "bert-base-uncased"
         self.translation_model = "facebook/m2m100_418M"
 
     def query(self, text, model):
@@ -231,121 +231,100 @@ class CalendlyClient:
             logging.error(f"Failed to schedule Calendly event: {str(e)}")
             return None
 
-# Gradio Interface Functions
+# Initialize Clients
 hf_client = HuggingFaceClient()
 sf_client = SalesforceClient(SALESFORCE_BASE_URL, SALESFORCE_HEADERS)
 twilio_client = TwilioClient(TWILIO_ACCOUNT_SID, TWILIO_AUTH_TOKEN, TWILIO_PHONE)
 calendly_client = CalendlyClient(CALENDLY_API_TOKEN)
 
-# State to store patient phone number
-patient_phone_state = gr.State(value=None)
-
-def register_patient(name_first, name_last, dob, Gender, height, weight, marital, phone, email, 
-                    address_city, address_state, postal, 
-                    med_yes_no, med_list, emergency_first, emergency_last, emergency_relation, 
-                    emergency_number, language, consent_status):
-    if not all([name_first, name_last, dob, Gender, height, weight, marital, phone, email, 
-                 address_city, address_state, postal, language, consent_status]) or \
+# State to store patient context from URL
+patient_state = gr.State(value=None)
+
+# Get URL parameters
+def get_url_params():
+    query = os.getenv("QUERY_STRING", "")
+    params = parse_qs(query)
+    patient_id = params.get("patientId", [None])[0]
+    patient_name = params.get("patientName", [None])[0]
+    if patient_id and patient_name:
+        return {"patientId": patient_id, "patientName": patient_name}
+    return None
+
+# Initialize patient state from URL
+initial_patient = get_url_params()
+if initial_patient:
+    patient_state.value = initial_patient
+
+# Automated Follow-up Message
+def send_automated_followup(patient_id, action, phone=None):
+    try:
+        patients = sf_client.query_records(f"SELECT Phone__c, ConsentGiven__c, Name FROM Patient__c WHERE Id = '{patient_id}'")
+        if not patients or patients[0]["ConsentGiven__c"] != "Approved":
+            return
+        phone = phone or patients[0]["Phone__c"]
+        consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patient_id}' ORDER BY CreatedDate DESC LIMIT 1")
+        consent_method = consent[0]["Method__c"] if consent else "SMS"
+        messages = {
+            "registration": f"Welcome, {patients[0]['Name']}! Your registration is complete. A follow-up is scheduled.",
+            "survey": f"Dear {patients[0]['Name']}, thank you for your survey response. We’ll follow up soon.",
+            "appointment": f"Hi {patients[0]['Name']}, your appointment is confirmed. Details will follow."
+        }
+        message = messages.get(action, "A follow-up action has been triggered.")
+        twilio_client.send_message(phone, message, method=consent_method.lower())
+        logging.info(f"Automated follow-up sent for {action} to {phone}")
+    except Exception as e:
+        logging.error(f"Error sending automated follow-up: {str(e)}")
+
+# Registration Function
+def register_patient(name_first, name_last, dob, gender, height, weight, marital, phone, email, 
+                    address_city, address_state, postal, med_yes_no, med_list, 
+                    emergency_first, emergency_last, emergency_relation, emergency_number, language, consent_status):
+    if not all([name_first, name_last, dob, gender, height, weight, marital, phone, email, 
+                address_city, address_state, postal, language, consent_status]) or \
        (med_yes_no == "Yes" and not med_list):
-        return "All fields are required.", gr.State(value=None)
+        return "All fields are required.", None
     try:
         patient_data = {
             "First_Name__c": name_first,
             "Last_Name__c": name_last,
-            "Date_of_Birth__c": dob,
-            "Gender__c": Gender,
-            "Height__c": height,
-            "Weight__c": weight,
-            "Marital_Status__c": marital,
+            "DateOfBirth__c": dob,
+            "Gender__c": gender,
+            "Height__c": float(height),
+            "Weight__c": float(weight),
+            "MaritalStatus__c": marital,
             "Phone__c": phone,
             "Email__c": email,
             "City__c": address_city,
             "State__c": address_state,
-            "Postal_Code__c": postal,
-            "Medications__c": med_list if med_yes_no == "Yes" else "",
-            "Emergency_First_Name__c": emergency_first or "",
-            "Emergency_Last_Name__c": emergency_last or "",
-            "Emergency_Relation__c": emergency_relation or "",
-            "Emergency_Number__c": emergency_number or "",
+            "PostalCode__c": postal,
+            "Medication__c": med_list if med_yes_no == "Yes" else "None",
+            "EmergencyContactName__c": f"{emergency_first} {emergency_last}".strip() or "",
+            "EmergencyContactRelationship__c": emergency_relation or "",
+            "EmergencyContactNumber__c": emergency_number or "",
             "Language__c": language,
-            "ConsentGiven__c": consent_status
+            "ConsentGiven__c": consent_status == "Approved"
         }
         result = sf_client.create_record("Patient__c", patient_data)
-        if isinstance(result, str):  # Error message returned
-            return result, gr.State(value=None)
+        if isinstance(result, str):
+            return result, None
         if result:
             logging.info(f"Registered patient: {result}")
             message = f"Welcome! You are registered. Your patient ID is {result}."
-            if consent_status in ["Approved", "Awaiting"]:
-                twilio_client.send_message(phone, message, method=consent_status.lower().replace(" ", "")[:3] if consent_status == "Awaiting" else "sms")
-            return f"Patient registered successfully! Your patient ID is {result}. You can use this ID for future actions.", gr.State(value=phone)
-        return "Failed to register patient. Please try again or contact support.", gr.State(value=None)
+            if consent_status == "Approved":
+                twilio_client.send_message(phone, message, method="sms")
+            send_automated_followup(result, "registration", phone)
+            return f"Patient registered successfully! Your patient ID is {result}.", {"patientId": result, "patientName": f"{name_first} {name_last}"}
+        return "Failed to register patient. Please try again or contact support.", None
     except Exception as e:
         logging.error(f"Error registering patient: {str(e)}")
-        return f"Error registering patient: {str(e)}. Please try again or contact support.", gr.State(value=None)
-
-def submit_consent(patient_phone, consent_method):
-    if not patient_phone or not consent_method:
-        return "All fields are required."
-    try:
-        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()}
-            result = sf_client.create_record("Consent__c", consent_data)
-            if isinstance(result, str):  # Error message returned
-                return result
-            if result:
-                message = f"Your consent method has been updated to {consent_method}."
-                twilio_client.send_message(patient_phone, message, method=consent_method.lower())
-                logging.info(f"Consent recorded for patient: {patients[0]['Name']}")
-                return "Consent recorded successfully! Your consent is now updated."
-            return "Failed to record consent. Please try again or contact support."
-        return "Patient not found. Please register the patient first."
-    except Exception as e:
-        logging.error(f"Error submitting consent: {str(e)}")
-        return f"Error submitting consent: {str(e)}. Please try again or contact support."
-
-def schedule_followup(patient_phone, message_template, follow_up_date):
-    if not patient_phone or not message_template or not follow_up_date:
-        return "All fields are required."
-    try:
-        follow_up_date = datetime.strptime(follow_up_date, "%Y-%m-%d").date()
-        patients = sf_client.query_records(f"SELECT Id, ConsentGiven__c, Language__c, Name FROM Patient__c WHERE Phone__c = '{patient_phone}'")
-        if not patients:
-            return "Patient not found. Please register the patient first."
-        if patients[0]["ConsentGiven__c"] != "Approved":
-            return "Consent not approved for messaging. Please update consent status."
-        consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patients[0]['Id']}' ORDER BY CreatedDate DESC LIMIT 1")
-        consent_method = consent[0]["Method__c"] if consent else "SMS"
-        followup_data = {
-            "Patient__c": patients[0]["Id"], "FollowUpDate__c": follow_up_date.isoformat(),
-            "MessageTemplate__c": message_template, "Status__c": "Scheduled"
-        }
-        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. Please try again or contact support."
-        send_result = twilio_client.send_message(patient_phone, message_template, method=consent_method.lower())
-        if send_result is True:
-            sf_client.create_record("MessageLog__c", {
-                "Patient__c": patients[0]["Id"], "MessageText__c": message_template,
-                "Direction__c": "Outbound", "Timestamp__c": datetime.utcnow().isoformat()
-            })
-            logging.info(f"Follow-up scheduled and message sent for patient: {patients[0]['Name']}")
-            return "Follow-up scheduled successfully, and a message has been sent!"
-        return f"Failed to send message: {send_result}. Please try again or contact support."
-    except ValueError:
-        return "Invalid date format. Please use YYYY-MM-DD."
-    except Exception as e:
-        logging.error(f"Error scheduling follow-up: {str(e)}")
-        return f"Error scheduling follow-up: {str(e)}. Please try again or contact support."
+        return f"Error registering patient: {str(e)}. Please try again or contact support.", None
 
+# Appointment Scheduling
 def schedule_appointment(patient_email, event_type_slug="30min"):
-    if not patient_email or not event_type_slug:
-        return "All fields are required."
+    if not patient_email:
+        return "Patient Email is required."
     try:
-        patients = sf_client.query_records(f"SELECT Id, Phone__c, ConsentGiven__c, Language__c, Name FROM Patient__c WHERE Email__c = '{patient_email}'")
+        patients = sf_client.query_records(f"SELECT Id, Phone__c, ConsentGiven__c, Name FROM Patient__c WHERE Email__c = '{patient_email}'")
         if not patients:
             return "Patient not found. Please register the patient first."
         event_type_uuid = calendly_client.get_event_type_uuid(event_type_slug)
@@ -355,14 +334,16 @@ def schedule_appointment(patient_email, event_type_slug="30min"):
         if event_time:
             appointment_data = {"DateTime__c": event_time, "Status__c": "Scheduled", "Patient__c": patients[0]["Id"]}
             result = sf_client.create_record("Appointment__c", appointment_data)
-            if isinstance(result, str):  # Error message returned
+            if isinstance(result, str):
                 return result
             if result:
-                consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patients[0]['Id']}' ORDER BY CreatedDate DESC LIMIT 1")
-                consent_method = consent[0]["Method__c"] if consent else "SMS"
-                message = f"Your appointment is scheduled for {event_time}. Please confirm."
+                consent_method = "SMS"
+                if patients[0]["ConsentGiven__c"]:
+                    consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patients[0]['Id']}' ORDER BY CreatedDate DESC LIMIT 1")
+                    consent_method = consent[0]["Method__c"] if consent else "SMS"
+                message = f"Hi {patients[0]['Name']}, your appointment is scheduled for {event_time}. Please confirm."
                 twilio_client.send_message(patients[0]["Phone__c"], message, method=consent_method.lower())
-                logging.info(f"Appointment scheduled for {patient_email}")
+                send_automated_followup(patients[0]["Id"], "appointment", patients[0]["Phone__c"])
                 return f"Appointment scheduled successfully for {event_time}!"
             return "Failed to create appointment record. Please try again or contact support."
         return "Failed to schedule appointment. Please try again or contact support."
@@ -370,64 +351,66 @@ def schedule_appointment(patient_email, event_type_slug="30min"):
         logging.error(f"Error scheduling appointment: {str(e)}")
         return f"Error scheduling appointment: {str(e)}. Please try again or contact support."
 
+# Survey Submission
 def submit_survey(patient_phone, answer):
     if not patient_phone or not answer:
         return "Please provide an answer to the selected question."
-    default_questions = {
-        "en": ["How would you rate your visit today?", "Are you experiencing any discomfort?", "How satisfied are you with our staff?"],
-        "te": ["నిన్ను ఈ రోజు ఎలా అనిపించింది?", "మీకు ఏ రకమైన అసౌకర్యం ఉందా?", "మా సిబ్బందిపై మీ సంతృప్తి ఎంత?"],
-        "hi": ["आज आपकी यात्रा कैसी रही?", "क्या आपको कोई असुविधा हो रही है?", "हमारी टीम से आप कितने संतुष्ट हैं?"]
-    }
     try:
         patients = sf_client.query_records(f"SELECT Id, Language__c, Name FROM Patient__c WHERE Phone__c = '{patient_phone}'")
         if not patients:
             return "Patient not found. Please register the patient first."
-        language = patients[0]["Language__c"].lower()
-        question = default_questions.get(language, default_questions["en"])[0]  # Default to first question
-        survey_data = {"Question__c": question, "Answer__c": answer}
+        language = patients[0]["Language__c"].lower() or "en"
+        question = "How would you rate your visit today?"  # Default question
+        survey_data = {"Question__c": question, "Answer__c": answer, "Patient__c": patients[0]["Id"]}
         result = sf_client.create_record("Survey__c", survey_data)
-        if isinstance(result, str):  # Error message returned
+        if isinstance(result, str):
             return result
-        if not result:
-            return "Failed to submit survey. Please try again or contact support."
-        analysis = hf_client.analyze_response(answer, language)
-        if analysis:
-            symptom_data = {
-                "Patient__c": patients[0]["Id"], "ResponseText__c": answer,
-                "RiskScore__c": analysis["risk_level"], "Severity__c": analysis["severity_score"],
-                "Sentiment__c": analysis["sentiment_score"]
-            }
-            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}"
+        if result:
+            analysis = hf_client.analyze_response(answer, language)
+            if analysis:
+                symptom_data = {
+                    "Patient__c": patients[0]["Id"], "ResponseText__c": answer,
+                    "RiskScore__c": analysis["risk_level"], "Severity__c": analysis["severity_score"],
+                    "Sentiment__c": analysis["sentiment_score"]
                 }
-                sf_client.create_record("Case__c", case_data)
-            consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patients[0]['Id']}' ORDER BY CreatedDate DESC LIMIT 1")
-            consent_method = consent[0]["Method__c"] if consent else "SMS"
-            message = f"Thank you for your response to '{question}'. Your risk level is {analysis['risk_level']}."
-            twilio_client.send_message(patient_phone, message, method=consent_method.lower())
-            logging.info(f"Survey submitted for patient: {patients[0]['Name']}")
-            return f"Survey submitted successfully! Risk Level: {analysis['risk_level']}."
-        return "Failed to analyze survey response. Please try again or contact support."
+                symptom_result = sf_client.create_record("SymptomLog__c", symptom_data)
+                if isinstance(symptom_result, str):
+                    return symptom_result
+                if analysis["risk_level"] == "High":
+                    case_data = {
+                        "RelatedPatient__c": patients[0]["Id"], "Priority__c": "High",
+                        "Description__c": f"High-risk survey response: {answer}"
+                    }
+                    sf_client.create_record("Case__c", case_data)
+                consent_method = "SMS"
+                if patients[0]["ConsentGiven__c"]:
+                    consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patients[0]['Id']}' ORDER BY CreatedDate DESC LIMIT 1")
+                    consent_method = consent[0]["Method__c"] if consent else "SMS"
+                message = f"Thank you, {patients[0]['Name']}, for your response to '{question}'. Your risk level is {analysis['risk_level']}."
+                twilio_client.send_message(patient_phone, message, method=consent_method.lower())
+                send_automated_followup(patients[0]["Id"], "survey", patient_phone)
+                return f"Survey submitted successfully! Risk Level: {analysis['risk_level']}."
+            return "Failed to analyze survey response. Please try again or contact support."
+        return "Failed to submit survey. Please try again or contact support."
     except Exception as e:
         logging.error(f"Error submitting survey: {str(e)}")
         return f"Error submitting survey: {str(e)}. Please try again or contact support."
 
-def view_risk_dashboard():
+# Risk Dashboard
+def view_risk_dashboard(patient_id=None):
     try:
-        symptom_logs = sf_client.query_records("SELECT Patient__c, ResponseText__c, RiskScore__c, Severity__c, Sentiment__c FROM SymptomLog__c")
+        query = "SELECT Patient__c, ResponseText__c, RiskScore__c, Severity__c, Sentiment__c FROM SymptomLog__c"
+        if patient_id:
+            query += f" WHERE Patient__c = '{patient_id}'"
+        symptom_logs = sf_client.query_records(query)
         if symptom_logs:
             df = pd.DataFrame([
-                {"Patient": log["Patient__c"], "Response": log["ResponseText__c"], 
+                {"Patient ID": log["Patient__c"], "Response": log["ResponseText__c"], 
                  "Risk Score": log["RiskScore__c"], "Severity": log["Severity__c"], 
                  "Sentiment": log["Sentiment__c"]} for log in symptom_logs
             ])
             high_risk = df[df["Severity"] == "High"]
-            high_risk_info = high_risk.to_dict('records') if not high_risk.empty else "No high-risk cases."
+            high_risk_info = high_risk.to_dict('records') if not high_risk.empty else "No high-risk cases for this patient."
             logging.info("Risk dashboard loaded successfully.")
             return df, high_risk_info
         logging.info("No symptom logs found.")
@@ -436,6 +419,7 @@ def view_risk_dashboard():
         logging.error(f"Error loading risk dashboard: {str(e)}")
         return pd.DataFrame(), f"Error loading risk dashboard: {str(e)}. Please try again or contact support."
 
+# Escalate Case
 def escalate_case(patient_id, response_text):
     if not patient_id or not response_text:
         return "All fields are required."
@@ -445,13 +429,15 @@ def escalate_case(patient_id, response_text):
             "Description__c": f"High-risk response: {response_text}"
         }
         result = sf_client.create_record("Case__c", case_data)
-        if isinstance(result, str):  # Error message returned
+        if isinstance(result, str):
             return result
         if result:
             patients = sf_client.query_records(f"SELECT Phone__c FROM Patient__c WHERE Id = '{patient_id}'")
             if patients:
-                consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patient_id}' ORDER BY CreatedDate DESC LIMIT 1")
-                consent_method = consent[0]["Method__c"] if consent else "SMS"
+                consent_method = "SMS"
+                if patients[0]["ConsentGiven__c"]:
+                    consent = sf_client.query_records(f"SELECT Method__c FROM Consent__c WHERE Patient__c = '{patient_id}' ORDER BY CreatedDate DESC LIMIT 1")
+                    consent_method = consent[0]["Method__c"] if consent else "SMS"
                 message = "Your case has been escalated due to a high-risk response. A doctor will contact you soon."
                 twilio_client.send_message(patients[0]["Phone__c"], message, method=consent_method.lower())
             logging.info(f"Case created for patient: {patient_id}")
@@ -462,133 +448,110 @@ def escalate_case(patient_id, response_text):
         return f"Error escalating case: {str(e)}. Please try again or contact support."
 
 # Gradio Interface
-with gr.Blocks(theme=gr.themes.Soft(), title="AI-Powered Patient Follow-up Agent") as demo:
-    gr.Markdown("# AI-Powered Patient Follow-up Agent for Clinics")
-    
+with gr.Blocks(theme=gr.themes.Soft(), title="Clinic AI Assistant") as demo:
+    gr.Markdown("# Clinic AI Assistant - Patient Management Dashboard")
+    gr.Markdown("Welcome to your personalized clinic dashboard. Log in via Salesforce to access your data.")
+
+    # Get patient context from URL or state
+    patient_context = patient_state.value or {}
+    patient_id = patient_context.get("patientId")
+    patient_name = patient_context.get("patientName")
+
     with gr.Tabs():
-        with gr.Tab("Patient Registration"):
+        with gr.Tab("Patient Dashboard", id=0):
+            gr.Markdown(f"## Welcome, {patient_name or 'Guest'}! (Patient ID: {patient_id or 'Not Logged In'})")
+            if patient_id:
+                dashboard_table, high_risk_info = view_risk_dashboard(patient_id)
+                gr.Dataframe(dashboard_table, label="Your Symptom Logs", interactive=False)
+                gr.Textbox(high_risk_info, label="High-Risk Cases", interactive=False)
+            else:
+                gr.Markdown("Please log in via Salesforce to view your dashboard.")
+
+        with gr.Tab("Register New Patient", id=1):
             with gr.Row():
                 with gr.Column():
-                    name_first = gr.Textbox(label="First Name", placeholder="First Name")
-                    name_last = gr.Textbox(label="Last Name", placeholder="Last Name")
+                    name_first = gr.Textbox(label="First Name", placeholder="John", info="Enter your first name")
+                    name_last = gr.Textbox(label="Last Name", placeholder="Doe", info="Enter your last name")
                 with gr.Column():
-                    dob_input = gr.Textbox(label="Date of Birth", placeholder="MM-DD-YYYY")
-                    Gender_input = gr.Dropdown(label="Gender", choices=["Please Select", "Male", "Female", "Other"], value="Please Select")
+                    dob_input = gr.Textbox(label="Date of Birth (MM-DD-YYYY)", placeholder="01-15-1990", info="e.g., MM-DD-YYYY")
+                    gender_input = gr.Dropdown(label="Gender", choices=["Male", "Female", "Other", "Prefer Not to Say"], value="Prefer Not to Say")
             with gr.Row():
-                height_input = gr.Textbox(label="Height (inches)", placeholder="e.g., 65")
-                weight_input = gr.Textbox(label="Weight (pounds)", placeholder="e.g., 150")
-            marital_input = gr.Dropdown(label="Marital Status", choices=["Please Select", "Single", "Married", "Divorced", "Widowed"], value="Please Select")
+                height_input = gr.Textbox(label="Height (inches)", placeholder="65", info="e.g., 65 inches")
+                weight_input = gr.Textbox(label="Weight (pounds)", placeholder="150", info="e.g., 150 lbs")
+            marital_input = gr.Dropdown(label="Marital Status", choices=["Single", "Married", "Divorced", "Widowed"], value="Single")
             with gr.Row():
-                phone_input = gr.Textbox(label="Contact Number", placeholder="(000) 000-0000")
-                email_input = gr.Textbox(label="Email", placeholder="patient@example.com")
+                phone_input = gr.Textbox(label="Contact Number", placeholder="+1234567890", info="e.g., +12025550123")
+                email_input = gr.Textbox(label="Email", placeholder="john.doe@clinic.com", info="e.g., your.email@domain.com")
             with gr.Row():
                 with gr.Column():
-                    address_city = gr.Textbox(label="City", placeholder="City")
-                    address_state = gr.Textbox(label="State/Province", placeholder="State/Province")
-            postal_input = gr.Textbox(label="Postal/Zip Code", placeholder="Postal/Zip Code")
+                    address_city = gr.Textbox(label="City", placeholder="New York", info="e.g., New York")
+                    address_state = gr.Textbox(label="State", placeholder="NY", info="e.g., NY")
+            postal_input = gr.Textbox(label="Postal Code", placeholder="10001", info="e.g., 10001")
             with gr.Row():
-                med_yes_no = gr.Radio(label="Taking any medications, currently?", choices=["Yes", "No"], value="No")
-                med_list = gr.Textbox(label="Please list it here", placeholder="List medications here", lines=2, visible=False)
-            with gr.Accordion("In case of emergency", open=False):
+                med_yes_no = gr.Radio(label="Taking Medications?", choices=["Yes", "No"], value="No")
+                med_list = gr.Textbox(label="Medication List", placeholder="e.g., Aspirin, 100mg", lines=2, visible=False)
+            with gr.Accordion("Emergency Contact", open=False):
                 with gr.Row():
                     with gr.Column():
-                        emergency_first = gr.Textbox(label="First Name", placeholder="First Name")
-                        emergency_last = gr.Textbox(label="Last Name", placeholder="Last Name")
+                        emergency_first = gr.Textbox(label="First Name", placeholder="Jane")
+                        emergency_last = gr.Textbox(label="Last Name", placeholder="Doe")
                     with gr.Column():
-                        emergency_relation = gr.Textbox(label="Relationship", placeholder="e.g., Spouse")
-                        emergency_number = gr.Textbox(label="Contact Number", placeholder="(000) 000-0000")
-            language_input = gr.Dropdown(["English", "Telugu", "Hindi"], label="Language")
-            consent_input = gr.Dropdown(["SMS", "WhatsApp", "Awaiting"], label="Consent Method")
-            register_button = gr.Button("Submit")
-            register_output = gr.Textbox(label="Result")
-
-            # Update visibility of medication list based on radio selection
+                        emergency_relation = gr.Textbox(label="Relationship", placeholder="Spouse")
+                        emergency_number = gr.Textbox(label="Contact Number", placeholder="+1234567890")
+            language_input = gr.Dropdown(["English", "Telugu", "Hindi"], label="Preferred Language", value="English")
+            consent_input = gr.Dropdown(["SMS", "WhatsApp", "Awaiting"], label="Consent Method", value="SMS")
+            register_button = gr.Button("Register Patient")
+            register_output = gr.Textbox(label="Registration Result")
+
             def toggle_med_list(med_choice):
                 return gr.update(visible=med_choice == "Yes")
 
-            med_yes_no.change(
-                fn=toggle_med_list,
-                inputs=med_yes_no,
-                outputs=med_list
-            )
-
+            med_yes_no.change(fn=toggle_med_list, inputs=med_yes_no, outputs=med_list)
             register_button.click(
                 fn=register_patient,
-                inputs=[name_first, name_last, dob_input, Gender_input, height_input, weight_input, marital_input, 
-                        phone_input, email_input, address_city, address_state, 
-                        postal_input, med_yes_no, med_list, emergency_first, emergency_last, emergency_relation, 
-                        emergency_number, language_input, consent_input],
-                outputs=[register_output, patient_phone_state]
+                inputs=[name_first, name_last, dob_input, gender_input, height_input, weight_input, marital_input, 
+                        phone_input, email_input, address_city, address_state, postal_input, med_yes_no, med_list, 
+                        emergency_first, emergency_last, emergency_relation, emergency_number, language_input, consent_input],
+                outputs=[register_output, patient_state]
             )
 
-        with gr.Tab("Consent Form"):
-            consent_phone_input = gr.Textbox(label="Patient Phone Number", value=lambda x: x, interactive=False) if patient_phone_state.value else gr.Textbox(label="Patient Phone Number", placeholder="+1234567890")
-            consent_method_input = gr.Dropdown(["SMS", "WhatsApp"], label="Consent Method")
-            consent_button = gr.Button("Submit Consent")
-            consent_output = gr.Textbox(label="Result")
-            consent_button.click(
-                fn=submit_consent,
-                inputs=[consent_phone_input, consent_method_input],
-                outputs=consent_output
-            )
-
-        with gr.Tab("Follow-up Messaging"):
-            followup_phone_input = gr.Textbox(label="Patient Phone Number", value=lambda x: x, interactive=False) if patient_phone_state.value else gr.Textbox(label="Patient Phone Number", placeholder="+1234567890")
-            message_template_input = gr.Textbox(label="Message Template", value="How are you feeling today?")
-            followup_date_input = gr.Textbox(label="Follow-up Date (YYYY-MM-DD)", value=date.today().isoformat())
-            followup_button = gr.Button("Schedule Follow-up")
-            followup_output = gr.Textbox(label="Result")
-            followup_button.click(
-                fn=schedule_followup,
-                inputs=[followup_phone_input, message_template_input, followup_date_input],
-                outputs=followup_output
-            )
-
-        with gr.Tab("Appointment Scheduling"):
-            appt_email_input = gr.Textbox(label="Patient Email", placeholder="patient@example.com")
-            appt_event_input = gr.Textbox(label="Event Type Slug", value="30min", placeholder="e.g., 30min")
-            appt_button = gr.Button("Schedule Appointment")
-            appt_output = gr.Textbox(label="Result")
+        with gr.Tab("Schedule Appointment", id=2):
+            appt_email_input = gr.Textbox(label="Patient Email", placeholder="john.doe@clinic.com", value=patient_context.get("email", ""))
+            appt_event_input = gr.Dropdown(label="Appointment Duration", choices=["15min", "30min", "60min"], value="30min", info="Select appointment length")
+            appt_button = gr.Button("Book Appointment")
+            appt_output = gr.Textbox(label="Appointment Result")
             appt_button.click(
                 fn=schedule_appointment,
                 inputs=[appt_email_input, appt_event_input],
                 outputs=appt_output
             )
 
-        with gr.Tab("Survey"):
-            survey_phone_input = gr.Textbox(label="Patient Phone Number", value=lambda x: x, interactive=False) if patient_phone_state.value else gr.Textbox(label="Patient Phone Number", placeholder="+1234567890")
+        with gr.Tab("Submit Survey", id=3):
+            survey_phone_input = gr.Textbox(label="Patient Phone Number", placeholder="+1234567890", value=patient_context.get("phone", ""), info="e.g., +12025550123")
             survey_question_input = gr.Dropdown(label="Select Question", choices=[
                 "How would you rate your visit today?",
                 "Are you experiencing any discomfort?",
                 "How satisfied are you with our staff?"
-            ], interactive=True)
-            survey_answer_input = gr.Textbox(label="Your Answer", lines=4)
+            ], value="How would you rate your visit today?")
+            survey_answer_input = gr.Textbox(label="Your Answer", lines=4, placeholder="e.g., Very satisfied")
             survey_button = gr.Button("Submit Survey")
-            survey_output = gr.Textbox(label="Result")
+            survey_output = gr.Textbox(label="Survey Result")
             survey_button.click(
                 fn=submit_survey,
                 inputs=[survey_phone_input, survey_answer_input],
                 outputs=survey_output
             )
 
-        with gr.Tab("Risk Dashboard"):
-            dashboard_button = gr.Button("Load Dashboard")
-            dashboard_table = gr.Dataframe(label="Symptom Logs")
-            high_risk_output = gr.Textbox(label="High-Risk Cases")
-            dashboard_button.click(
-                fn=view_risk_dashboard,
-                outputs=[dashboard_table, high_risk_output]
+        with gr.Tab("Escalate Case", id=4):
+            escalate_id_input = gr.Textbox(label="Patient ID", placeholder="e.g., 001xxxxxxxxxxxx", value=patient_context.get("patientId", ""), info="Enter Patient ID for escalation")
+            escalate_response_input = gr.Textbox(label="Response Text", placeholder="e.g., Severe pain reported", info="Describe the issue")
+            escalate_button = gr.Button("Escalate Case")
+            escalate_output = gr.Textbox(label="Escalation Result")
+            escalate_button.click(
+                fn=escalate_case,
+                inputs=[escalate_id_input, escalate_response_input],
+                outputs=escalate_output
             )
-            with gr.Row():
-                escalate_id_input = gr.Textbox(label="Patient ID (for escalation)")
-                escalate_response_input = gr.Textbox(label="Response Text")
-                escalate_button = gr.Button("Escalate Case")
-                escalate_output = gr.Textbox(label="Result")
-                escalate_button.click(
-                    fn=escalate_case,
-                    inputs=[escalate_id_input, escalate_response_input],
-                    outputs=escalate_output
-                )
 
 # Launch Gradio app
 if __name__ == "__main__":