Update src/streamlit_app.py

src/streamlit_app.py

@@ -67,7 +67,28 @@ st.markdown(
     """,
     unsafe_allow_html=True)
 expander = st.expander("**Important notes**")
-expander.write(""" **How to Use the HR.ai web app:** 1. Type or paste your text into the text area, then press Ctrl + Enter.2. Click the 'Results' button to extract and tag entities in your text data. Results are presented in easy-to-read tables, visualized in an interactive tree map, pie chart and bar chart, and are available for download along with a Glossary of tags.      **How to Use the Question-Answering feature:**1. Type or paste your text into the text area, then press Ctrl + Enter. 2. Click the 'Add Question' button to add your question to the Record of Questions. You can manage your questions by deleting them one by one.3. Click the 'Extract Answers' button to extract the answer to your question.Results are presented in an easy-to-read table, visualized in an interactive tree map, and is available for download.   **Entities:** "Email", "Phone_number", "Street_address", "City", "Country", "Date_of_birth", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Terminated", "Retired", "Job_title", "Date", "Organization", "Role", "Performance_score", "Leave_of_absence", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance", "Pay_rate", "Annual_salary", "Tax", "Deductions", "Interview_type", "Applicant", "Referral", "Job_board", "Recruiter", "Offer_letter", "Agreement", "Certification", "Skill"**Usage Limits:** You can request results unlimited times for one (1) month.      **Supported Languages:** English     **Technical issues:** If your connection times out, please refresh the page or reopen the app's URL. For any errors or inquiries, please contact us at info@nlpblogs.com""")
+expander.write(""" **How to Use the HR.ai web app:** 
+
+1. Type or paste your text into the text area, then press Ctrl + Enter.
+2. Click the 'Results' button to extract and tag entities in your text data. 
+Results are presented in easy-to-read tables, visualized in an interactive tree map, pie chart and bar chart, and are available for download along with a Glossary of tags.      
+
+**How to Use the Question-Answering feature:**
+
+1. Type or paste your text into the text area, then press Ctrl + Enter. 
+2. Click the 'Add Question' button to add your question to the Record of Questions. You can manage your questions by deleting them one by one.
+3. Click the 'Extract Answers' button to extract the answer to your question.
+
+Results are presented in an easy-to-read table, visualized in an interactive tree map, and is available for download.   
+
+**Entities:** "Email", "Phone_number", "Street_address", "City", "Country", "Date_of_birth", "Marital_status", "Person", "Full_time", "Part_time", "Contract", "Terminated", "Retired", "Job_title", "Date", "Organization", "Role", "Performance_score", "Leave_of_absence", "Retirement_plan", "Bonus", "Stock_options", "Health_insurance", "Pay_rate", "Annual_salary", "Tax", "Deductions", "Interview_type", "Applicant", "Referral", "Job_board", "Recruiter", "Offer_letter", "Agreement", "Certification", "Skill"**Usage Limits:** You can request results unlimited times for one (1) month.      
+
+**Supported Languages:** English     
+
+**Technical issues:** If your connection times out, please refresh the page or reopen the app's URL. 
+
+For any errors or inquiries, please contact us at info@nlpblogs.com""")
+
 with st.sidebar:
     st.write("Use the following code to embed the web app on your website. Feel free to adjust the width and height values to fit your page.")
     code = '''
@@ -132,10 +153,10 @@ with tab1:
     # Load model for this tab
     model_hr = load_gliner_model("HR_AI")
     
-    # Define the word limit
+    # Define the word limit for this tab
     word_limit = 200
     
-    text = st.text_area("Type or paste your text below (max 200 words), and then press Ctrl + Enter", height=250, key='my_text_area_hr')
+    text = st.text_area(f"Type or paste your text below (max {word_limit} words), and then press Ctrl + Enter", height=250, key='my_text_area_hr')
     
     # Calculate and display the word count
     word_count = len(text.split())
@@ -237,9 +258,19 @@ with tab1:
 with tab2:
     # Load model for this tab
     model_qa = load_gliner_model("InfoFinder")
-    user_text = st.text_area("Type or paste your text below, and then press Ctrl + Enter", height=250, key='my_text_area_infofinder')
+    
+    # Define the word limit for this tab
+    word_limit_qa = 500
+    
+    user_text = st.text_area(f"Type or paste your text below (max {word_limit_qa} words), and then press Ctrl + Enter", height=250, key='my_text_area_infofinder')
+    
+    # Calculate and display the word count
+    word_count_qa = len(user_text.split())
+    st.markdown(f"**Word count:** {word_count_qa}/{word_limit_qa}")
+    
     def clear_text_qa():
         st.session_state['my_text_area_infofinder'] = ""
+    
     st.button("Clear text", on_click=clear_text_qa, key="clear_qa")
     st.subheader("Question-Answering", divider="green")
     question_input = st.text_input("Ask wh-questions. **Wh-questions begin with what, when, where, who, whom, which, whose, why and how. We use them to ask for specific information.**")
@@ -269,6 +300,8 @@ with tab2:
     if st.button("Extract Answers"):
         if not user_text.strip():
             st.warning("Please enter some text to analyze.")
+        elif word_count_qa > word_limit_qa:
+            st.warning(f"Your text exceeds the {word_limit_qa} word limit. Please shorten it to continue.")
         elif not st.session_state.user_labels:
             st.warning("Please define at least one question.")
         else: