Index: firmware/App/Modes/ModePostTreat.h
===================================================================
diff -u -r8cdd4c1f2945181d31659605fb3777a967b753de -rcd5be724d5a3ba7457e761191d82f278654d7f5c
--- firmware/App/Modes/ModePostTreat.h	(.../ModePostTreat.h)	(revision 8cdd4c1f2945181d31659605fb3777a967b753de)
+++ firmware/App/Modes/ModePostTreat.h	(.../ModePostTreat.h)	(revision cd5be724d5a3ba7457e761191d82f278654d7f5c)
@@ -1,14 +1,14 @@
 /**************************************************************************
 *
-* Copyright (c) 2019-2021 Diality Inc. - All Rights Reserved.
+* Copyright (c) 2019-2024 Diality Inc. - All Rights Reserved.
 *
 * THIS CODE MAY NOT BE COPIED OR REPRODUCED IN ANY FORM, IN PART OR IN
 * WHOLE, WITHOUT THE EXPLICIT PERMISSION OF THE COPYRIGHT OWNER.
 *
 * @file    ModePostTreat.h
 *
-* @author  (last)      Dara Navaei
-* @date    (last)      12-Nov-2021
+* @author  (last)      Sean Nash
+* @date    (last)      09-Aug-2023
 *
 * @author  (original)  Dara Navaei
 * @date    (original)  05-Nov-2019
@@ -78,6 +78,13 @@
     U32 waterSampleTestResult;                  ///< Pass or fail
 } TREATMENT_LOG_DATA_PAYLOAD_T;
 
+/// Post treatment state data publish
+typedef struct
+{
+    U32 currentPostTreatmentState;               ///< Post treatment current state.
+    U32 currentPostTxDrainState;                 ///< Post treatment drain state.
+} POST_TREATMENT_STATE_DATA_T;
+
 // ********** public function prototypes **********
 
 void initPostTreatmentMode( void );                                 // Initialize this module
@@ -88,6 +95,7 @@
 void sendTreatmentLogDataToUI( void );
 
 void signalUserConfirmPatientDisconnection( void );
+void requestPostTxNext( void );
 void signalUserConfirmDisposableRemoval( void );
 void signalAlarmActionToPostTreatmentMode( ALARM_ACTION_T action ); // Execute alarm action as appropriate for post-treatment mode