Update app/routes.py

app/routes.py

@@ -53,47 +53,76 @@ def landing():
         return redirect(url_for('auth.dashboard')) 
     return render_template('landing.html')
 
-@api.route('/rank-resumes', methods=['GET', 'POST'])
-def rank_resumes():
-    if request.method == 'GET':
-        return render_template("upload_rank_form.html")  # Optional: A form to test this route manually
+# @api.route('/rank-resumes', methods=['GET', 'POST'])
+# def rank_resumes():
+#     if request.method == 'GET':
+#         return render_template("upload_rank_form.html")  # Optional: A form to test this route manually
 
-    # 1. Extract and validate API key
-    auth_header = request.headers.get('Authorization')
-    if not auth_header or not auth_header.startswith("Bearer "):
-        return render_template("error.html", error="Missing or invalid Authorization header"), 401
+#     # 1. Extract and validate API key
+#     auth_header = request.headers.get('Authorization')
+#     if not auth_header or not auth_header.startswith("Bearer "):
+#         return render_template("error.html", error="Missing or invalid Authorization header"), 401
 
-    api_key = auth_header.split(" ")[1]
+#     api_key = auth_header.split(" ")[1]
 
-    # 2. Query Supabase to validate API key
-    response = requests.get(
-        f"{SUPABASE_URL}/rest/v1/users?api_key=eq.{api_key}",
-        headers=HEADERS
-    )
-    if response.status_code != 200 or not response.json():
-        return render_template("error.html", error="Invalid API key"), 403
+#     # 2. Query Supabase to validate API key
+#     response = requests.get(
+#         f"{SUPABASE_URL}/rest/v1/users?api_key=eq.{api_key}",
+#         headers=HEADERS
+#     )
+#     if response.status_code != 200 or not response.json():
+#         return render_template("error.html", error="Invalid API key"), 403
 
-    # 3. Parse form data
+#     # 3. Parse form data
+#     job_description = request.form.get('job_description')
+#     resume_files = request.files.getlist('resumes')
+
+#     if not job_description:
+#         return render_template("error.html", error="Missing job description"), 400
+#     if not resume_files:
+#         return render_template("error.html", error="No resumes uploaded"), 400
+
+#     # 4. Rank resumes
+#     try:
+#         results = rank_resumes_by_semantic_similarity(resume_files, job_description)
+
+#         return render_template(
+#             "resume_rankings.html",
+#             job_description=job_description,
+#             ranked_resumes=results
+#         )
+
+#     except Exception as e:
+#         return render_template("error.html", error=f"Error processing resumes: {str(e)}"), 500
+
+
+@auth.route('/rank-resumes', methods=['POST'])
+def rank_resumes_page():
+    api_key = request.form.get('api_key')
     job_description = request.form.get('job_description')
     resume_files = request.files.getlist('resumes')
 
-    if not job_description:
-        return render_template("error.html", error="Missing job description"), 400
-    if not resume_files:
-        return render_template("error.html", error="No resumes uploaded"), 400
+    if not api_key or not job_description or not resume_files:
+        flash("All fields are required.")
+        return redirect(url_for('auth.upload_rank_form'))
 
-    # 4. Rank resumes
-    try:
-        results = rank_resumes_by_semantic_similarity(resume_files, job_description)
+    # Validate API key with Supabase
+    response = requests.get(
+        f"{SUPABASE_URL}/rest/v1/users?api_key=eq.{api_key}",
+        headers=HEADERS
+    )
 
-        return render_template(
-            "resume_rankings.html",
-            job_description=job_description,
-            ranked_resumes=results
-        )
+    if response.status_code != 200 or not response.json():
+        flash("Invalid API key.")
+        return redirect(url_for('auth.upload_rank_form'))
 
+    try:
+        rankings = rank_resumes_by_semantic_similarity(resume_files, job_description)
     except Exception as e:
-        return render_template("error.html", error=f"Error processing resumes: {str(e)}"), 500
+        flash(f"Ranking error: {str(e)}")
+        return redirect(url_for('auth.upload_rank_form'))
+
+    return render_template('resume_rankings.html', rankings=rankings, job_description=job_description)