Update main.py

main.py

@@ -66,7 +66,11 @@ def predict_single_review(text_rating: TextRatingRequest):
     #     spans = [{'label': text, 'color': '', 'value': '', 'sentiment': '', 'score': ''}]
     #     has_sentiments = False
     # processed_data['spans'] = spans
-    processed_data, has_sentiments = process_single_comment(raw_data, LANGUAGE_MODEL, SENTIMENT_MODEL, LABELS )
+    try:
+        processed_data, has_sentiments = process_single_comment(raw_data, LANGUAGE_MODEL, SENTIMENT_MODEL, LABELS )
+    except Exception as e:
+        print("error during prediction: ", e)
+        return {"error":e}
     end_time = time.time()
     print(f"Time taken to process the data : {end_time - start_time} seconds")
     return {"processed_data": processed_data, "has_sentiments":has_sentiments}
@@ -79,7 +83,11 @@ def predict_all_reviews(reviews: AllTextRatingRequest):
     start_time = time.time()
     for review in reviews:
         raw_data = {"text":review.get('text', str()), "star_rating":review.get('rating', 5), "skip":skip}
-        processed_data, has_sentiments = process_single_comment(raw_data, LANGUAGE_MODEL, SENTIMENT_MODEL, LABELS )
+        try:
+            processed_data, has_sentiments = process_single_comment(raw_data, LANGUAGE_MODEL, SENTIMENT_MODEL, LABELS )
+        except Exception as e:
+            print("error during prediction: ", e)
+            return {"error":e}
         processed_data_list.append({"processed_data":processed_data, "has_sentiments":has_sentiments})
     end_time = time.time()
     print(f"Time taken to process the data : {end_time - start_time} seconds")
@@ -102,7 +110,11 @@ def predict_file_responses(file: UploadFile = File(...)):
             star_rating = row['STAR RATING']
             review_id = row['REVIEWID']
             raw_data = {"text":text, "star_rating":star_rating, "skip":False}
-            processed_data, has_sentiments = process_single_comment(raw_data,LANGUAGE_MODEL, SENTIMENT_MODEL, LABELS ) 
+            try:
+                processed_data, has_sentiments = process_single_comment(raw_data,LANGUAGE_MODEL, SENTIMENT_MODEL, LABELS ) 
+            except Exception as e:
+                print("error during prediction: ", e)
+                return {"error":e}
             now = datetime.now()
             print(f"Processed review with index {index} at time {now.time()}")
             processed_data_list.append({"processed_data":processed_data, "has_sentiments":has_sentiments, "review_id":review_id})