broadfield-dev commited on
Commit
2aa963e
·
verified ·
1 Parent(s): 8a1ae42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -118,11 +118,15 @@ def index():
118
  def search():
119
  query = request.form.get('search')
120
  if not query:
 
121
  return render_template("index.html", categorized_articles={}, has_articles=False, loading=False)
122
 
123
  try:
124
  logger.info(f"Searching for: {query}")
 
125
  results = vector_db.similarity_search(query, k=10)
 
 
126
  enriched_articles = []
127
  seen_keys = set()
128
  for doc in results:
@@ -138,7 +142,7 @@ def search():
138
  "link": link,
139
  "description": meta.get("original_description", "No Description"),
140
  "category": meta.get("category", "Uncategorized"),
141
- "published": meta.get("published", "Unknown Date"),
142
  "image": meta.get("image", "svg"),
143
  })
144
 
@@ -147,6 +151,7 @@ def search():
147
  cat = article["category"]
148
  categorized_articles.setdefault(cat, []).append(article)
149
 
 
150
  return render_template("index.html", categorized_articles=categorized_articles, has_articles=bool(enriched_articles), loading=False)
151
  except Exception as e:
152
  logger.error(f"Search error: {e}")
 
118
  def search():
119
  query = request.form.get('search')
120
  if not query:
121
+ logger.info("Empty search query received")
122
  return render_template("index.html", categorized_articles={}, has_articles=False, loading=False)
123
 
124
  try:
125
  logger.info(f"Searching for: {query}")
126
+ # Embed the query using the same embedding model as vector_db
127
  results = vector_db.similarity_search(query, k=10)
128
+ logger.info(f"Search returned {len(results)} results")
129
+
130
  enriched_articles = []
131
  seen_keys = set()
132
  for doc in results:
 
142
  "link": link,
143
  "description": meta.get("original_description", "No Description"),
144
  "category": meta.get("category", "Uncategorized"),
145
+ "published": published,
146
  "image": meta.get("image", "svg"),
147
  })
148
 
 
151
  cat = article["category"]
152
  categorized_articles.setdefault(cat, []).append(article)
153
 
154
+ logger.info(f"Found {len(enriched_articles)} unique articles across {len(categorized_articles)} categories")
155
  return render_template("index.html", categorized_articles=categorized_articles, has_articles=bool(enriched_articles), loading=False)
156
  except Exception as e:
157
  logger.error(f"Search error: {e}")