Update app/routes.py

app/routes.py

@@ -342,6 +342,27 @@ def generate_api():
 def generate_api_key():
     return f"rk_live_{secrets.token_urlsafe(24)}"
 
+@auth.route('/generate-initial-api-key', methods=['POST'])
+def generate_initial_api_key():
+    if 'user_id' not in session:
+        flash('Please log in.')
+        return redirect(url_for('auth.login'))
+
+    user_id = session['user_id']
+    new_key = generate_api_key()
+
+    res = requests.patch(
+        f"{SUPABASE_URL}/rest/v1/users?id=eq.{user_id}",
+        headers={**HEADERS, "Content-Type": "application/json"},
+        json={"api_key": new_key}
+    )
+
+    if res.status_code == 200 or res.status_code == 204:
+        flash("API key generated successfully.")
+    else:
+        flash("Failed to generate API key. Please try again.")
+    return redirect(url_for('auth.api_keys'))
+
 @auth.route('/regenerate-api-key', methods=['POST'])
 def regenerate_api_key():
     if 'user_id' not in session:
@@ -357,11 +378,10 @@ def regenerate_api_key():
         json={"api_key": new_key}
     )
 
-    if res.status_code != 200:
-        flash("API key regenerated .")
-        return redirect(url_for('auth.api_keys'))
-
-    flash("API key regenerated successfully.")
+    if res.status_code == 200 or res.status_code == 204:
+        flash("API key regenerated successfully.")
+    else:
+        flash("Failed to regenerate API key. Please try again.")
     return redirect(url_for('auth.api_keys'))