import gradio as gr
import json
from datetime import datetime
from patient_registration import register_patient
from test_selection import select_tests, get_tests_by_category
from billing import fetch_billing

# Load data
def load_data():
    try:
        with open("data.json", "r") as file:
            return json.load(file)
    except FileNotFoundError:
        return {}

# Save data
def save_data(data):
    with open("data.json", "w") as file:
        json.dump(data, file)

# Patient Registration Tab
def registration_interface(name, father_name, age, phone, address, pincode):
    # Get today's date
    today = datetime.now()
    # Generate patient ID with last 5 digits of the phone number
    patient_id = f"{today.year}{today.month:02d}{today.day:02d}{str(phone)[-5:]}"
    
    # Proceed with the rest of the function
    data = load_data()
    data[patient_id] = {
        "name": name,
        "father_name": father_name,
        "age": age,
        "phone": phone,
        "address": address,
        "pincode": pincode,
        "tests": [],
        "total_cost": 0
    }
    save_data(data)
    return f"Patient Registered. Patient ID: {patient_id}"



# Tests Selection Tab
def test_interface(categories):
    available_tests = get_tests_by_category(categories)
    return gr.update(choices=available_tests)  # Update dropdown with available tests

def confirm_tests_interface(patient_id, selected_tests):
    data = load_data()
    response = select_tests(patient_id, selected_tests, data)
    save_data(data)
    return response

# Billing Tab
def billing_interface(patient_id):
    data = load_data()
    billing_info = fetch_billing(patient_id, data)
    return billing_info

# Gradio Interface
with gr.Blocks() as app:
    gr.Markdown("# Sathkrutha LIMS - Patient Management and Billing App")

    with gr.Tab("Patient Registration"):
        with gr.Row():
            name = gr.Textbox(label="Patient Name", placeholder="Enter patient name", elem_id="name_field", lines=1)
            father_name = gr.Textbox(label="Father/Husband Name", placeholder="Enter father or husband name", elem_id="father_name_field", lines=1)
        with gr.Row():
            age = gr.Number(label="Age", value=None, minimum=0, maximum=99, elem_id="age_field")
            phone = gr.Number(label="Phone Number", value=None, elem_id="phone_field")
        with gr.Row():
            address = gr.Textbox(label="Address", placeholder="Enter address", lines=2)
            pincode = gr.Number(label="Pincode", value=None, elem_id="pincode_field")
        register_button = gr.Button("Register Patient")
        registration_output = gr.Textbox(label="Registration Output")

        register_button.click(registration_interface, [name, father_name, age, phone, address, pincode], registration_output)

    with gr.Tab("Tests"):
        patient_id_test = gr.Textbox(label="Patient ID")
        categories = gr.CheckboxGroup(
            ["Haematology", "Clinical Pathology", "Biochemistry", "Microbiology", "Specific Diseases", "Serology", "Radiology", "Other Diagnostic Tests"],
            label="Select Test Categories"
        )
        available_tests = gr.CheckboxGroup(label="Available Tests")
        confirm_button = gr.Button("Confirm Tests")
        test_output = gr.Textbox(label="Test Selection Output")

        categories.change(test_interface, inputs=categories, outputs=available_tests)  # Update available tests based on selected categories
        confirm_button.click(confirm_tests_interface, [patient_id_test, available_tests], test_output)

    with gr.Tab("Billing"):
        patient_id_bill = gr.Textbox(label="Patient ID")
        fetch_button = gr.Button("Fetch Billing")
        billing_output = gr.Textbox(label="Billing Information")
        fetch_button.click(billing_interface, [patient_id_bill], billing_output)

# Custom CSS to adjust field sizes
app.css = """
#name_field, #father_name_field {
    max-width: 750px;
}

#age_field {
    max-width: 300px;
}

#phone_field {
    max-width: 1100px;
}

#pincode_field {
    max-width: 400px;
}
"""

app.launch()