Update app/routes.py

app/routes.py

@@ -9,11 +9,12 @@ from .ranker import rank_resumes_by_semantic_similarity
 import traceback
 from werkzeug.security import generate_password_hash, check_password_hash
 from .supabase_client import get_user_by_email, create_user
-import os, requests
 import uuid
 from .supabase_config import SUPABASE_URL, SUPABASE_API_KEY
 import secrets
-
+import os
+import requests
+api = Blueprint('api', __name__)
 
 SUPABASE_URL = os.getenv("SUPABASE_URL")
 SUPABASE_API_KEY = os.getenv("SUPABASE_API_KEY")
@@ -40,47 +41,41 @@ def embed():
     
 
 
-@auth.route('/rank-resumes', methods=['POST'])
+
+
+@api.route('/rank-resumes', methods=['POST'])
 def rank_resumes():
+    # 1. Extract and validate API key
     auth_header = request.headers.get('Authorization')
-    if not auth_header or not auth_header.startswith('Bearer '):
+    if not auth_header or not auth_header.startswith("Bearer "):
         return jsonify({"error": "Missing or invalid Authorization header"}), 401
 
-    api_key = auth_header.split(" ")[1].strip()
+    api_key = auth_header.split(" ")[1]
 
-    #  Supabase: Check user by API key
-    headers = {
-        "apikey": SUPABASE_API_KEY,
-        "Authorization": f"Bearer {SUPABASE_API_KEY}"
-    }
-    user_check = requests.get(
+    # 2. Query Supabase to validate API key
+    response = requests.get(
         f"{SUPABASE_URL}/rest/v1/users?api_key=eq.{api_key}",
-        headers=headers
+        headers=HEADERS
     )
 
-    if user_check.status_code != 200 or not user_check.json():
+    if response.status_code != 200 or not response.json():
         return jsonify({"error": "Invalid API key"}), 403
 
-    # Proceed to ranking
-    resumes = request.files.getlist('resumes')
+    # 3. Parse form data
     job_description = request.form.get('job_description')
+    resume_files = request.files.getlist('resumes')
 
-    if not resumes or not job_description:
-        return jsonify({"error": "Missing resumes or job description"}), 400
-
-    ranked = rank_resumes_by_semantic_similarity(resumes, job_description)
-
-    return jsonify({
-        "job_description": job_description,
-        "ranked_resumes": ranked
-    })
-
-
-
-
-
-auth = Blueprint('auth', __name__)
+    if not job_description:
+        return jsonify({"error": "Missing job_description field"}), 400
+    if not resume_files:
+        return jsonify({"error": "No resumes uploaded"}), 400
 
+    # 4. Rank resumes
+    try:
+        results = rank_resumes_by_semantic_similarity(resume_files, job_description)
+        return jsonify({"results": results}), 200
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
 
 
 @auth.route('/register', methods=['GET', 'POST'])