Update config.py

config.py

@@ -1,6 +1,11 @@
+import os
+
+
 COLLECT_PREDICTION_DATA = 1
-DATABASE = "test_rss_news_db_cat_pred_sim_news"
-COLLECTION = "test_rss_news_cat_pred_sim_news"
+PRODUCTION_URL = os.getenv("PRODUCTION_DB_URL")
+PRODUCTION_DATABASE = "test_rss_news_db_cat_pred_sim_news"
+PRODUCTION_COLLECTION = "test_rss_news_cat_pred_sim_news"
+PREDICTION_URL = os.getenv("PREDICTION_DB_URL")
 PREDICTION_DATABASE = "test_indian_news_predictions_db"
 PREDICTION_COLLECTION = "test_indian_news_predictions_coll"
 NEWS_EXTRACTOR_URL_TIMEOUT = 5