Update app.py

app.py

@@ -4,9 +4,9 @@ import re
 import time
 import requests
 
-if "visibility" not in st.session_state:
-    st.session_state.visibility = "visible"
-    st.session_state.disabled = False
+# if "visibility" not in st.session_state:
+#     st.session_state.visibility = "visible"
+#     st.session_state.disabled = False
 
 
 HF_SPACES_API_KEY = st.secrets["HF_token"]
@@ -75,16 +75,16 @@ def proc():
 
 # st.text_area('enter text', on_change=proc, key='text_key')
 
-claim_text = st.text_input(
-    "Enter your claim 👇",
-    label_visibility="visible",
-    # placeholder=st.session_state.placeholder,
-)
+# claim_text = st.text_input(
+#     "Enter your claim 👇",
+#     label_visibility="visible",
+#     # placeholder=st.session_state.placeholder,
+# )
 
-if claim_text:
-    st.write("You entered: ", claim_text)
+# if claim_text:
+#     st.write("You entered: ", claim_text)
     
-# 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')
 
 # form_claim = st.form(key='my_claim')
 # form_claim.text_input(label='Enter your claim')
@@ -92,21 +92,21 @@ if claim_text:
 
 
 
-evidence_text = st.text_input(
-    "Enter your evidence 👇",
-    label_visibility="visible",
-    # disabled=st.session_state.disabled,
-    # placeholder=st.session_state.placeholder,
-)
+# evidence_text = st.text_input(
+#     "Enter your evidence 👇",
+#     label_visibility="visible",
+#     # disabled=st.session_state.disabled,
+#     # placeholder=st.session_state.placeholder,
+# )
 
-if evidence_text:
-    st.write("You entered: ", evidence_text)
+# if evidence_text:
+#     st.write("You entered: ", evidence_text)
         
 # evidence_text=st.text_area("Enter your evidence:")
 
-# form_evidence = st.form(key='my_evidence')
-# form_evidence.text_input(label='Enter your evidence')
-# evidence_text = form_evidence.form_submit_button(label='Submit')
+form_evidence = st.form(key='my_evidence')
+form_evidence.text_input(label='Enter your evidence')
+evidence_text = form_evidence.form_submit_button(label='Submit')
 
 if evidence_text:
     st.caption(':green[Kindly hold on for a few minutes while the QA pairs are being generated]')
@@ -122,6 +122,7 @@ predictor = Predictor.from_path("structured-prediction-srl-bert.tar.gz")
 
 #---------------------------------------------------------------
 #---------------------------------------------------------------
+@st.cache
 def claim(text):
     import re
     def remove_special_chars(text):
@@ -328,6 +329,7 @@ def claim(text):
     data=df[["claim","who","what","why","when","where"]].copy()    
     return data
 #-------------------------------------------------------------------------
+@st.cache
 def split_ws(input_list, delimiter="<sep>"):
     output_list = []
     for item in input_list:
@@ -339,13 +341,14 @@ def split_ws(input_list, delimiter="<sep>"):
     return output_list
 
 #--------------------------------------------------------------------------    
+@st.cache
 def calc_rouge_l_score(list_of_evidence, list_of_ans):
     scorer = rouge_scorer.RougeScorer(['rougeL'], use_stemmer=True)
     scores = scorer.score(' '.join(list_of_evidence), ' '.join(list_of_ans))
     return scores['rougeL'].fmeasure
 #-------------------------------------------------------------------------
     
-
+@st.cache
 def rephrase_question_who(question):
     if not question.lower().startswith("who"):
         words = question.split()
@@ -354,6 +357,7 @@ def rephrase_question_who(question):
     else:
         return question
 #------------------------------------------------------------------------    
+@st.cache
 def gen_qa_who(df):
     list_of_ques_who=[]
     list_of_ans_who=[]
@@ -405,7 +409,7 @@ def gen_qa_who(df):
             rouge_l_scores="Not verifiable"
     return list_of_ques_who,list_of_ans_who,rouge_l_scores,list_of_evidence_answer_who
 #------------------------------------------------------------    
-
+@st.cache
 def rephrase_question_what(question):
     if not question.lower().startswith("what"):
         words = question.split()
@@ -414,6 +418,7 @@ def rephrase_question_what(question):
     else:
         return question
 #----------------------------------------------------------        
+@st.cache
 def gen_qa_what(df):
     list_of_ques_what=[]
     list_of_ans_what=[]
@@ -465,7 +470,7 @@ def gen_qa_what(df):
             rouge_l_scores="Not verifiable"
     return list_of_ques_what,list_of_ans_what,rouge_l_scores,list_of_evidence_answer_what
 #----------------------------------------------------------    
-
+@st.cache
 def rephrase_question_why(question):
     if not question.lower().startswith("why"):
         words = question.split()
@@ -475,6 +480,7 @@ def rephrase_question_why(question):
         return question
 
 #---------------------------------------------------------        
+@st.cache
 def gen_qa_why(df):
     list_of_ques_why=[]
     list_of_ans_why=[]
@@ -526,7 +532,7 @@ def gen_qa_why(df):
     return list_of_ques_why,list_of_ans_why,rouge_l_scores,list_of_evidence_answer_why
 
 #---------------------------------------------------------    
-
+@st.cache
 def rephrase_question_when(question):
     if not question.lower().startswith("when"):
         words = question.split()
@@ -535,6 +541,7 @@ def rephrase_question_when(question):
     else:
         return question
 #---------------------------------------------------------        
+@st.cache
 def gen_qa_when(df):
     list_of_ques_when=[]
     list_of_ans_when=[]
@@ -586,7 +593,7 @@ def gen_qa_when(df):
     return list_of_ques_when,list_of_ans_when,rouge_l_scores,list_of_evidence_answer_when
 
 #------------------------------------------------------   
-
+@st.cache
 def rephrase_question_where(question):
     if not question.lower().startswith("where"):
         words = question.split()
@@ -595,6 +602,7 @@ def rephrase_question_where(question):
     else:
         return question
 #------------------------------------------------------      
+@st.cache
 def gen_qa_where(df):
     list_of_ques_where=[]
     list_of_ans_where=[]