Update app.py

app.py

@@ -4,10 +4,14 @@ import os
 import logging
 
 # Set up logging
-logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
-# Path to CSV file in the root directory of your project
-CSV_FILE = "patients.csv"
+# Ensure the "files" directory exists
+FILES_DIR = "files"
+os.makedirs(FILES_DIR, exist_ok=True)
+
+# Path to CSV file in the "files" directory of your project
+CSV_FILE = os.path.join(FILES_DIR, "patients.csv")
 
 # Initialize the CSV file if it does not exist
 if not os.path.exists(CSV_FILE):
@@ -21,8 +25,11 @@ if not os.path.exists(CSV_FILE):
 # Function to handle registration
 def register_patient(name, age, phone):
     try:
+        logging.info(f"Starting registration for {name}, Age: {age}, Phone: {phone}")
+        
         # Load existing data
         df = pd.read_csv(CSV_FILE)
+        logging.debug(f"Loaded existing data: {df}")
         
         # Generate new patient ID
         new_id = len(df) + 1
@@ -34,16 +41,19 @@ def register_patient(name, age, phone):
             "Age": [age],
             "Phone Number": [phone]
         })
+        logging.info(f"New patient data: {new_data}")
+
         df = pd.concat([df, new_data], ignore_index=True)
+        logging.debug(f"Updated DataFrame: {df}")
         
         # Save updated data
         df.to_csv(CSV_FILE, index=False)
+        logging.info(f"Data saved to CSV file successfully at {CSV_FILE}.")
         
-        logging.info(f"Successfully registered patient {name} with ID {new_id}")
-        return f"Patient {name} with ID {new_id} registered successfully!"
+        return f"Patient {name} with ID {new_id} registered successfully!", CSV_FILE
     except Exception as e:
         logging.error(f"An error occurred while registering patient: {e}")
-        return f"An error occurred: {e}"
+        return f"An error occurred: {e}", None
 
 # Define Gradio interface
 with gr.Blocks() as demo:
@@ -54,8 +64,13 @@ with gr.Blocks() as demo:
         phone_number = gr.Textbox(label="Phone Number", placeholder="Enter phone number")
     register_button = gr.Button("Register")
     status = gr.Textbox(label="Status", interactive=False)
-    
-    register_button.click(fn=register_patient, inputs=[patient_name, age, phone_number], outputs=status)
+    download_button = gr.File(label="Download CSV")
+
+    def update_status_and_file(*inputs):
+        message, file_path = register_patient(*inputs)
+        return message, file_path
+
+    register_button.click(fn=update_status_and_file, inputs=[patient_name, age, phone_number], outputs=[status, download_button])
 
 # Launch the app
 if __name__ == "__main__":