Update rss_processor.py

rss_processor.py

@@ -1,5 +1,6 @@
 import os
 import feedparser
+import sys
 from huggingface_hub import HfApi, login
 from langchain.vectorstores import Chroma
 from langchain.embeddings import HuggingFaceEmbeddings
@@ -172,4 +173,9 @@ def upload_to_hf_hub():
                     logger.info(f"Uploaded {file} to {REPO_ID}")
                 except Exception as e:
                     logger.error(f"Error uploading file {file}: {e}")
-        logger.info(f"Database uploaded to: {REPO_ID}")
+        logger.info(f"Database uploaded to: {REPO_ID}")
+
+if __name__ == "__main__":
+    if len(sys.argv) > 1 and sys.argv[1] == "load_feeds":
+        articles = fetch_rss_feeds()
+        process_and_store_articles(articles)