broadfield-dev commited on
Commit
cb518f2
·
verified ·
1 Parent(s): dd2367e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -1,13 +1,21 @@
1
  import os
2
  from flask import Flask, render_template, request
3
  from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db
 
4
 
5
  app = Flask(__name__)
6
 
 
 
 
 
7
  @app.route('/', methods=['GET', 'POST'])
8
  def index():
 
9
  articles = fetch_rss_feeds()
 
10
  process_and_store_articles(articles)
 
11
  stored_docs = vector_db.similarity_search("news", k=len(articles))
12
  enriched_articles = [
13
  {
@@ -17,14 +25,16 @@ def index():
17
  "category": doc.metadata["category"],
18
  "sentiment": doc.metadata["sentiment"],
19
  "published": doc.metadata["published"],
20
- "image": doc.metadata.get("image", "https://via.placeholder.com/150"), # Default image if none
21
  }
22
  for doc in stored_docs
23
  ]
 
24
 
25
  if request.method == 'POST':
26
  query = request.form.get('search')
27
  if query:
 
28
  results = vector_db.similarity_search(query, k=10)
29
  enriched_articles = [
30
  {
@@ -34,10 +44,11 @@ def index():
34
  "category": doc.metadata["category"],
35
  "sentiment": doc.metadata["sentiment"],
36
  "published": doc.metadata["published"],
37
- "image": doc.metadata.get("image", "https://via.placeholder.com/150"),
38
  }
39
  for doc in results
40
  ]
 
41
 
42
  categorized_articles = {}
43
  for article in enriched_articles:
 
1
  import os
2
  from flask import Flask, render_template, request
3
  from rss_processor import fetch_rss_feeds, process_and_store_articles, vector_db
4
+ import logging
5
 
6
  app = Flask(__name__)
7
 
8
+ # Setup logging
9
+ logging.basicConfig(level=logging.INFO)
10
+ logger = logging.getLogger(__name__)
11
+
12
  @app.route('/', methods=['GET', 'POST'])
13
  def index():
14
+ logger.info("Starting to fetch RSS feeds")
15
  articles = fetch_rss_feeds()
16
+ logger.info(f"Fetched {len(articles)} articles")
17
  process_and_store_articles(articles)
18
+ logger.info("Articles processed and stored")
19
  stored_docs = vector_db.similarity_search("news", k=len(articles))
20
  enriched_articles = [
21
  {
 
25
  "category": doc.metadata["category"],
26
  "sentiment": doc.metadata["sentiment"],
27
  "published": doc.metadata["published"],
28
+ "image": doc.metadata.get("image", "svg"), # Use "svg" as a flag for default
29
  }
30
  for doc in stored_docs
31
  ]
32
+ logger.info(f"Enriched {len(enriched_articles)} articles for display")
33
 
34
  if request.method == 'POST':
35
  query = request.form.get('search')
36
  if query:
37
+ logger.info(f"Processing search query: {query}")
38
  results = vector_db.similarity_search(query, k=10)
39
  enriched_articles = [
40
  {
 
44
  "category": doc.metadata["category"],
45
  "sentiment": doc.metadata["sentiment"],
46
  "published": doc.metadata["published"],
47
+ "image": doc.metadata.get("image", "svg"),
48
  }
49
  for doc in results
50
  ]
51
+ logger.info(f"Search returned {len(enriched_articles)} results")
52
 
53
  categorized_articles = {}
54
  for article in enriched_articles: