From 929b3299484719b6655ca552ae6810d6ba1c2b34 Mon Sep 17 00:00:00 2001
From: sharath <29162020+tnfssc@users.noreply.github.com>
Date: Fri, 27 Jun 2025 15:39:58 +0000
Subject: [PATCH] hotfix(billing): integrate BillingModal into
 ChatSettingsDialog, MessageInput, and InputNode components

---
 .../components/thread/chat-input/chat-settings-dialog.tsx | 8 ++++++++
 .../src/components/thread/chat-input/message-input.tsx    | 8 ++++++++
 .../src/components/thread/chat-input/model-selector.tsx   | 7 -------
 frontend/src/components/workflows/nodes/InputNode.tsx     | 8 ++++++++
 4 files changed, 24 insertions(+), 7 deletions(-)

diff --git a/frontend/src/components/thread/chat-input/chat-settings-dialog.tsx b/frontend/src/components/thread/chat-input/chat-settings-dialog.tsx
index f6279b90..49ab99d4 100644
--- a/frontend/src/components/thread/chat-input/chat-settings-dialog.tsx
+++ b/frontend/src/components/thread/chat-input/chat-settings-dialog.tsx
@@ -14,6 +14,7 @@ import { Label } from '@/components/ui/label';
 import { ModelSelector } from './model-selector';
 import { SubscriptionStatus } from './_use-model-selection';
 import { cn } from '@/lib/utils';
+import { BillingModal } from '@/components/billing/billing-modal';
 
 interface ChatSettingsDialogProps {
   selectedModel: string;
@@ -91,6 +92,13 @@ export function ChatSettingsDialog({
                 setBillingModalOpen={setBillingModalOpen}
               />
             </div>
+            {/* Billing Modal */}
+            <BillingModal
+              open={billingModalOpen}
+              onOpenChange={setBillingModalOpen}
+              returnUrl={typeof window !== 'undefined' ? window.location.href : '/'}
+            />
+
             <p className="text-xs text-muted-foreground">
               Choose the AI model that best fits your needs. Premium models offer better performance.
             </p>
diff --git a/frontend/src/components/thread/chat-input/message-input.tsx b/frontend/src/components/thread/chat-input/message-input.tsx
index b5e5d731..dfb31b24 100644
--- a/frontend/src/components/thread/chat-input/message-input.tsx
+++ b/frontend/src/components/thread/chat-input/message-input.tsx
@@ -14,6 +14,7 @@ import { useFeatureFlag } from '@/lib/feature-flags';
 import { TooltipContent } from '@/components/ui/tooltip';
 import { Tooltip } from '@/components/ui/tooltip';
 import { TooltipProvider, TooltipTrigger } from '@radix-ui/react-tooltip';
+import { BillingModal } from '@/components/billing/billing-modal';
 
 interface MessageInputProps {
   value: string;
@@ -200,6 +201,13 @@ export const MessageInput = forwardRef<HTMLTextAreaElement, MessageInputProps>(
               />
             )}
             
+            {/* Billing Modal */}
+            <BillingModal
+              open={billingModalOpen}
+              onOpenChange={setBillingModalOpen}
+              returnUrl={typeof window !== 'undefined' ? window.location.href : '/'}
+            />
+
             <VoiceRecorder
               onTranscription={onTranscription}
               disabled={loading || (disabled && !isAgentRunning)}
diff --git a/frontend/src/components/thread/chat-input/model-selector.tsx b/frontend/src/components/thread/chat-input/model-selector.tsx
index 04a31e87..ba64c319 100644
--- a/frontend/src/components/thread/chat-input/model-selector.tsx
+++ b/frontend/src/components/thread/chat-input/model-selector.tsx
@@ -763,13 +763,6 @@ export const ModelSelector: React.FC<ModelSelectorProps> = ({
         mode={dialogMode}
       />
 
-      {/* Billing Modal */}
-      <BillingModal
-        open={billingModalOpen}
-        onOpenChange={setBillingModalOpen}
-        returnUrl={typeof window !== 'undefined' ? window.location.href : '/'}
-      />
-
       {paywallOpen && (
         <PaywallDialog
           open={true}
diff --git a/frontend/src/components/workflows/nodes/InputNode.tsx b/frontend/src/components/workflows/nodes/InputNode.tsx
index 9afd8b07..e7f5d166 100644
--- a/frontend/src/components/workflows/nodes/InputNode.tsx
+++ b/frontend/src/components/workflows/nodes/InputNode.tsx
@@ -18,6 +18,7 @@ import { ScheduleConfigDialog } from "../scheduling/ScheduleConfigDialog";
 import { ScheduleConfig } from "../scheduling/types";
 import { ModelSelector } from "@/components/thread/chat-input/model-selector";
 import { useModelSelection } from "@/components/thread/chat-input/_use-model-selection";
+import { BillingModal } from "@/components/billing/billing-modal";
 
 interface InputNodeData {
   label?: string;
@@ -233,6 +234,13 @@ const InputNode = memo(({ data, selected, id }: NodeProps) => {
               </div>
             </div>
 
+            {/* Billing Modal */}
+            <BillingModal
+              open={billingModalOpen}
+              onOpenChange={setBillingModalOpen}
+              returnUrl={typeof window !== 'undefined' ? window.location.href : '/'}
+            />
+
             <div className="space-y-2">
               <Label className="text-sm font-medium">Trigger Type *</Label>
               <Select