broadfield-dev commited on
Commit
2c0d9b1
·
verified ·
1 Parent(s): 15033cb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -119,11 +119,10 @@ 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
 
@@ -152,10 +151,13 @@ def search():
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}")
158
- return render_template("index.html", categorized_articles={}, has_articles=False, loading=False)
159
 
160
  @app.route('/check_loading')
161
  def check_loading():
 
119
  query = request.form.get('search')
120
  if not query:
121
  logger.info("Empty search query received")
122
+ return jsonify({"articles": {}, "has_articles": False}), 200
123
 
124
  try:
125
  logger.info(f"Searching for: {query}")
 
126
  results = vector_db.similarity_search(query, k=10)
127
  logger.info(f"Search returned {len(results)} results")
128
 
 
151
  categorized_articles.setdefault(cat, []).append(article)
152
 
153
  logger.info(f"Found {len(enriched_articles)} unique articles across {len(categorized_articles)} categories")
154
+ return jsonify({
155
+ "articles": categorized_articles,
156
+ "has_articles": bool(enriched_articles)
157
+ }), 200
158
  except Exception as e:
159
  logger.error(f"Search error: {e}")
160
+ return jsonify({"error": "Failed to perform search. Please try again.", "articles": {}, "has_articles": False}), 500
161
 
162
  @app.route('/check_loading')
163
  def check_loading():