Update news_category_similar_news_prediction.py

news_category_similar_news_prediction.py

@@ -165,7 +165,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
             # final_df.loc[(final_df['title'].str.contains('Zodiac Sign', case=False)) | (final_df['title'].str.contains('Horoscope', case=False)), 'category'] = 'SCIENCE'
             # logger.warning('Updated category of articles having Zodiac Sign in title to SCIENCE')
 
-            final_df = process_prediction_df(final_df, df_type="production & archive")
+            final_df = process_prediction_df(final_df, df_type="production & prediction")
             prediction_df = final_df.copy()
             
         else:
@@ -192,8 +192,8 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
                 final_df = pd.concat([old_news, new_news], axis=0, ignore_index=True)
 
                 final_df = process_prediction_df(final_df, df_type="production")
-                archive_df = new_news.copy()
-                archive_df = process_prediction_df(archive_df, df_type="archive")
+                prediction_df = new_news.copy()
+                prediction_df = process_prediction_df(prediction_df, df_type="prediction")
                 
                 # final_df.drop_duplicates(subset='url', keep='first', inplace=True)
                 # final_df.reset_index(drop=True, inplace=True)
@@ -206,7 +206,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
                 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()
-                archive_df = final_df.copy()
+                prediction_df = final_df.copy()
                 
         
         if len(final_df) == 0:
@@ -216,7 +216,7 @@ def predict_news_category_similar_news(old_news: pd.DataFrame, new_news: pd.Data
     except Exception as e:
         logger.warning(f'Unexcpected error in predict_news_category()\n{e}')
         final_df = None
-        archive_df = final_df.copy()
+        prediction_df = final_df.copy()
         db_updation_required = 0
-    return final_df, archive_df, db_updation_required
+    return final_df, prediction_df, db_updation_required