Update functions.py

functions.py

@@ -68,6 +68,20 @@ def split_text_by_token_limit(text, tokenizer, max_tokens):
     return chunks
 
 
+# def safe_sentiment(sentiment_pipeline, text, length, progress_bar):
+#     try:
+#         result = sentiment_pipeline(text)[0]
+#     except Exception as e:
+#         result = None
+
+#     if "count" not in st.session_state:
+#         st.session_state.count = 0
+#     st.session_state.count += 1
+    
+#     progress = st.session_state.count / length
+#     progress_bar.progress(progress)
+#     return result
+
 def safe_sentiment(sentiment_pipeline, text, length, progress_bar):
     try:
         result = sentiment_pipeline(text)[0]
@@ -77,13 +91,16 @@ def safe_sentiment(sentiment_pipeline, text, length, progress_bar):
     if "count" not in st.session_state:
         st.session_state.count = 0
     st.session_state.count += 1
-    
+
     progress = st.session_state.count / length
+
+    #  Clamp the progress value between 0.0 and 1.0
+    progress = min(max(progress, 0.0), 1.0)
+
     progress_bar.progress(progress)
     return result
 
 
-
 def safe_sentiment_batch(sentiment_pipeline, texts):
     try:
         results = sentiment_pipeline(texts)