diff --git a/frontend/src/components/AccountMenu.tsx b/frontend/src/components/AccountMenu.tsx index 1846df4d0..5f8271e2e 100644 --- a/frontend/src/components/AccountMenu.tsx +++ b/frontend/src/components/AccountMenu.tsx @@ -86,10 +86,11 @@ export function AccountMenu() { const hasStripeAccount = billingStatus?.stripe_customer_id !== null; const productName = billingStatus?.product_name || ""; const isPro = productName.toLowerCase().includes("pro"); + const isMax = productName.toLowerCase().includes("max"); const isStarter = productName.toLowerCase().includes("starter"); const isTeamPlan = productName.toLowerCase().includes("team"); - const showUpgrade = !isPro && !isTeamPlan; - const showManage = (isPro || isStarter || isTeamPlan) && hasStripeAccount; + const showUpgrade = !isMax && !isTeamPlan; + const showManage = (isPro || isMax || isStarter || isTeamPlan) && hasStripeAccount; // Fetch team status if user has team plan const { data: teamStatus } = useQuery({ diff --git a/frontend/src/components/BillingStatus.tsx b/frontend/src/components/BillingStatus.tsx index 81fa41183..9ff291df1 100644 --- a/frontend/src/components/BillingStatus.tsx +++ b/frontend/src/components/BillingStatus.tsx @@ -48,7 +48,7 @@ export function BillingStatus() { } const isFree = billingStatus.product_name.toLowerCase().includes("free"); - const isPro = billingStatus.product_name.toLowerCase().includes("pro"); + const isMax = billingStatus.product_name.toLowerCase().includes("max"); const getChatsText = () => { if (isFree) { @@ -58,7 +58,7 @@ export function BillingStatus() { return `Free Plan — ${billingStatus.chats_remaining} Message${billingStatus.chats_remaining === 1 ? "" : "s"} Left This Week`; } if (!billingStatus.can_chat) { - if (isPro) { + if (isMax) { return "Contact us to increase your limits"; } return "You've run out of messages, upgrade to keep chatting!"; @@ -80,7 +80,7 @@ export function BillingStatus() { - - - ); - })} - + )} + + + + ); + })} + + ); + })()}