Update app.py

app.py

@@ -36,22 +36,20 @@ interpreter, label_encoder, tokenizer = load_model()
 @cross_origin()
 def update_news():
     logger.warning('Entering update_news()')
-    status_json = "{'status':'success'}"
+    status_json = "{'status':'success', 'message':'success'}"
     status_code = 200
     try:
         db_read = DBRead()
         db_write = DBWrite()
         old_news = db_read.read_news_from_db()
-        if old_news is None:
-            logger.warning('No old news Found')
         new_news = get_news()
-        if new_news is None:
-            raise Exception('No news news Found')
         news_df = predict_news_category(old_news, new_news, interpreter, label_encoder, tokenizer)
+        if news_df is None:
+            raise Exception('Could not generate category predictions')
         news_json = [*json.loads(news_df.reset_index(drop=True).to_json(orient="index")).values()]
         db_write.insert_news_into_db(news_json)
-    except:
-        status_json = "{'status':'failure'}"
+    except Exception as e:
+        status_json = "{'status':'failure', 'message':'" + str(e) + "'}"
         status_code = 500
         raise
     logger.warning('Exiting update_news()')