Update app.py

app.py

@@ -8,6 +8,7 @@ import requests
 from PIL import Image
 import itertools
 
+st.set_page_config(page_title="FACTIFY - 5WQA", layout="wide")
 
 HF_SPACES_API_KEY = st.secrets["HF_token"]
 
@@ -116,11 +117,17 @@ def proc():
     st.write(st.session_state.text_key)
 
     
-claim_text=st.text_area("Enter your claim:", on_change=proc, key='text_key')
+# claim_text=st.text_area("Enter your claim:", on_change=proc, key='text_key')
 
 
         
-evidence_text=st.text_area("Enter your evidence:")
+# evidence_text=st.text_area("Enter your evidence:")
+with st.form(key="claim_evidence_form",clear_on_submit=True):
+    claim_text = st.text_input()
+    evidence_text = st.text_input()
+    submitted = st.form_submit_button("Submit")
+
+        
 
 # form_evidence = st.form(key='my_evidence')
 # form_evidence.text_input(label='Enter your evidence')
@@ -382,68 +389,70 @@ def qa_list_gen(claim,srl_list,evidence):
     return list_of_qa_pipeline    
     
 #-------------------------------------------------------------------------
-
-    
-if claim_text and 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"]])
-
-    
-    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))
+# Handle form submission
+if submitted and claim_text and evidence_text:
+    if not claim_text or not claim_text:
+        st.warning("You need to input both the claim and evidence and then press Submit")
+    else:
+        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"]])
     
-    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)                 
+        
+        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)