Update app.py

app.py

@@ -1,18 +1,9 @@
 import gradio as gr
 import json
 from datetime import datetime
-
-# Sample test categories for demonstration
-TEST_CATEGORIES = {
-    "Haematology": ["Complete Blood Count", "Haemoglobin"],
-    "Clinical Pathology": ["Urinalysis", "Stool Analysis"],
-    "Biochemistry": ["Liver Function Test", "Kidney Function Test"],
-    "Microbiology": ["Culture Test", "Sensitivity Test"],
-    "Specific Diseases": ["Malaria", "Dengue"],
-    "Serology": ["HIV Test", "Hepatitis Test"],
-    "Radiology": ["X-ray", "CT Scan"],
-    "Other Diagnostic Tests": ["Blood Sugar", "Cholesterol"]
-}
+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():
@@ -69,35 +60,28 @@ def registration_interface(name, father_name, age, phone, address, pincode):
     
     # Save data and updated sequence
     save_data(patients, last_sequence)
-    return f"Patient Registered. Patient ID: {patient_id}", patient_id
+    return f"Patient Registered. Patient ID: {patient_id}"
 
 # Tests Selection Tab
 def test_interface(categories):
-    available_tests = []
-    for category in categories:
-        available_tests.extend(TEST_CATEGORIES.get(category, []))
-    return available_tests
+    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):
     patients, last_sequence = load_data()  # Load both patients and last_sequence
-    if patient_id not in patients:
-        return "Patient ID not found."
-    patients[patient_id]["tests"] = selected_tests
+    response = select_tests(patient_id, selected_tests, patients)  # Update patients directly
     save_data(patients, last_sequence)  # Save both patients and last_sequence
-    return "Tests confirmed for patient."
+    return response
 
 # Billing Tab
 def billing_interface(patient_id):
     patients, _ = load_data()  # Load patients and ignore last_sequence
     if patient_id in patients:
-        return f"Billing for {patient_id}: Tests: {patients[patient_id]['tests']}"
+        billing_info = fetch_billing(patient_id, patients)
+        return billing_info
     else:
         return "Invalid Patient ID. Please check the ID."
 
-# Copy functionality
-def copy_patient_id(patient_id):
-    return f"{patient_id} copied to clipboard!"
-
 # Gradio Interface
 with gr.Blocks() as app:
     gr.Markdown("<h1 style='text-align: center;'>Sathkrutha LIMS</h1>", elem_id="header")
@@ -112,29 +96,15 @@ with gr.Blocks() as app:
         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")
-        patient_id_display = gr.Textbox(label="Patient ID", interactive=False)  # Display Patient ID here
-        copy_button = gr.Button("Copy Patient ID")
-        copy_output = gr.Textbox(label="Copy Status", interactive=False)
-
-        register_button.click(
-            registration_interface, 
-            [name, father_name, age, phone, address, pincode], 
-            [registration_output, patient_id_display]
-        )
 
-        copy_button.click(
-            copy_patient_id,
-            [patient_id_display],  # Pass the Patient ID to the function
-            copy_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(
-            list(TEST_CATEGORIES.keys()),
+            ["Haematology", "Clinical Pathology", "Biochemistry", "Microbiology", "Specific Diseases", "Serology", "Radiology", "Other Diagnostic Tests"],
             label="Select Test Categories"
         )
         available_tests = gr.CheckboxGroup(label="Available Tests")
@@ -246,4 +216,4 @@ h1, h2, h3, h4, h5, h6 {
 }
 """
 
-app.launch()
+app.launch()