Update app.py

app.py

@@ -1,7 +1,8 @@
 import os
-from flask import Flask, render_template, request
+from flask import Flask, render_template, request, Response
 from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db
 import logging
+import time
 
 app = Flask(__name__)
 
@@ -9,14 +10,25 @@ app = Flask(__name__)
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-@app.route('/', methods=['GET', 'POST'])
-def index():
-    logger.info("Starting to fetch RSS feeds")
+@app.route('/')
+def loading():
+    return render_template("loading.html")
+
+@app.route('/load_feeds', methods=['GET'])
+def load_feeds():
+    logger.info("Starting to fetch and process RSS feeds")
+    start_time = time.time()
     articles = fetch_rss_feeds()
     logger.info(f"Fetched {len(articles)} articles")
     process_and_store_articles(articles)
     logger.info("Articles processed and stored")
-    stored_docs = vector_db.similarity_search("news", k=len(articles))
+    end_time = time.time()
+    logger.info(f"RSS feed loading took {end_time - start_time:.2f} seconds")
+    return Response("Feeds loaded", status=200)
+
+@app.route('/index', methods=['GET'])
+def index():
+    stored_docs = vector_db.similarity_search("news", k=1000)  # Increased k to ensure all unique articles
     # Use a set to ensure unique articles by title and link
     unique_articles = {}
     for doc in stored_docs: