Update app.py

app.py

@@ -364,13 +364,13 @@ def gen_qa_who(df):
                               "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]}""")
+                    list_of_ques_who.append(f"""Q{j+1} :\n {question_ids}""")
+                    list_of_ans_who.append(f"""Claim :\n {answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     time.sleep(5)
                     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}""")
+                        list_of_evidence_answer_who.append(f"""Answer retrieved from evidence :\n {answer_evidence}""")
                     else:
                         answer_evidence=""
                         list_of_evidence_answer_who.append(f"""No mention of 'who'in any related documents.""")
@@ -425,12 +425,12 @@ def gen_qa_what(df):
                                          "min_length": 100,"wait_for_model":True})[0]['generated_text'].capitalize()
                     question_ids = rephrase_question_what(question_ids)
                     list_of_ques_what.append(f"""Q{j+1}:{question_ids}""")
-                    list_of_ans_what.append(f"""Ans{j+1}:{answer[j]}""")
+                    list_of_ans_what.append(f"""Claim :\n {answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     time.sleep(5)
                     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}""")
+                        list_of_evidence_answer_what.append(f"""Answer retrieved from evidence :\n {answer_evidence}""")
                         
                     else:
                         answer_evidence=""
@@ -487,12 +487,12 @@ def gen_qa_why(df):
                                          "min_length": 100,"wait_for_model":True})[0]['generated_text'].capitalize()
                     question_ids = rephrase_question_why(question_ids)
                     list_of_ques_why.append(f"""Q{j+1}:{question_ids}""")
-                    list_of_ans_why.append(f"""Ans{j+1}:{answer[j]}""")
+                    list_of_ans_why.append(f"""Claim :\n {answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     time.sleep(5)
                     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_why.append(f"""Evidence{j+1}:{answer_evidence}""")
+                        list_of_evidence_answer_why.append(f"""Answer retrieved from evidence :\n {answer_evidence}""")
                     else:
                         answer_evidence=""
                         list_of_evidence_answer_why.append(f"""No mention of 'why'in any related documents.""")
@@ -548,12 +548,12 @@ def gen_qa_when(df):
                                          "min_length": 100,"wait_for_model":True})[0]['generated_text'].capitalize()
                     question_ids = rephrase_question_when(question_ids)
                     list_of_ques_when.append(f"""Q{j+1}:{question_ids}""")
-                    list_of_ans_when.append(f"""Ans{j+1}:{answer[j]}""")
+                    list_of_ans_when.append(f"""Claim :\n {answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     time.sleep(5)
                     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_when.append(f"""Evidence{j+1}:{answer_evidence}""")
+                        list_of_evidence_answer_when.append(f"""Answer retrieved from evidence :\n {answer_evidence}""")
                     else:
                         answer_evidence=""
                         list_of_evidence_answer_when.append(f"""No mention of 'when'in any related documents.""")
@@ -609,12 +609,12 @@ def gen_qa_where(df):
                                          "min_length": 100,"wait_for_model":True})[0]['generated_text'].capitalize()
                     question_ids = rephrase_question_where(question_ids)
                     list_of_ques_where.append(f"""Q{j+1}:{question_ids}""")
-                    list_of_ans_where.append(f"""Ans{j+1}:{answer[j]}""")
+                    list_of_ans_where.append(f"""Claim :\n {answer[j]}""")
                     input_evidence = f"answer_the_next_question_from_context: {question_ids} context: {evidence}"
                     time.sleep(5)
                     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_where.append(f"""Evidence{j+1}:{answer_evidence}""")
+                        list_of_evidence_answer_where.append(f"""Answer retrieved from evidence :\n {answer_evidence}""")
                     else:
                         answer_evidence=""
                         list_of_evidence_answer_where.append(f"""No mention of 'where'in any related documents.""")
@@ -658,18 +658,88 @@ def gen_qa_where(df):
 #         # st.write(gen_qa_who(df))
 #         # st.table(final_df)
 
+# if claim_text and evidence_text:
+#     st.write("You entered: ", claim_text)
+#     st.write("You entered: ", evidence_text)
+#     st.caption(':green[Kindly hold on for a few minutes while the QA pairs are being generated]')
+#     df=claim(claim_text)
+#     df["evidence"]=evidence_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)
+    # st.table(final_df)
+    # st.write(df["claim"])
+    # st.write(df["evidence"])
+
 if claim_text and evidence_text:
     st.write("You entered: ", claim_text)
     st.write("You entered: ", evidence_text)
     st.caption(':green[Kindly hold on for a few minutes while the QA pairs are being generated]')
     df=claim(claim_text)
     df["evidence"]=evidence_text
+    lst1=gen_qa_who(df6)
+    lst2=gen_qa_what(df6)
+    lst3=gen_qa_when(df6)
+    lst4=gen_qa_where(df6)
+    lst5=gen_qa_why(df6)
+
+    output1=[]
+    if 'No claims' in lst1[0]:
+        output1=[item for item in lst1]
+    else:
+        for i in range(len(lst1[0])):
+            output1.append(lst1[0][i])
+            output1.append(lst1[1][i])
+            output1.append(lst1[2][i])
+            output1.append(lst1[3][i])
+    output2=[]
+    if 'No claims' in lst2[0]:
+        output2=[item for item in lst2]
+    else:
+        for i in range(len(lst2[0])):
+            output2.append(lst2[0][i])
+            output2.append(lst2[1][i])
+            output2.append(lst2[2][i])
+            output2.append(lst2[3][i])
+    output3=[]
+    if 'No claims' in lst3[0]:
+        output3=[item for item in lst3]
+    else:
+        for i in range(len(lst3[0])):
+            output3.append(lst3[0][i])
+            output3.append(lst3[1][i])
+            output3.append(lst3[2][i])
+            output3.append(lst3[3][i])
+    output4=[]
+    if 'No claims' in lst4[0]:
+        output4=[item for item in lst4]
+    else:
+        for i in range(len(lst4[0])):
+            output4.append(lst4[0][i])
+            output4.append(lst4[1][i])
+            output4.append(lst4[2][i])
+            output4.append(lst4[3][i])
+    output5=[]
+    if 'No claims' in lst5[0]:
+        output5=[item for item in lst5]
+    else:
+        for i in range(len(lst5[0])):
+            output5.append(lst5[0][i])
+            output5.append(lst5[1][i])
+            output5.append(lst5[2][i])
+            output5.append(lst5[3][i])
+    max_rows = max(len(output1), len(output2), len(output3), len(output4), len(output5))
     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)
+    # add the data to the dataframe
+    final_df['Who Claims'] = output1 + [''] * (max_rows - len(output1))
+    final_df['What Claims'] = output2 + [''] * (max_rows - len(output2))
+    final_df['When Claims'] = output3 + [''] * (max_rows - len(output3))
+    final_df['Where Claims'] = output4 + [''] * (max_rows - len(output4))
+    final_df['Why Claims'] = output5 + [''] * (max_rows - len(output5))    
     st.table(final_df)
     st.write(df["claim"])
-    st.write(df["evidence"])
+    st.write(df["evidence"])    
+