Update main.py

main.py

@@ -31,7 +31,7 @@ def predict_single_review(text_rating: TextRatingRequest):
     #     spans = [{'label': text, 'color': '', 'value': '', 'sentiment': '', 'score': ''}]
     #     has_sentiments = False
     # processed_data['spans'] = spans
-    processed_data = process_single_comment(raw_data)
+    processed_data, has_sentiments = process_single_comment(raw_data)
     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}
@@ -44,8 +44,8 @@ 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('star_rating', 1), "skip":skip}
-        processed_data = process_single_comment(raw_data)
-        porcessed_data_list.append(processed_data)
+        processed_data, has_sentiments = process_single_comment(raw_data)
+        porcessed_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")
     return processed_data_list