diff --git a/question_generation_main.py b/question_generation_main.py new file mode 100644 index 0000000..1cc3419 --- /dev/null +++ b/question_generation_main.py @@ -0,0 +1,31 @@ +'''This module ties together the +questions generation and incorrect answer +generation modules +''' +from question_extraction import QuestionExtractor +from incorrect_answer_generation import IncorrectAnswerGenerator + +class QuestionGeneration: + '''This class contains the method + to generate questions + ''' + + def __init__(self, num_questions, num_options): + self.num_questions = num_questions + self.num_options = num_options + self.question_extractor = QuestionExtractor(num_questions) + + def generate_questions_dict(self, document): + self.questions_dict = self.question_extractor.get_questions_dict(document) + self.incorrect_answer_generator = IncorrectAnswerGenerator(document) + + for i in range(1, self.num_questions+1): + if i not in self.questions_dict: + continue + self.questions_dict[i]["options"] \ + = self.incorrect_answer_generator.get_all_options_dict( + self.questions_dict[i]["answer"], + self.num_options + ) + + return self.questions_dict