update app.py

app.py

@@ -22,6 +22,8 @@ def process_text(keywords, stage, medical_history):
                 return gptService.getShortTermGoals(keywords,medical_history)
             elif stage == Stage.LONG_TERM_GOALS.value:
                 return gptService.getLongTermGoals(keywords,medical_history)
+            elif stage == Stage.RECOMMENDATION.value:
+                return gptService.getRecommendation(keywords,medical_history)
             else:
                 print("Unknown stage")
                 raise gr.Error('Unknown stage encountered')
@@ -45,7 +47,7 @@ medical_history = gr.Textbox(lines=7, label="Medical History", placeholder="Ente
 # stage_choices = [stage.value for stage in Stage]
 
 # To add few stages
-stage_choices = [Stage.INITIAL_ASSESSMENT.value,Stage.FOLLOWUP_ASSESSMENT.value,Stage.DETAILED_EVALUATION.value,Stage.DISCHARGE_SUMMARY.value,Stage.LONG_TERM_GOALS.value]
+stage_choices = [Stage.INITIAL_ASSESSMENT.value,Stage.SHORT_TERM_GOALS.value,Stage.LONG_TERM_GOALS.value, Stage.RECOMMENDATION.value]
 
 stage = gr.Dropdown(choices=stage_choices, label="Stage")