ENCOA-311

src/components/ExamEditor/ExercisePicker/index.tsx

@@ -13,12 +13,13 @@ import { BsArrowRepeat } from "react-icons/bs";
 interface ExercisePickerProps {
     module: string;
     sectionId: number;
-    difficulty: string;
     extraArgs?: Record<string, any>;
     levelSectionId?: number;
     level?: boolean;
 }
 
+const DIFFICULTIES: string[] = ["A1", "A2", "B1", "B2", "C1", "C2"];
+
 const ExercisePicker: React.FC<ExercisePickerProps> = ({
     module,
     sectionId,
@@ -26,7 +27,7 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
     levelSectionId,
     level = false
 }) => {
-    const { currentModule, dispatch } = useExamEditorStore();
+    const { currentModule } = useExamEditorStore();
     const { difficulty, sections } = useExamEditorStore((store) => store.modules[level ? "level" : currentModule]);
     const section = sections.find((s) => s.sectionId === (level ? levelSectionId : sectionId));
 
@@ -67,6 +68,9 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
                 }),
                 ...(config.params.max_words !== undefined && {
                     max_words: Number(config.params.max_words)
+                }),
+                ...(DIFFICULTIES.includes(config.params.difficulty as string) && {
+                    difficulty: config.params.difficulty
                 })
             };
         });
@@ -100,8 +104,8 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
                     method: 'POST',
                     body: {
                         ...context,
-                        exercises: exercises,
-                        difficulty: difficulty
+                        exercises,
+                        difficulty
                     }
                 },
                 (data: any) => [{
@@ -112,7 +116,11 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
             );
         } else if (module === "writing") {
             configurations.forEach((config) => {
-                let queryParams = config.params.topic !== '' ? { topic: config.params.topic as string } : undefined;
+                let queryParams = {
+                    difficulty: config.params.difficulty ? config.params.difficulty as string: difficulty,
+                    ...(config.params.topic !== '' && { topic: config.params.topic as string })
+                };
+                  
                 generate(
                     config.type === 'writing_letter' ? 1 : 2,
                     "writing",
@@ -122,7 +130,8 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
                         queryParams
                     },
                     (data: any) => [{
-                        prompt: data.question
+                        prompt: data.question,
+                        difficulty: data.difficulty
                     }],
                     levelSectionId,
                     level
@@ -135,6 +144,7 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
                         topic: config.params.topic as string,
                         first_topic: config.params.first_topic as string,
                         second_topic: config.params.second_topic as string,
+                        difficulty: config.params.difficulty ? config.params.difficulty as string: difficulty,
                     }).filter(([_, value]) => value && value !== '')
                 );
                 let query = Object.keys(queryParams).length === 0 ? undefined : queryParams;
@@ -152,19 +162,22 @@ const ExercisePicker: React.FC<ExercisePickerProps> = ({
                                 return [{
                                     prompts: data.questions,
                                     first_topic: data.first_topic,
-                                    second_topic: data.second_topic
+                                    second_topic: data.second_topic,
+                                    difficulty: data.difficulty
                                 }];
                             case 2:
                                 return [{
                                     topic: data.topic,
                                     question: data.question,
                                     prompts: data.prompts,
-                                    suffix: data.suffix
+                                    suffix: data.suffix,
+                                    difficulty: data.difficulty
                                 }];
                             case 3:
                                 return [{
                                     topic: data.topic,
-                                    questions: data.questions
+                                    questions: data.questions,
+                                    difficulty: data.difficulty
                                 }];
                             default:
                                 return [data];