Use ad choice component

Author: jahooma

cli/src/chat.tsx

@@ -11,9 +11,8 @@ import {
 } from 'react'
 import { useShallow } from 'zustand/react/shallow'
 
-import { getAdsEnabled, handleAdsDisable } from './commands/ads'
+import { getAdsEnabled } from './commands/ads'
 import { routeUserPrompt, addBashMessageToHistory } from './commands/router'
-import { AdBanner } from './components/ad-banner'
 import { ChoiceAdBanner } from './components/choice-ad-banner'
 import { ChatInputBar } from './components/chat-input-bar'
 import { LoadPreviousButton } from './components/load-previous-button'
@@ -175,13 +174,7 @@ export const Chat = ({
   })
   const hasSubscription = subscriptionData?.hasSubscription ?? false
 
-  const { ad, adData, recordImpression } = useGravityAd({ enabled: IS_FREEBUFF || !hasSubscription })
-  const [adsManuallyDisabled, setAdsManuallyDisabled] = useState(false)
-
-  const handleDisableAds = useCallback(() => {
-    handleAdsDisable()
-    setAdsManuallyDisabled(true)
-  }, [])
+  const { adData, recordImpression } = useGravityAd({ enabled: IS_FREEBUFF || !hasSubscription })
 
   // Set initial mode from CLI flag on mount
   useEffect(() => {
@@ -1466,19 +1459,11 @@ export const Chat = ({
           />
         )}
 
-        {ad && (IS_FREEBUFF || (!adsManuallyDisabled && getAdsEnabled())) && (
-          adData?.variant === 'choice' ? (
-            <ChoiceAdBanner
-              ads={adData.ads}
-              onImpression={recordImpression}
-            />
-          ) : (
-            <AdBanner
-              ad={ad}
-              onDisableAds={handleDisableAds}
-              isFreeMode={IS_FREEBUFF}
-            />
-          )
+        {adData && (IS_FREEBUFF || getAdsEnabled()) && (
+          <ChoiceAdBanner
+            ads={adData.variant === 'choice' ? adData.ads : [adData.ad]}
+            onImpression={recordImpression}
+          />
         )}
 
         {reviewMode ? (

cli/src/components/ad-banner.tsx

@@ -2,7 +2,6 @@ import { TextAttributes } from '@opentui/core'
 import { useRenderer } from '@opentui/react'
 import React, { useMemo, useState } from 'react'
 
-import { AdBanner } from './ad-banner'
 import { Button } from './button'
 import { ChoiceAdBanner } from './choice-ad-banner'
 import { FreebuffModelSelector } from './freebuff-model-selector'
@@ -74,7 +73,7 @@ export const WaitingRoomScreen: React.FC<WaitingRoomScreenProps> = ({
   // forceStart bypasses the "wait for first user message" gate inside the hook,
   // which would otherwise block ads here since no conversation exists yet.
   // Uses Carbon (BuySellAds); in-chat ads still use the Gravity default.
-  const { ad, adData, recordImpression } = useGravityAd({
+  const { adData, recordImpression } = useGravityAd({
     enabled: true,
     forceStart: true,
     provider: 'carbon',
@@ -263,21 +262,17 @@ export const WaitingRoomScreen: React.FC<WaitingRoomScreenProps> = ({
       </box>
 
       {/* Ad banner pinned to the bottom, same look-and-feel as in chat. */}
-      {ad && (
+      {adData && (
         <box style={{ flexShrink: 0 }}>
-          {adData?.variant === 'choice' ? (
-            <ChoiceAdBanner
-              ads={adData.ads}
-              onImpression={recordImpression}
-            />
-          ) : (
-            <AdBanner ad={ad} onDisableAds={() => {}} isFreeMode />
-          )}
+          <ChoiceAdBanner
+            ads={adData.variant === 'choice' ? adData.ads : [adData.ad]}
+            onImpression={recordImpression}
+          />
         </box>
       )}
 
       {/* Horizontal separator (mirrors chat input divider style) */}
-      {!ad && (
+      {!adData && (
         <text style={{ fg: theme.muted, flexShrink: 0 }}>
           {'─'.repeat(terminalWidth)}
         </text>