Merge pull request #75 from EyeSeeTea/feat/move-add-another-treatment…

…-button feat: move add another treatment button to the bottom of section
EyeSeeTea · Dec 19, 2024 · 0a24903 · 0a24903
2 parents 41bdbc0 + b73aaf9
commit 0a24903
Showing 1 changed file with 21 additions and 19 deletions.
diff --git a/src/webapp/components/survey/SurveyForm.tsx b/src/webapp/components/survey/SurveyForm.tsx
@@ -116,25 +116,7 @@ export const SurveyForm: React.FC<SurveyFormProps> = props => {
                     return (
                         <PaddedDiv key={stage.id}>
                             <Typography>{i18n.t(`Stage - ${stage.title}`)}</Typography>
-                            {stage.repeatable && (
-                                <RightAlignedDiv>
-                                    <Button
-                                        variant="contained"
-                                        color="primary"
-                                        onClick={() => addProgramStage(stage.code)}
-                                    >
-                                        {i18n.t(`Add Another ${stage.title}`)}
-                                    </Button>
-                                    {stage.isAddedByUser && (
-                                        <CancelButton
-                                            variant="outlined"
-                                            onClick={() => removeProgramStage(stage.id)}
-                                        >
-                                            {i18n.t(`Remove ${stage.title}`)}
-                                        </CancelButton>
-                                    )}
-                                </RightAlignedDiv>
-                            )}
+
                             {stage.sections.map(section => {
                                 if (!section.isVisible || section.isAntibioticSection) return null;
 
@@ -172,6 +154,26 @@ export const SurveyForm: React.FC<SurveyFormProps> = props => {
                                     />
                                 );
                             })}
+
+                            {stage.repeatable && (
+                                <RightAlignedDiv>
+                                    <Button
+                                        variant="contained"
+                                        color="primary"
+                                        onClick={() => addProgramStage(stage.code)}
+                                    >
+                                        {i18n.t(`Add Another ${stage.title}`)}
+                                    </Button>
+                                    {stage.isAddedByUser && (
+                                        <CancelButton
+                                            variant="outlined"
+                                            onClick={() => removeProgramStage(stage.id)}
+                                        >
+                                            {i18n.t(`Remove ${stage.title}`)}
+                                        </CancelButton>
+                                    )}
+                                </RightAlignedDiv>
+                            )}
                         </PaddedDiv>
                     );
                 })}