Update database.py

database.py

@@ -78,6 +78,18 @@ def get_db_data(db_path_glaucoma, db_path_cataract):
     else:
         return [], []
 
+def clear_database(db_path, table_name):
+    """Clear all records from the specified table in the database."""
+    try:
+        conn = sqlite3.connect(db_path)
+        cursor = conn.cursor()
+        cursor.execute(f"DELETE FROM {table_name}")
+        conn.commit()
+        conn.close()
+        return f"All records from {table_name} table in {db_path} have been deleted."
+    except Exception as e:
+        return f"Error deleting records from {table_name} table in {db_path}: {str(e)}"
+
 def format_db_data(glaucoma_data, cataract_data):
     """ Format the database data for display """
     formatted_data = ""