Update app.py

app.py

@@ -1,7 +1,7 @@
 import os
 import threading
 from flask import Flask, render_template, request, jsonify
-from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db
+from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db, download_from_hf_hub, upload_to_hf_hub
 import logging
 import time
 from datetime import datetime
@@ -25,6 +25,8 @@ def load_feeds_in_background():
         process_and_store_articles(articles)
         last_update_time = time.time()  # Update timestamp when new articles are added
         logger.info("Background feed processing complete")
+        # Upload updated DB to Hugging Face Hub
+        upload_to_hf_hub()
         loading_complete = True
     except Exception as e:
         logger.error(f"Error in background feed loading: {e}")
@@ -35,6 +37,11 @@ def index():
     global loading_complete
     loading_complete = False  # Reset on each load
 
+    # Ensure Chroma DB is downloaded from Hugging Face Hub on first load
+    if not os.path.exists("chroma_db"):
+        logger.info("Downloading Chroma DB from Hugging Face Hub...")
+        download_from_hf_hub()
+
     # Start background feed loading
     threading.Thread(target=load_feeds_in_background, daemon=True).start()