Update app.py

app.py

@@ -36,7 +36,6 @@ def load_feeds_in_background():
 def index():
     global loading_complete, last_update_time
 
-    # Check if the database needs to be loaded (first time or empty)
     db_exists = os.path.exists("chroma_db") and vector_db.get().get('documents')
     if not db_exists:
         loading_complete = False
@@ -44,13 +43,14 @@ def index():
         download_from_hf_hub()
         threading.Thread(target=load_feeds_in_background, daemon=True).start()
     elif not loading_complete:
-        pass  # Let background loading continue
+        pass
     else:
         loading_complete = True
 
     try:
-        # Retrieve all articles from Chroma DB
         all_docs = vector_db.get(include=['documents', 'metadatas'])
+        total_docs = len(all_docs['documents']) if all_docs.get('documents') else 0
+        logger.info(f"Total articles in DB: {total_docs}")
         if not all_docs.get('metadatas'):
             logger.info("No articles in DB yet")
             return render_template("index.html", categorized_articles={}, has_articles=False, loading=not loading_complete)