Update app.py

app.py

@@ -391,68 +391,67 @@ def qa_list_gen(claim,srl_list,evidence):
 #-------------------------------------------------------------------------
 # Handle form submission
 if submitted and claim_text and evidence_text:
-    if not claim_text or not evidence_text:
-        st.caption(':green[Kindly hold on for a few minutes while the QA pairs are being generated]')
-        srl_list = list(itertools.chain(*[list(s) for s in srl(claim_text)]))
-        qa_list=qa_list_gen(claim_text,srl_list,evidence_text)
-        list_who = []
-        list_what = []
-        list_when = []
-        list_where = []
-        list_why = []
-        list_misc = []
-        
-        for item in qa_list:
-            question = item[0]
-            if any(x in question.lower() for x in ['who', 'what', 'when', 'where', 'why']):
-                if 'who' in question.lower():
-                    list_who.append(item)
-                elif 'what' in question.lower():
-                    list_what.append(item)
-                elif 'when' in question.lower():
-                    list_when.append(item)
-                elif 'where' in question.lower():
-                    list_where.append(item)
-                elif 'why' in question.lower():
-                    list_why.append(item)
-            else:
-                list_misc.append(item)
-        lists = [list_who, list_when, list_why, list_where, list_what]
-        for j, lst in enumerate(lists):
-            for i, l in enumerate(lst):
-                if l:  # check if list is not empty
-                    l[0] = f"Q{i+1}: {l[0]}"
-                    l[1] = f"Claim:- {l[1]}"
-                    if l[2]:
-                        l[2] = f"answer retrieved from evidence:- {l[2]}"
-                    else:
-                        l[2] = f"answer retrieved from evidence:- No mention of '{['who', 'when', 'why', 'where', 'what'][j]}' in any related documents."
-                        
-        for i in range(len(lists)):
-            if not lists[i]:
-                lists[i].extend([["No claims", "", f"No mention of '{['who', 'when', 'why', 'where', 'what'][i]}' in any related documents.", "❔ Not verifiable"]])
+    st.caption(':green[Kindly hold on for a few minutes while the QA pairs are being generated]')
+    srl_list = list(itertools.chain(*[list(s) for s in srl(claim_text)]))
+    qa_list=qa_list_gen(claim_text,srl_list,evidence_text)
+    list_who = []
+    list_what = []
+    list_when = []
+    list_where = []
+    list_why = []
+    list_misc = []
     
-        
-        final_df = pd.DataFrame(columns=['Who Claims', 'What Claims', 'When Claims', 'Where Claims', 'Why Claims', 'Misc Claims'])
-        
-        all_items_who = [item for item_list in list_who for item in item_list]
-        all_items_what = [item for item_list in list_what for item in item_list]
-        all_items_when = [item for item_list in list_when for item in item_list]
-        all_items_where = [item for item_list in list_where for item in item_list]
-        all_items_why = [item for item_list in list_why for item in item_list]
-        all_items_misc = [item for item_list in list_misc for item in item_list]
-        
-        
-        max_rows = max(len(all_items_who), len(all_items_what), len(all_items_when), len(all_items_where), len(all_items_why), len(all_items_misc))
-        
-        final_df['Who Claims'] = all_items_who + [''] * (max_rows - len(all_items_who))
-        final_df['What Claims'] = all_items_what + [''] * (max_rows - len(all_items_what))
-        final_df['When Claims'] = all_items_when + [''] * (max_rows - len(all_items_when))
-        final_df['Where Claims'] = all_items_where + [''] * (max_rows - len(all_items_where))
-        final_df['Why Claims'] = all_items_why + [''] * (max_rows - len(all_items_why))  
-        final_df['Misc Claims'] = all_items_misc + [''] * (max_rows - len(all_items_misc))  
-        st.write(f"""Claim : {claim_text}""")
-        st.write(f"""Evidence : {evidence_text}""")  
-        st.table(final_df) 
-    else:
-        st.warning("You need to input both the claim and evidence and then press Submit")
+    for item in qa_list:
+        question = item[0]
+        if any(x in question.lower() for x in ['who', 'what', 'when', 'where', 'why']):
+            if 'who' in question.lower():
+                list_who.append(item)
+            elif 'what' in question.lower():
+                list_what.append(item)
+            elif 'when' in question.lower():
+                list_when.append(item)
+            elif 'where' in question.lower():
+                list_where.append(item)
+            elif 'why' in question.lower():
+                list_why.append(item)
+        else:
+            list_misc.append(item)
+    lists = [list_who, list_when, list_why, list_where, list_what]
+    for j, lst in enumerate(lists):
+        for i, l in enumerate(lst):
+            if l:  # check if list is not empty
+                l[0] = f"Q{i+1}: {l[0]}"
+                l[1] = f"Claim:- {l[1]}"
+                if l[2]:
+                    l[2] = f"answer retrieved from evidence:- {l[2]}"
+                else:
+                    l[2] = f"answer retrieved from evidence:- No mention of '{['who', 'when', 'why', 'where', 'what'][j]}' in any related documents."
+                    
+    for i in range(len(lists)):
+        if not lists[i]:
+            lists[i].extend([["No claims", "", f"No mention of '{['who', 'when', 'why', 'where', 'what'][i]}' in any related documents.", "❔ Not verifiable"]])
+
+    
+    final_df = pd.DataFrame(columns=['Who Claims', 'What Claims', 'When Claims', 'Where Claims', 'Why Claims', 'Misc Claims'])
+    
+    all_items_who = [item for item_list in list_who for item in item_list]
+    all_items_what = [item for item_list in list_what for item in item_list]
+    all_items_when = [item for item_list in list_when for item in item_list]
+    all_items_where = [item for item_list in list_where for item in item_list]
+    all_items_why = [item for item_list in list_why for item in item_list]
+    all_items_misc = [item for item_list in list_misc for item in item_list]
+    
+    
+    max_rows = max(len(all_items_who), len(all_items_what), len(all_items_when), len(all_items_where), len(all_items_why), len(all_items_misc))
+    
+    final_df['Who Claims'] = all_items_who + [''] * (max_rows - len(all_items_who))
+    final_df['What Claims'] = all_items_what + [''] * (max_rows - len(all_items_what))
+    final_df['When Claims'] = all_items_when + [''] * (max_rows - len(all_items_when))
+    final_df['Where Claims'] = all_items_where + [''] * (max_rows - len(all_items_where))
+    final_df['Why Claims'] = all_items_why + [''] * (max_rows - len(all_items_why))  
+    final_df['Misc Claims'] = all_items_misc + [''] * (max_rows - len(all_items_misc))  
+    st.write(f"""Claim : {claim_text}""")
+    st.write(f"""Evidence : {evidence_text}""")  
+    st.table(final_df) 
+else:
+    st.warning("You need to input both the claim and evidence and then press Submit")