Update service.py

service.py

@@ -17,6 +17,7 @@ class Stage(Enum):
     EVALUATION = "Evaluation"
     DETAILED_EVALUATION = "Detailed Evaluation"
     PROGRESS_NOTE = "Progress Note"
+    RECOMMENDATION="Recommendations"
     DISCHARGE_SUMMARY = "Discharge Summary"
     SHORT_TERM_GOALS = "Short Term Goals"
     LONG_TERM_GOALS = "Long Term Goals"
@@ -59,7 +60,7 @@ class GPT_Service:
     
     def formulateUserPrompt(self, assessmentPhase, keywords, medicalHistory):
         patientData= f"""
-        Given we are in the {assessmentPhase} phase with a patient who has articulated experiences of {keywords}.
+        Keywords: {keywords}.
         {f"Patient has the following medical history: {medicalHistory}" if medicalHistory is not None and medicalHistory != ""
         else ""}
         """
@@ -88,6 +89,11 @@ class GPT_Service:
         userPrompt= self.formulateUserPrompt(Stage.EVALUATION.value,keywords,medicalHistory)
         response= self.request_gpt_for_response(Prompts.EVALUATION_PROMPT.value,userPrompt)
         return self.processOutputResults(response)
+    
+    def getRecommendation(self, keywords, medicalHistory):
+        userPrompt= self.formulateUserPrompt(Stage.RECOMMENDATION.value,keywords,medicalHistory)
+        response= self.request_gpt_for_response(Prompts.RECOMMENDATION_PROMPT.value,userPrompt)
+        return self.processOutputResults(response)
 
     def getDetailedEvaluation(self, keywords,medicalHistory):
         userPrompt= self.formulateUserPrompt(Stage.DETAILED_EVALUATION.value,keywords,medicalHistory)