Enhance subscription handling by adding subscription_plan_key to response and updating plan key formatting in Pricing Plans page

Author: Bharani0012

src/backend/base/langflow/services/paddle/subscriptions.py

@@ -72,16 +72,31 @@ async def has_active_subscription(
         return {
             "has_access": False,
             "subscription_status": None,
+            "subscription_plan_key": None,
             "paddle_subscription_id": None,
             "organisation_created_by": created_by,
         }
 
-    status = await get_subscription_status(subscription_id=sub_id, client=client)
+    paddle_client = client or get_paddle_client()
+    subscription = await asyncio.to_thread(
+        paddle_client.subscriptions.get,
+        sub_id,
+    )
+    status = getattr(subscription, "status", None)
+    if status:
+        status = str(status).lower()
     has_access = status in ACTIVE_SUBSCRIPTION_STATUSES if status else False
+    subscription_custom_data = _normalize_custom_data(
+        getattr(subscription, "custom_data", None)
+    )
+    plan_key = subscription_custom_data.get("plan_key")
+    if plan_key is not None:
+        plan_key = str(plan_key)
     logger.info(f"Subscription {sub_id} - status: {status}, has_access: {has_access}, active_statuses: {ACTIVE_SUBSCRIPTION_STATUSES}")
     return {
         "has_access": has_access,
         "subscription_status": status,
+        "subscription_plan_key": plan_key,
         "paddle_subscription_id": sub_id,
         "organisation_created_by": created_by,
     }
@@ -417,4 +432,4 @@ async def retry_with_backoff(
             )
             await asyncio.sleep(delay)
 
-    raise last_exception
+    raise last_exception

src/frontend/src/pages/SettingsPage/pages/PricingPlansPage/index.tsx

@@ -5,21 +5,41 @@ import ForwardedIconComponent from "@/components/common/genericIconComponent";
 import Loading from "@/components/ui/loading";
 import { api } from "@/controllers/API/api";
 import { getURL } from "@/controllers/API/helpers/constants";
+import planConfigData from "../../../../../public/plan_config.json";
 
 type BillingAccessResponse = {
   has_access?: boolean;
   subscription_status?: string | null;
+  subscription_plan_key?: string | null;
   paddle_subscription_id?: string | null;
 };
 
+const PLAN_NAME_BY_KEY = Object.fromEntries(
+  (planConfigData.plans ?? []).map((plan) => [
+    String(plan?.paddle?.plan_key ?? "").toLowerCase(),
+    String(plan?.name ?? "").trim(),
+  ]),
+) as Record<string, string>;
+
+function formatPlanKey(planKey: string): string {
+  return planKey
+    .replace(/[_-]+/g, " ")
+    .replace(/\s+/g, " ")
+    .trim()
+    .replace(/\b\w/g, (char) => char.toUpperCase());
+}
+
 function SelectedPlan({ billing }: { billing: BillingAccessResponse | null }) {
   const selectedPlan = useMemo(() => {
     if (!billing?.has_access) return "Free";
 
-    const status = (billing.subscription_status ?? "").toLowerCase();
-    if (status === "trialing") return "Paid (Trialing)";
+    const planKey = (billing.subscription_plan_key ?? "").toLowerCase();
+    const planName = PLAN_NAME_BY_KEY[planKey];
+    if (planName) return planName;
+    if (planKey) return formatPlanKey(planKey);
 
-    return "Paid";
+    const status = (billing.subscription_status ?? "").toLowerCase();
+    return status === "trialing" ? "Paid (Trialing)" : "Paid";
   }, [billing]);
 
   return (