Update news_category_prediction.py

news_category_prediction.py

@@ -45,6 +45,7 @@ def cols_check(new_cols, old_cols):
     
 def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interpreter, label_encoder, tokenizer):
     try:
+        db_updation_required = 1
         logger.warning('Entering predict_news_category()')
         logger.warning(f'old news: {old_news}\nnew_news: {new_news}')
         if not isinstance(new_news, pd.DataFrame):
@@ -92,7 +93,9 @@ def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interp
                 final_df.reset_index(drop=True, inplace=True)
                 final_df.loc[final_df['pred_proba']<CLASSIFIER_THRESHOLD, 'category'] = 'OTHERS'
             else:
-                raise Exception('INFO: Old & New Articles are the same. There is no requirement of updating them in the database. Database is not updated.')
+                logger.warning('INFO: Old & New Articles are the same. There is no requirement of updating them in the database. Database is not updated.')
+                db_updation_required = 0
+                final_df = old_news.copy()
                 
         
         if len(final_df) == 0:
@@ -102,5 +105,5 @@ def predict_news_category(old_news: pd.DataFrame, new_news: pd.DataFrame, interp
     except Exception as e:
         logger.warning(f'Unexcpected error in predict_news_category()\n{e}')
         return None
-    return final_df
+    return final_df, db_updation_required