Aakash Goel commited on
Commit
f8665e5
·
1 Parent(s): f5ce1f3
Files changed (2) hide show
  1. code/helper.py +2 -2
  2. code/quiz_gen_new3.py +6 -15
code/helper.py CHANGED
@@ -79,13 +79,13 @@ def filter_overlap_words(l):
79
 
80
  def get_keywords(originaltext,summarytext):
81
  keywords = get_nouns_multipartite(originaltext)
82
- print ("keywords unsummarized: ",keywords)
83
  keyword_processor = KeywordProcessor()
84
  for keyword in keywords:
85
  keyword_processor.add_keyword(keyword)
86
  keywords_found = keyword_processor.extract_keywords(summarytext)
87
  keywords_found = list(set(keywords_found))
88
- print("keywords_found in summarized: ",keywords_found)
89
  important_keywords =[]
90
  for keyword in keywords:
91
  if keyword in keywords_found:
 
79
 
80
  def get_keywords(originaltext,summarytext):
81
  keywords = get_nouns_multipartite(originaltext)
82
+ #print ("keywords unsummarized: ",keywords)
83
  keyword_processor = KeywordProcessor()
84
  for keyword in keywords:
85
  keyword_processor.add_keyword(keyword)
86
  keywords_found = keyword_processor.extract_keywords(summarytext)
87
  keywords_found = list(set(keywords_found))
88
+ #print("keywords_found in summarized: ",keywords_found)
89
  important_keywords =[]
90
  for keyword in keywords:
91
  if keyword in keywords_found:
code/quiz_gen_new3.py CHANGED
@@ -79,14 +79,14 @@ start_time = str(datetime.datetime.now())
79
  if raw_text != None and raw_text != '':
80
  summary_text = summarizer(raw_text,summary_model,summary_tokenizer)
81
  ans_list = get_keywords(raw_text,summary_text)
82
- print("Ans list: {}".format(ans_list))
83
  questions = []
84
  option1=[]
85
  option2=[]
86
  option3=[]
87
  option4=[]
88
  for idx,ans in enumerate(ans_list):
89
- print("IDX: {}, ANS: {}".format(idx, ans))
90
  ques = get_question(summary_text,ans,question_model,question_tokenizer)
91
  other_options = get_related_word(ans)
92
  final_options, ans_index = get_final_option_list(ans,other_options)
@@ -112,21 +112,12 @@ if raw_text != None and raw_text != '':
112
  st.markdown("-----")
113
  questions.append(ques)
114
  output_path = "results/df_quiz_log_file_v1.csv"
115
- st.write(len(ans_list),len(questions),len(option1),len(option2),len(option3),len(option4))
116
- st.write("Answer: {}".format(ans_list))
117
- st.write(questions)
118
- st.write(option1)
119
- st.write(option2)
120
- st.write(option3)
121
- st.write(option4)
122
- st.write([str(raw_text)]*len(ans_list))
123
- st.write([start_time]*len(ans_list))
124
  res_df = pd.DataFrame({"TimeStamp":[start_time]*len(ans_list),\
125
  "Input":[str(raw_text)]*len(ans_list),\
126
- "Question":questions,"Option1":[option1],\
127
- "Option2":[option2],\
128
- "Option3":[option3],\
129
- "Option4":[option4],\
130
  "Correct Answer":ans_list})
131
  res_df.to_csv(output_path, mode='a', index=False, sep="\t", header= not os.path.exists(output_path))
132
  # st.dataframe(pd.read_csv(output_path,sep="\t").tail(5))
 
79
  if raw_text != None and raw_text != '':
80
  summary_text = summarizer(raw_text,summary_model,summary_tokenizer)
81
  ans_list = get_keywords(raw_text,summary_text)
82
+ #print("Ans list: {}".format(ans_list))
83
  questions = []
84
  option1=[]
85
  option2=[]
86
  option3=[]
87
  option4=[]
88
  for idx,ans in enumerate(ans_list):
89
+ #print("IDX: {}, ANS: {}".format(idx, ans))
90
  ques = get_question(summary_text,ans,question_model,question_tokenizer)
91
  other_options = get_related_word(ans)
92
  final_options, ans_index = get_final_option_list(ans,other_options)
 
112
  st.markdown("-----")
113
  questions.append(ques)
114
  output_path = "results/df_quiz_log_file_v1.csv"
 
 
 
 
 
 
 
 
 
115
  res_df = pd.DataFrame({"TimeStamp":[start_time]*len(ans_list),\
116
  "Input":[str(raw_text)]*len(ans_list),\
117
+ "Question":questions,"Option1":option1,\
118
+ "Option2":option2,\
119
+ "Option3":option3,\
120
+ "Option4":option4,\
121
  "Correct Answer":ans_list})
122
  res_df.to_csv(output_path, mode='a', index=False, sep="\t", header= not os.path.exists(output_path))
123
  # st.dataframe(pd.read_csv(output_path,sep="\t").tail(5))