Update app.py

app.py

@@ -25,21 +25,28 @@ def generate_patient_id(phone, last_sequence):
     current_year = today.year
     current_month = today.month
     
+    # Check if year or month has changed
     if last_sequence["year"] != current_year or last_sequence["month"] != current_month:
+        # Reset sequence for new year/month
         last_sequence["year"] = current_year
         last_sequence["month"] = current_month
         last_sequence["number"] = 1
     else:
+        # Increment sequence number
         last_sequence["number"] += 1
     
+    # Format patient ID
     patient_id = f"{current_year}{current_month:02d}{last_sequence['number']:05d}"
     return patient_id
 
 # Patient Registration Tab
 def registration_interface(name, father_name, age, phone, address, pincode):
     patients, last_sequence = load_data()
+    
+    # Generate patient ID
     patient_id = generate_patient_id(phone, last_sequence)
     
+    # Add patient details to data
     patients[patient_id] = {
         "name": name,
         "father_name": father_name,
@@ -51,9 +58,30 @@ def registration_interface(name, father_name, age, phone, address, pincode):
         "total_cost": 0
     }
     
+    # Save data and updated sequence
     save_data(patients, last_sequence)
     return f"Patient Registered. Patient ID: {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):
+    patients, last_sequence = load_data()  # Load both patients and last_sequence
+    response = select_tests(patient_id, selected_tests, patients)  # Update patients directly
+    save_data(patients, last_sequence)  # Save both patients and last_sequence
+    return response
+
+# Billing Tab
+def billing_interface(patient_id):
+    patients, _ = load_data()  # Load patients and ignore last_sequence
+    if patient_id in patients:
+        billing_info = fetch_billing(patient_id, patients)
+        return billing_info
+    else:
+        return "Invalid Patient ID. Please check the ID."
+
 # Gradio Interface
 with gr.Blocks() as app:
     gr.Markdown("<h1 style='text-align: center;'>Sathkrutha LIMS</h1>", elem_id="header")
@@ -70,21 +98,43 @@ with gr.Blocks() as app:
             pincode = gr.Number(label="Pincode", value=None, elem_id="pincode_field")
         register_button = gr.Button("Register Patient")
         registration_output = gr.Textbox(label="Registration Output")
-        hidden_patient_id = gr.Textbox(visible=False)  # Hidden field to store the patient ID
+        hidden_patient_id = gr.Textbox(visible=False, elem_id="hidden_patient_id")  # Hidden field to store the patient ID
 
         register_button.click(registration_interface, [name, father_name, age, phone, address, pincode], [registration_output, hidden_patient_id])
 
         # Custom HTML for Copy ID Button
-        copy_id_html = """
+        gr.HTML("""
         <button onclick="copyPatientID()" style="background-color: #3b74f2; color: white; padding: 10px; border-radius: 5px; border: none; cursor: pointer;">Copy Patient ID</button>
         <script>
             function copyPatientID() {
-                navigator.clipboard.writeText(document.querySelector('#hidden_patient_id textarea').value);
-                alert('Patient ID copied to clipboard');
+                const patientID = document.querySelector('#hidden_patient_id input').value;
+                navigator.clipboard.writeText(patientID).then(() => {
+                    alert('Patient ID copied to clipboard');
+                }).catch(err => {
+                    alert('Failed to copy Patient ID');
+                });
             }
         </script>
-        """
-        gr.HTML(copy_id_html)
+        """)
+
+    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 styling
 app.css = """