Added Calculate Score Functionality

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import pipeline
 import re
 from collections import Counter
+import string
 
 @st.cache_resource
 def load_qa_pipeline():
@@ -11,17 +12,13 @@ def normalize_answer(s):
     """Lower text and remove punctuation, articles and extra whitespace."""
     def remove_articles(text):
         return re.sub(r'\b(a|an|the)\b', ' ', text)
-
     def white_space_fix(text):
         return ' '.join(text.split())
-
     def remove_punc(text):
         exclude = set(string.punctuation)
         return ''.join(ch for ch in text if ch not in exclude)
-
     def lower(text):
         return text.lower()
-
     return white_space_fix(remove_articles(remove_punc(lower(s))))
 
 def compute_exact_match(prediction, ground_truth):
@@ -47,9 +44,25 @@ def main():
 
     # User input for context
     context = st.text_area("Enter the context:", height=200)
+    context = context.strip()  # Remove leading/trailing whitespace
 
     # User input for question
     question = st.text_input("Enter your question:")
+    question = question.strip()  # Remove leading/trailing whitespace
+
+    # Check for context and question length
+    if len(context) > 1500:
+        st.warning("Context should not exceed 1500 characters.")
+        return
+    if len(question) > 150:
+        st.warning("Question should not exceed 150 characters.")
+        return
+
+    # Option to calculate scores
+    calculate_scores = st.checkbox("Calculate scores")
+
+    if calculate_scores:
+        actual_answer = st.text_input("Enter the actual answer:")
 
     if st.button("Get Answer"):
         if context and question:
@@ -61,22 +74,13 @@ def main():
             st.write(result['answer'])
             st.write(f"Confidence: {result['score']:.2f}")
 
-            # Store the result for later use
-            st.session_state.last_answer = result['answer']
-
-            # Show option to calculate scores
-            st.subheader("Calculate Scores")
-            if st.checkbox("Show score calculation"):
-                actual_answer = st.text_input("Enter the actual answer:")
-                if st.button("Calculate Scores"):
-                    if actual_answer:
-                        em_score = compute_exact_match(result['answer'], actual_answer)
-                        f1_score = compute_f1(result['answer'], actual_answer)
-                        st.subheader("Scores:")
-                        st.write(f"Exact Match: {em_score}")
-                        st.write(f"F1 Score: {f1_score:.4f}")
-                    else:
-                        st.warning("Please enter the actual answer.")
+            # Calculate and display scores if option is selected
+            if calculate_scores and actual_answer:
+                em_score = compute_exact_match(result['answer'], actual_answer)
+                f1_score = compute_f1(result['answer'], actual_answer)
+                st.subheader("Scores:")
+                st.write(f"Exact Match: {em_score}")
+                st.write(f"F1 Score: {f1_score:.4f}")
         else:
             st.warning("Please provide both context and question.")