updated questions generator

src/models/bot_response_schema.py

@@ -26,7 +26,7 @@ class AssessmentPredictionsResponse(BaseModel):
 
 class AssessmentSuggestion(BaseModel):
     assessment_type:str
-    frequency:str
+    duration:int
     Justification:str
 
 
@@ -41,3 +41,6 @@ class AchievementPrediction(BaseModel):
     goal_achievement:GoalAchivementPrediction
     goal_achievement_new_assessment:str
 
+
+class Areas(BaseModel):
+    areas:List[str]

src/models/questions_response.py

@@ -1,4 +1,4 @@
-from pydantic import BaseModel
+'''from pydantic import BaseModel
 from typing import List, Dict
 
 class Question(BaseModel):
@@ -9,5 +9,25 @@ class Question(BaseModel):
 
 class AssementQuestion(BaseModel):
     number: int
-    questions: List[Question]
+    questions: List[Question]'''
 
+
+from pydantic import BaseModel
+from typing import List, Dict
+
+
+class Question(BaseModel):
+    area_tag: int
+    assigned_to: int
+    questions: str
+    role: int
+
+class FrequencyQuestions(BaseModel):
+    frequency_number: int
+    items: List[Question]
+
+class Questions(BaseModel):
+    questions: List[FrequencyQuestions]
+
+class AssessmentQuestions(BaseModel):
+    questions: Questions