fix badge logic

Author: icecrasher321

apps/sim/app/workspace/[workspaceId]/w/components/sidebar/components/usage-indicator/usage-indicator.tsx

@@ -22,7 +22,7 @@ const PLAN_NAMES = {
 } as const
 
 interface UsageIndicatorProps {
-  onClick?: (badgeType: 'add' | 'upgrade') => void
+  onClick?: () => void
 }
 
 export function UsageIndicator({ onClick }: UsageIndicatorProps) {
@@ -39,7 +39,7 @@ export function UsageIndicator({ onClick }: UsageIndicatorProps) {
   // Show skeleton while loading
   if (isLoading) {
     return (
-      <div className={CONTAINER_STYLES} onClick={() => onClick?.('upgrade')}>
+      <div className={CONTAINER_STYLES} onClick={() => onClick?.()}>
         <div className='space-y-2'>
           {/* Plan and usage info skeleton */}
           <div className='flex items-center justify-between'>
@@ -69,13 +69,10 @@ export function UsageIndicator({ onClick }: UsageIndicatorProps) {
   // Determine badge to show
   const billingStatus = useSubscriptionStore.getState().getBillingStatus()
   const isBlocked = billingStatus === 'blocked'
-  const showAddBadge =
-    !isBlocked && planType !== 'free' && planType !== 'enterprise' && usage.percentUsed >= 50
-  const badgeText = isBlocked ? 'Payment Failed' : planType === 'free' ? 'Upgrade' : 'Add'
-  const badgeType = isBlocked ? 'upgrade' : planType === 'free' ? 'upgrade' : 'add'
+  const badgeText = isBlocked ? 'Payment Failed' : planType === 'free' ? 'Upgrade' : undefined
 
   return (
-    <div className={CONTAINER_STYLES} onClick={() => onClick?.(badgeType)}>
+    <div className={CONTAINER_STYLES} onClick={() => onClick?.()}>
       <div className='space-y-2'>
         {/* Plan and usage info */}
         <div className='flex items-center justify-between'>
@@ -88,9 +85,7 @@ export function UsageIndicator({ onClick }: UsageIndicatorProps) {
             >
               {PLAN_NAMES[planType]}
             </span>
-            {(showAddBadge || planType === 'free') && (
-              <Badge className={GRADIENT_BADGE_STYLES}>{badgeText}</Badge>
-            )}
+            {badgeText ? <Badge className={GRADIENT_BADGE_STYLES}>{badgeText}</Badge> : null}
           </div>
           <span className='text-muted-foreground text-xs tabular-nums'>
             {isBlocked ? 'Payment required' : `$${usage.current.toFixed(2)} / $${usage.limit}`}

apps/sim/app/workspace/[workspaceId]/w/components/sidebar/sidebar.tsx

@@ -1008,10 +1008,9 @@ export function Sidebar() {
           style={{ bottom: `${navigationBottom + SIDEBAR_HEIGHTS.NAVIGATION + SIDEBAR_GAP}px` }} // Navigation height + gap
         >
           <UsageIndicator
-            onClick={(badgeType) => {
+            onClick={() => {
               const isBlocked = useSubscriptionStore.getState().getBillingStatus() === 'blocked'
-
-              if (isBlocked || badgeType === 'add') {
+              if (isBlocked) {
                 if (typeof window !== 'undefined') {
                   window.dispatchEvent(
                     new CustomEvent('open-settings', { detail: { tab: 'subscription' } })