Spaces:
Sleeping
Sleeping
Aakash Goel
commited on
Commit
·
f8665e5
1
Parent(s):
f5ce1f3
log add 3
Browse files- code/helper.py +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":
|
127 |
-
"Option2":
|
128 |
-
"Option3":
|
129 |
-
"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))
|