Update register_patient.py

register_patient.py

@@ -10,18 +10,13 @@ def initialize_csv():
     if not os.path.exists(patient_file_path):
         with open(patient_file_path, mode='w', newline='') as file:
             writer = csv.writer(file)
-            writer.writerow(["Patient ID", "Name", "Age (Years)", "Age (Months)", "Age (Days)", "Gender", "Address", "Contact Number"])
+            writer.writerow(["Patient ID", "Name", "Age", "Age Unit", "Gender", "Address", "Contact Number"])
 
-def register_patient(name, age_selection, gender, contact):
+def register_patient(name, age, age_unit, gender, contact):
     """Register a new patient and return the registration status."""
     address = "Hyderabad"
     initialize_csv()  # Ensure the CSV is initialized
     
-    # Calculate age based on the selection
-    age_years = age_selection.get("Years", 0)
-    age_months = age_selection.get("Months", 0)
-    age_days = age_selection.get("Days", 0)
-    
     # Read the last Patient ID and increment it
     with open(patient_file_path, mode='r') as file:
         reader = csv.reader(file)
@@ -33,7 +28,7 @@ def register_patient(name, age_selection, gender, contact):
     # Write the new patient details to the CSV file
     with open(patient_file_path, mode='a', newline='') as file:
         writer = csv.writer(file)
-        writer.writerow([new_id, name, age_years, age_months, age_days, gender, address, contact])
+        writer.writerow([new_id, name, age, age_unit, gender, address, contact])
     
     return f"Patient ID: {new_id}, Name: {name} registered successfully."
 
@@ -49,12 +44,10 @@ def create_gradio_interface():
                 # Define Gradio inputs
                 name = gr.Textbox(label="Patient Name", placeholder="Enter patient name")
                 
-                # Age inputs using checkboxes
-                age_years = gr.Number(label="Age (Years)", value=0)
-                age_months = gr.Number(label="Age (Months)", value=0)
-                age_days = gr.Number(label="Age (Days)", value=0)
-                age_selection = gr.CheckboxGroup(choices=["Years", "Months", "Days"], label="Select Age Unit(s)")
-
+                # Age input with number and unit selection similar to gender buttons
+                age = gr.Number(label="Age", value=0)
+                age_unit = gr.Radio(["Years", "Months", "Days"], label="Age Unit", value="Years")
+                
                 contact_number = gr.Textbox(label="Contact Number", placeholder="Enter contact number")
                 
             with gr.Column():
@@ -65,24 +58,12 @@ def create_gradio_interface():
         result = gr.Textbox(label="Result", interactive=False)
         
         # Define Register button
-        def on_register_click(name, age_selection, age_years, age_months, age_days, gender, contact_number):
-            # Create a dictionary to map the selected age units to their values
-            age_dict = {}
-            if "Years" in age_selection:
-                age_dict["Years"] = age_years
-            if "Months" in age_selection:
-                age_dict["Months"] = age_months
-            if "Days" in age_selection:
-                age_dict["Days"] = age_days
-
-            return register_patient(name, age_dict, gender, contact_number)
-        
         register_button = gr.Button("Register")
         
         # Define the registration logic
         register_button.click(
-            fn=on_register_click,
-            inputs=[name, age_selection, age_years, age_months, age_days, gender, contact_number],
+            fn=register_patient,
+            inputs=[name, age, age_unit, gender, contact_number],
             outputs=result
         )