Update app.py

app.py

@@ -1,4 +1,4 @@
-from flask import Flask, request, jsonify
+from flask import Flask, request, jsonify, send_from_directory
 from flask_cors import CORS
 from sentence_transformers import SentenceTransformer
 from pinecone import Pinecone
@@ -335,6 +335,7 @@ def semantic_search(query: str, top_k=1, category_filter=None, language='en'):
                         "summary": summary,
                         "costLKR": costLKR,
                         "category": category,
+                        "badge": proposal_data.get("badge", ""),  # Add badge field
                         "pdfUrl": f"assets/pdfs/{file_path}" if file_path else "",
                         "thumbUrl": f"assets/thumbs/{thumb_url}" if thumb_url else "",
                         "score": score,
@@ -393,6 +394,7 @@ def get_all_proposals(category_filter=None, language='en'):
                     "summary": summary,
                     "costLKR": costLKR,
                     "category": category,
+                    "badge": proposal_data.get("badge", ""),  # Add badge field
                     "pdfUrl": f"assets/pdfs/{file_path}" if file_path else "",
                     "thumbUrl": f"assets/thumbs/{thumb_url}" if thumb_url else "",
                     "score": 1.0,  # Default score for all proposals
@@ -560,6 +562,21 @@ def get_stats():
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
+@app.route('/assets/<path:filename>')
+def serve_assets(filename):
+    """Serve static assets like badge images"""
+    try:
+        # Check if the file exists in the Budget_Proposals copy-2/assets directory
+        assets_dir = os.path.join("Budget_Proposals copy-2", "assets")
+        if os.path.exists(os.path.join(assets_dir, filename)):
+            return send_from_directory(assets_dir, filename)
+        else:
+            # Fallback to current directory assets
+            return send_from_directory("assets", filename)
+    except Exception as e:
+        logger.error(f"Error serving asset {filename}: {e}")
+        return jsonify({"error": f"Asset not found: {filename}"}), 404
+
 @app.route('/', methods=['GET'])
 def home():
     """Home endpoint with API documentation"""