File size: 950 Bytes
98dc5b0 1214fa7 98dc5b0 1214fa7 98dc5b0 1214fa7 98dc5b0 1214fa7 98dc5b0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
from generateDistractors.senseToVec import S2V
from keyExtractor.rake import KeyExtractor
from questionGeneration.questionGeneration import QuestionGeneration
from summarizer.summarizer import Summarizer
import gradio as gr
sense2Vec = S2V()
Key = KeyExtractor()
Question = QuestionGeneration()
Summary = Summarizer()
def run(text):
result = []
summarized_text = Summary.summarizer(text)
print("summarized_text done")
imp_keywords = Key.get_keywords(text)
print("keywords done: ", imp_keywords)
for answer in imp_keywords:
ques = Question.get_question(summarized_text,answer)
print("question done")
distractors = sense2Vec.execute(answer)
print("distractors done")
result.append({
"question": ques,
"answer": answer.capitalize(),
"distractors": distractors
})
return result
if __name__ == '__main__':
demo = gr.Interface(fn=run, inputs="text", outputs="json")
demo.launch() |