Update app/routes.py

app/routes.py

@@ -450,6 +450,22 @@ def try_api():
     resumes = request.files.getlist('resumes')
     job_description = request.form.get('job_description')
 
+    if 'user_id' not in session:
+        flash('Please log in to Test the API.')
+        return redirect(url_for('auth.login'))
+
+    user_id = session['user_id']
+
+    # Fetch user info
+    user_res = requests.get(
+        f"{SUPABASE_URL}/rest/v1/users?id=eq.{user_id}&select=*",
+        headers=HEADERS
+    )
+    if user_res.status_code != 200 or not user_res.json():
+        flash('User not found.')
+        return redirect(url_for('auth.login'))
+    user = user_res.json()[0]
+    
     if not api_key:
         flash("API key is required.")
         return redirect(url_for('auth.try_api'))