Update app.py

app.py

@@ -44,7 +44,7 @@ def update_news():
         db_write = DBWrite()
         old_news = db_read.read_news_from_db()
         new_news = get_news()
-        news_df = predict_news_category(old_news, new_news, interpreter, label_encoder, tokenizer)
+        news_df, is_db_updation_required = predict_news_category(old_news, new_news, interpreter, label_encoder, tokenizer)
         if news_df is None:
             raise Exception('Could not generate category predictions. Aborting the database insertion. No new articles are inserted into the collection.')
             
@@ -53,9 +53,12 @@ def update_news():
         logger.warning(f'Old News count: {old_news_count}\nNew News count: {new_news_count}')
         if new_news_count < old_news_count:
             raise Exception('New news count < Old news count. Aborting the database insertion. No new articles are inserted into the collection.')
-        
-        news_json = [*json.loads(news_df.reset_index(drop=True).to_json(orient="index")).values()]
-        db_write.insert_news_into_db(news_json)
+
+        if is_db_updation_required:
+            news_json = [*json.loads(news_df.reset_index(drop=True).to_json(orient="index")).values()]
+            db_write.insert_news_into_db(news_json)
+        else:
+            logger.warning('DB is not updated as it is not required.')
     except Exception as e:
         status_json = "{'status':'failure', 'message':'" + str(e) + "'}"
         status_code = 500