@@ -1,4 +1,5 @@
 import React, { FC, useCallback, useEffect, useRef } from 'react'
+import { Typography } from '@mui/material'
 import { TextFieldProps } from '@mui/material/TextField/TextField'
 
 import classes from './model-input.module.scss'
@@ -7,6 +8,8 @@ import { PredictPrice } from './predict-price'
 import { LoadImage } from '#/app/components/input_components/load_image'
 import { SendBtn } from '#/app/components/input_components/send_button'
 import { IModelInputs } from '#/shared/api/models/models'
+import { TooltipCustom } from '#/shared'
+import { SvgIcon } from '#/shared/ui/svg'
 
 function buildTypeVersionsMap(inputs: IModelInputs[]): Record<string, string[]> {
 	const byType = new Map<string, string[][]>()
@@ -217,7 +220,18 @@ export const ModelInput: FC<Input & TextFieldProps> = ({
 			></textarea>
 
 			<div className={classes.endAdornment}>
-				<PredictPrice token={Number(predictedPrice)} />
+				{styles === 'chats' ? (
+					<PredictPrice token={Number(predictedPrice)} />
+				) : (
+					<TooltipCustom title='Стоимость генерации' placement='top'>
+						<div className={classes.predictPrice}>
+							<SvgIcon className={classes.generationIcon} url='/svg/chatgpt/generation-icon.svg' width={12} height={16} />
+							<Typography fontSize={14} fontWeight={600} color='#7F7DF3'>
+								{Math.ceil(Number(predictedPrice))}
+							</Typography>
+						</div>
+					</TooltipCustom>
+				)}
 				{hasAttachInput && (
 						<>
 							{!blocked && (
@@ -6,6 +6,13 @@ import classes from './predict-price.module.scss'
 const HIGH_COST_TOOLTIP =
 	'Генерация может выйти очень дорогой, т.к вы ввели очень большой запрос или в чате накопились большие сообщения. Если хотите снизить стоимость генерации, создайте новый чат или уменьшите количество запросов'
 
+const MOCK_TOKENS = [1, 76, 151] as const
+
+function getMockToken(): number {
+	const randomIndex = Math.floor(Math.random() * MOCK_TOKENS.length)
+	return MOCK_TOKENS[randomIndex]
+}
+
 function tierModifierClass(token: number): string {
 	if (token < 75) return classes['predictPrice--low']
 	if (token < 150) return classes['predictPrice--mid']
@@ -13,8 +20,10 @@ function tierModifierClass(token: number): string {
 }
 
 export function PredictPrice({ token }: { token: number }) {
-	const modifier = tierModifierClass(token)
-	const isHighCost = token >= 150
+	const mockToken = getMockToken()
+	// const mockToken = token
+	const modifier = tierModifierClass(mockToken)
+	const isHighCost = mockToken >= 150
 
 	return (
 		<TooltipCustom
@@ -24,7 +33,7 @@ export function PredictPrice({ token }: { token: number }) {
 		>
 			<div className={[classes.predictPrice, modifier].filter(Boolean).join(' ')}>
 				<SvgIcon className={classes.generationIcon} url='/svg/chatgpt/generation-icon.svg' width={12} height={16} />
-				<span className={classes.value}>{token}</span>
+				<span className={classes.value}>{mockToken}</span>
 			</div>
 		</TooltipCustom>
 	)