Update app.py

app.py

@@ -344,41 +344,41 @@ def gen_qa_who(df):
         claim=df['claim'][i]
         answer= split_ws(df["who"])
         evidence=df["evidence"][i]
-        #time.sleep(10)
+        time.sleep(10)
         if srl!="":
-        # try:
-            for j in range(0,len(answer)):
-                FACT_TO_GENERATE_QUESTION_FROM = f"""{answer[j]} [SEP] {claim}"""
-                #FACT_TO_GENERATE_QUESTION_FROM = f"""generate_who_based_question_from_context_using_the_next_answer:{answer[j]} [SEP] context:{claim}"""
-                #time.sleep(10)
-                question_ids = query({"inputs":FACT_TO_GENERATE_QUESTION_FROM, 
-                          "num_beams":5, 
-                          "early_stopping":True,
-                                     "min_length": 100,"wait_for_model":True})[0]['generated_text'].capitalize()
-                question_ids = rephrase_question_who(question_ids)
-                list_of_ques_who.append(f"""Q{j+1}:{question_ids}""")
-                list_of_ans_who.append(f"""Ans{j+1}:{answer[j]}""")
-                input_evidence = f"answer: {question_ids} context: {evidence}"
-                #time.sleep(10)
-                answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True}) #[0]['generated_text']
-                # if answer_evidence.lower() in evidence.lower():
-                #     list_of_evidence_answer_who.append(f"""Evidence{j+1}:{answer_evidence}""")
-                # else:
-                #     answer_evidence=""
-                #     list_of_evidence_answer_who.append(f"""No mention of 'who'in any related documents.""")
-                list_of_evidence_answer_who.append(f"""Evidence{j+1}:{answer_evidence}""")
-                threshold = 0.2
-                # list_of_pairs = [(answer_evidence, answer[j])]
-                # rouge_l_score = calc_rouge_l_score(answer_evidence, answer[j])
-                # if rouge_l_score >= threshold:
-                #     verification_status = 'Verified Valid'
-                # elif rouge_l_score == 0:
-                #     verification_status = 'Not verifiable'
-                # else:
-                #     verification_status = 'Verified False'
-                # rouge_l_scores.append(verification_status)
-            # except:
-            #     pass
+            try:
+                for j in range(0,len(answer)):
+                    FACT_TO_GENERATE_QUESTION_FROM = f"""{answer[j]} [SEP] {claim}"""
+                    #FACT_TO_GENERATE_QUESTION_FROM = f"""generate_who_based_question_from_context_using_the_next_answer:{answer[j]} [SEP] context:{claim}"""
+                    #time.sleep(10)
+                    question_ids = query({"inputs":FACT_TO_GENERATE_QUESTION_FROM, 
+                              "num_beams":5, 
+                              "early_stopping":True,
+                                         "min_length": 100,"wait_for_model":True})[0]['generated_text'].capitalize()
+                    question_ids = rephrase_question_who(question_ids)
+                    list_of_ques_who.append(f"""Q{j+1}:{question_ids}""")
+                    list_of_ans_who.append(f"""Ans{j+1}:{answer[j]}""")
+                    input_evidence = f"answer: {question_ids} context: {evidence}"
+                    time.sleep(10)
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
+                    if answer_evidence.lower() in evidence.lower():
+                        list_of_evidence_answer_who.append(f"""Evidence{j+1}:{answer_evidence}""")
+                    else:
+                        answer_evidence=""
+                        list_of_evidence_answer_who.append(f"""No mention of 'who'in any related documents.""")
+                    list_of_evidence_answer_who.append(f"""Evidence{j+1}:{answer_evidence}""")
+                    threshold = 0.2
+                    list_of_pairs = [(answer_evidence, answer[j])]
+                    rouge_l_score = calc_rouge_l_score(answer_evidence, answer[j])
+                    if rouge_l_score >= threshold:
+                        verification_status = 'Verified Valid'
+                    elif rouge_l_score == 0:
+                        verification_status = 'Not verifiable'
+                    else:
+                        verification_status = 'Verified False'
+                    rouge_l_scores.append(verification_status)
+                except:
+                    pass
         else:
             list_of_ques_who="No claims"
             list_of_ans_who=""
@@ -420,7 +420,7 @@ def gen_qa_what(df):
                     list_of_ans_what.append(f"""Ans{j+1}:{answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     #time.sleep(10)
-                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True})[0]['generated_text']
+                    answer_evidence = query_evidence({"inputs":input_evidence,"truncation":True,"wait_for_model":True})[0]['generated_text']
                     if answer_evidence.lower() in evidence.lower():
                         list_of_evidence_answer_what.append(f"""Evidence{j+1}:{answer_evidence}""")
                         
@@ -637,11 +637,13 @@ if claim_text:
         final_df = pd.DataFrame(columns=['Who Claims', 'What Claims', 'When Claims', 'Where Claims', 'Why Claims'])
 
         
-        # final_df["Who Claims"]=gen_qa_who(df)
-        # final_df["What Claims"]=gen_qa_what(df)
-        # final_df["When Claims"]=gen_qa_when(df)
-        # final_df["Where Claims"]=gen_qa_where(df)
-        # final_df["Why Claims"]=gen_qa_why(df)
+        final_df["Who Claims"]=gen_qa_who(df)
+        final_df["What Claims"]=gen_qa_what(df)
+        final_df["When Claims"]=gen_qa_when(df)
+        final_df["Where Claims"]=gen_qa_where(df)
+        final_df["Why Claims"]=gen_qa_why(df)
+        st.table(final_df)
+        st.write(df["claim"])
         st.write(df["evidence"])
-        st.write(gen_qa_who(df))
+        # st.write(gen_qa_who(df))
         # st.table(final_df)