You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

index.tsx 24KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676
  1. 'use client'
  2. import type { FC } from 'react'
  3. import React, { useCallback, useEffect, useRef, useState } from 'react'
  4. import { useTranslation } from 'react-i18next'
  5. import {
  6. RiBookmark3Line,
  7. RiErrorWarningFill,
  8. } from '@remixicon/react'
  9. import { useBoolean } from 'ahooks'
  10. import { usePathname, useRouter, useSearchParams } from 'next/navigation'
  11. import TabHeader from '../../base/tab-header'
  12. import { checkOrSetAccessToken } from '../utils'
  13. import MenuDropdown from './menu-dropdown'
  14. import RunBatch from './run-batch'
  15. import ResDownload from './run-batch/res-download'
  16. import useBreakpoints, { MediaType } from '@/hooks/use-breakpoints'
  17. import RunOnce from '@/app/components/share/text-generation/run-once'
  18. import { fetchSavedMessage as doFetchSavedMessage, fetchAppInfo, fetchAppParams, removeMessage, saveMessage } from '@/service/share'
  19. import type { SiteInfo } from '@/models/share'
  20. import type {
  21. MoreLikeThisConfig,
  22. PromptConfig,
  23. SavedMessage,
  24. TextToSpeechConfig,
  25. } from '@/models/debug'
  26. import AppIcon from '@/app/components/base/app-icon'
  27. import Badge from '@/app/components/base/badge'
  28. import { changeLanguage } from '@/i18n/i18next-config'
  29. import Loading from '@/app/components/base/loading'
  30. import { userInputsFormToPromptVariables } from '@/utils/model-config'
  31. import Res from '@/app/components/share/text-generation/result'
  32. import SavedItems from '@/app/components/app/text-generate/saved-items'
  33. import type { InstalledApp } from '@/models/explore'
  34. import { DEFAULT_VALUE_MAX_LEN, appDefaultIconBackground } from '@/config'
  35. import Toast from '@/app/components/base/toast'
  36. import type { VisionFile, VisionSettings } from '@/types/app'
  37. import { Resolution, TransferMethod } from '@/types/app'
  38. import { useAppFavicon } from '@/hooks/use-app-favicon'
  39. import LogoSite from '@/app/components/base/logo/logo-site'
  40. import cn from '@/utils/classnames'
  41. const GROUP_SIZE = 5 // to avoid RPM(Request per minute) limit. The group task finished then the next group.
  42. enum TaskStatus {
  43. pending = 'pending',
  44. running = 'running',
  45. completed = 'completed',
  46. failed = 'failed',
  47. }
  48. type TaskParam = {
  49. inputs: Record<string, any>
  50. }
  51. type Task = {
  52. id: number
  53. status: TaskStatus
  54. params: TaskParam
  55. }
  56. export type IMainProps = {
  57. isInstalledApp?: boolean
  58. installedAppInfo?: InstalledApp
  59. isWorkflow?: boolean
  60. }
  61. const TextGeneration: FC<IMainProps> = ({
  62. isInstalledApp = false,
  63. installedAppInfo,
  64. isWorkflow = false,
  65. }) => {
  66. const { notify } = Toast
  67. const { t } = useTranslation()
  68. const media = useBreakpoints()
  69. const isPC = media === MediaType.pc
  70. const searchParams = useSearchParams()
  71. const mode = searchParams.get('mode') || 'create'
  72. const [currentTab, setCurrentTab] = useState<string>(['create', 'batch'].includes(mode) ? mode : 'create')
  73. const router = useRouter()
  74. const pathname = usePathname()
  75. useEffect(() => {
  76. const params = new URLSearchParams(searchParams)
  77. if (params.has('mode')) {
  78. params.delete('mode')
  79. router.replace(`${pathname}?${params.toString()}`)
  80. }
  81. // eslint-disable-next-line react-hooks/exhaustive-deps
  82. }, [])
  83. // Notice this situation isCallBatchAPI but not in batch tab
  84. const [isCallBatchAPI, setIsCallBatchAPI] = useState(false)
  85. const isInBatchTab = currentTab === 'batch'
  86. const [inputs, doSetInputs] = useState<Record<string, any>>({})
  87. const inputsRef = useRef(inputs)
  88. const setInputs = useCallback((newInputs: Record<string, any>) => {
  89. doSetInputs(newInputs)
  90. inputsRef.current = newInputs
  91. }, [])
  92. const [appId, setAppId] = useState<string>('')
  93. const [siteInfo, setSiteInfo] = useState<SiteInfo | null>(null)
  94. const [canReplaceLogo, setCanReplaceLogo] = useState<boolean>(false)
  95. const [customConfig, setCustomConfig] = useState<Record<string, any> | null>(null)
  96. const [promptConfig, setPromptConfig] = useState<PromptConfig | null>(null)
  97. const [moreLikeThisConfig, setMoreLikeThisConfig] = useState<MoreLikeThisConfig | null>(null)
  98. const [textToSpeechConfig, setTextToSpeechConfig] = useState<TextToSpeechConfig | null>(null)
  99. // save message
  100. const [savedMessages, setSavedMessages] = useState<SavedMessage[]>([])
  101. const fetchSavedMessage = async () => {
  102. const res: any = await doFetchSavedMessage(isInstalledApp, installedAppInfo?.id)
  103. setSavedMessages(res.data)
  104. }
  105. const handleSaveMessage = async (messageId: string) => {
  106. await saveMessage(messageId, isInstalledApp, installedAppInfo?.id)
  107. notify({ type: 'success', message: t('common.api.saved') })
  108. fetchSavedMessage()
  109. }
  110. const handleRemoveSavedMessage = async (messageId: string) => {
  111. await removeMessage(messageId, isInstalledApp, installedAppInfo?.id)
  112. notify({ type: 'success', message: t('common.api.remove') })
  113. fetchSavedMessage()
  114. }
  115. // send message task
  116. const [controlSend, setControlSend] = useState(0)
  117. const [controlStopResponding, setControlStopResponding] = useState(0)
  118. const [visionConfig, setVisionConfig] = useState<VisionSettings>({
  119. enabled: false,
  120. number_limits: 2,
  121. detail: Resolution.low,
  122. transfer_methods: [TransferMethod.local_file],
  123. })
  124. const [completionFiles, setCompletionFiles] = useState<VisionFile[]>([])
  125. const handleSend = () => {
  126. setIsCallBatchAPI(false)
  127. setControlSend(Date.now())
  128. // eslint-disable-next-line ts/no-use-before-define
  129. setAllTaskList([]) // clear batch task running status
  130. // eslint-disable-next-line ts/no-use-before-define
  131. showResultPanel()
  132. }
  133. const [controlRetry, setControlRetry] = useState(0)
  134. const handleRetryAllFailedTask = () => {
  135. setControlRetry(Date.now())
  136. }
  137. const [allTaskList, doSetAllTaskList] = useState<Task[]>([])
  138. const allTaskListRef = useRef<Task[]>([])
  139. const getLatestTaskList = () => allTaskListRef.current
  140. const setAllTaskList = (taskList: Task[]) => {
  141. doSetAllTaskList(taskList)
  142. allTaskListRef.current = taskList
  143. }
  144. const pendingTaskList = allTaskList.filter(task => task.status === TaskStatus.pending)
  145. const noPendingTask = pendingTaskList.length === 0
  146. const showTaskList = allTaskList.filter(task => task.status !== TaskStatus.pending)
  147. const [currGroupNum, doSetCurrGroupNum] = useState(0)
  148. const currGroupNumRef = useRef(0)
  149. const setCurrGroupNum = (num: number) => {
  150. doSetCurrGroupNum(num)
  151. currGroupNumRef.current = num
  152. }
  153. const getCurrGroupNum = () => {
  154. return currGroupNumRef.current
  155. }
  156. const allSuccessTaskList = allTaskList.filter(task => task.status === TaskStatus.completed)
  157. const allFailedTaskList = allTaskList.filter(task => task.status === TaskStatus.failed)
  158. const allTasksFinished = allTaskList.every(task => task.status === TaskStatus.completed)
  159. const allTasksRun = allTaskList.every(task => [TaskStatus.completed, TaskStatus.failed].includes(task.status))
  160. const [batchCompletionRes, doSetBatchCompletionRes] = useState<Record<string, string>>({})
  161. const batchCompletionResRef = useRef<Record<string, string>>({})
  162. const setBatchCompletionRes = (res: Record<string, string>) => {
  163. doSetBatchCompletionRes(res)
  164. batchCompletionResRef.current = res
  165. }
  166. const getBatchCompletionRes = () => batchCompletionResRef.current
  167. const exportRes = allTaskList.map((task) => {
  168. const batchCompletionResLatest = getBatchCompletionRes()
  169. const res: Record<string, string> = {}
  170. const { inputs } = task.params
  171. promptConfig?.prompt_variables.forEach((v) => {
  172. res[v.name] = inputs[v.key]
  173. })
  174. let result = batchCompletionResLatest[task.id]
  175. // task might return multiple fields, should marshal object to string
  176. if (typeof batchCompletionResLatest[task.id] === 'object')
  177. result = JSON.stringify(result)
  178. res[t('share.generation.completionResult')] = result
  179. return res
  180. })
  181. const checkBatchInputs = (data: string[][]) => {
  182. if (!data || data.length === 0) {
  183. notify({ type: 'error', message: t('share.generation.errorMsg.empty') })
  184. return false
  185. }
  186. const headerData = data[0]
  187. let isMapVarName = true
  188. promptConfig?.prompt_variables.forEach((item, index) => {
  189. if (!isMapVarName)
  190. return
  191. if (item.name !== headerData[index])
  192. isMapVarName = false
  193. })
  194. if (!isMapVarName) {
  195. notify({ type: 'error', message: t('share.generation.errorMsg.fileStructNotMatch') })
  196. return false
  197. }
  198. let payloadData = data.slice(1)
  199. if (payloadData.length === 0) {
  200. notify({ type: 'error', message: t('share.generation.errorMsg.atLeastOne') })
  201. return false
  202. }
  203. // check middle empty line
  204. const allEmptyLineIndexes = payloadData.filter(item => item.every(i => i === '')).map(item => payloadData.indexOf(item))
  205. if (allEmptyLineIndexes.length > 0) {
  206. let hasMiddleEmptyLine = false
  207. let startIndex = allEmptyLineIndexes[0] - 1
  208. allEmptyLineIndexes.forEach((index) => {
  209. if (hasMiddleEmptyLine)
  210. return
  211. if (startIndex + 1 !== index) {
  212. hasMiddleEmptyLine = true
  213. return
  214. }
  215. startIndex++
  216. })
  217. if (hasMiddleEmptyLine) {
  218. notify({ type: 'error', message: t('share.generation.errorMsg.emptyLine', { rowIndex: startIndex + 2 }) })
  219. return false
  220. }
  221. }
  222. // check row format
  223. payloadData = payloadData.filter(item => !item.every(i => i === ''))
  224. // after remove empty rows in the end, checked again
  225. if (payloadData.length === 0) {
  226. notify({ type: 'error', message: t('share.generation.errorMsg.atLeastOne') })
  227. return false
  228. }
  229. let errorRowIndex = 0
  230. let requiredVarName = ''
  231. let moreThanMaxLengthVarName = ''
  232. let maxLength = 0
  233. payloadData.forEach((item, index) => {
  234. if (errorRowIndex !== 0)
  235. return
  236. promptConfig?.prompt_variables.forEach((varItem, varIndex) => {
  237. if (errorRowIndex !== 0)
  238. return
  239. if (varItem.type === 'string') {
  240. const maxLen = varItem.max_length || DEFAULT_VALUE_MAX_LEN
  241. if (item[varIndex].length > maxLen) {
  242. moreThanMaxLengthVarName = varItem.name
  243. maxLength = maxLen
  244. errorRowIndex = index + 1
  245. return
  246. }
  247. }
  248. if (!varItem.required)
  249. return
  250. if (item[varIndex].trim() === '') {
  251. requiredVarName = varItem.name
  252. errorRowIndex = index + 1
  253. }
  254. })
  255. })
  256. if (errorRowIndex !== 0) {
  257. if (requiredVarName)
  258. notify({ type: 'error', message: t('share.generation.errorMsg.invalidLine', { rowIndex: errorRowIndex + 1, varName: requiredVarName }) })
  259. if (moreThanMaxLengthVarName)
  260. notify({ type: 'error', message: t('share.generation.errorMsg.moreThanMaxLengthLine', { rowIndex: errorRowIndex + 1, varName: moreThanMaxLengthVarName, maxLength }) })
  261. return false
  262. }
  263. return true
  264. }
  265. const handleRunBatch = (data: string[][]) => {
  266. if (!checkBatchInputs(data))
  267. return
  268. if (!allTasksFinished) {
  269. notify({ type: 'info', message: t('appDebug.errorMessage.waitForBatchResponse') })
  270. return
  271. }
  272. const payloadData = data.filter(item => !item.every(i => i === '')).slice(1)
  273. const varLen = promptConfig?.prompt_variables.length || 0
  274. setIsCallBatchAPI(true)
  275. const allTaskList: Task[] = payloadData.map((item, i) => {
  276. const inputs: Record<string, any> = {}
  277. if (varLen > 0) {
  278. item.slice(0, varLen).forEach((input, index) => {
  279. const varSchema = promptConfig?.prompt_variables[index]
  280. inputs[varSchema?.key as string] = input
  281. if (!input) {
  282. if (varSchema?.type === 'string' || varSchema?.type === 'paragraph')
  283. inputs[varSchema?.key as string] = ''
  284. else
  285. inputs[varSchema?.key as string] = undefined
  286. }
  287. })
  288. }
  289. return {
  290. id: i + 1,
  291. status: i < GROUP_SIZE ? TaskStatus.running : TaskStatus.pending,
  292. params: {
  293. inputs,
  294. },
  295. }
  296. })
  297. setAllTaskList(allTaskList)
  298. setCurrGroupNum(0)
  299. setControlSend(Date.now())
  300. // clear run once task status
  301. setControlStopResponding(Date.now())
  302. // eslint-disable-next-line ts/no-use-before-define
  303. showResultPanel()
  304. }
  305. const handleCompleted = (completionRes: string, taskId?: number, isSuccess?: boolean) => {
  306. const allTaskListLatest = getLatestTaskList()
  307. const batchCompletionResLatest = getBatchCompletionRes()
  308. const pendingTaskList = allTaskListLatest.filter(task => task.status === TaskStatus.pending)
  309. const runTasksCount = 1 + allTaskListLatest.filter(task => [TaskStatus.completed, TaskStatus.failed].includes(task.status)).length
  310. const needToAddNextGroupTask = (getCurrGroupNum() !== runTasksCount) && pendingTaskList.length > 0 && (runTasksCount % GROUP_SIZE === 0 || (allTaskListLatest.length - runTasksCount < GROUP_SIZE))
  311. // avoid add many task at the same time
  312. if (needToAddNextGroupTask)
  313. setCurrGroupNum(runTasksCount)
  314. const nextPendingTaskIds = needToAddNextGroupTask ? pendingTaskList.slice(0, GROUP_SIZE).map(item => item.id) : []
  315. const newAllTaskList = allTaskListLatest.map((item) => {
  316. if (item.id === taskId) {
  317. return {
  318. ...item,
  319. status: isSuccess ? TaskStatus.completed : TaskStatus.failed,
  320. }
  321. }
  322. if (needToAddNextGroupTask && nextPendingTaskIds.includes(item.id)) {
  323. return {
  324. ...item,
  325. status: TaskStatus.running,
  326. }
  327. }
  328. return item
  329. })
  330. setAllTaskList(newAllTaskList)
  331. if (taskId) {
  332. setBatchCompletionRes({
  333. ...batchCompletionResLatest,
  334. [`${taskId}`]: completionRes,
  335. })
  336. }
  337. }
  338. const fetchInitData = async () => {
  339. if (!isInstalledApp)
  340. await checkOrSetAccessToken()
  341. return Promise.all([
  342. isInstalledApp
  343. ? {
  344. app_id: installedAppInfo?.id,
  345. site: {
  346. title: installedAppInfo?.app.name,
  347. prompt_public: false,
  348. copyright: '',
  349. icon: installedAppInfo?.app.icon,
  350. icon_background: installedAppInfo?.app.icon_background,
  351. },
  352. plan: 'basic',
  353. }
  354. : fetchAppInfo(),
  355. fetchAppParams(isInstalledApp, installedAppInfo?.id),
  356. !isWorkflow
  357. ? fetchSavedMessage()
  358. : {},
  359. ])
  360. }
  361. useEffect(() => {
  362. (async () => {
  363. const [appData, appParams]: any = await fetchInitData()
  364. const { app_id: appId, site: siteInfo, can_replace_logo, custom_config } = appData
  365. setAppId(appId)
  366. setSiteInfo(siteInfo as SiteInfo)
  367. setCanReplaceLogo(can_replace_logo)
  368. setCustomConfig(custom_config)
  369. changeLanguage(siteInfo.default_language)
  370. const { user_input_form, more_like_this, file_upload, text_to_speech }: any = appParams
  371. setVisionConfig({
  372. // legacy of image upload compatible
  373. ...file_upload,
  374. transfer_methods: file_upload.allowed_file_upload_methods || file_upload.allowed_upload_methods,
  375. // legacy of image upload compatible
  376. image_file_size_limit: appParams?.system_parameters?.image_file_size_limit,
  377. fileUploadConfig: appParams?.system_parameters,
  378. })
  379. const prompt_variables = userInputsFormToPromptVariables(user_input_form)
  380. setPromptConfig({
  381. prompt_template: '', // placeholder for future
  382. prompt_variables,
  383. } as PromptConfig)
  384. setMoreLikeThisConfig(more_like_this)
  385. setTextToSpeechConfig(text_to_speech)
  386. })()
  387. }, [])
  388. // Can Use metadata(https://beta.nextjs.org/docs/api-reference/metadata) to set title. But it only works in server side client.
  389. useEffect(() => {
  390. if (siteInfo?.title) {
  391. if (canReplaceLogo)
  392. document.title = `${siteInfo.title}`
  393. else
  394. document.title = `${siteInfo.title} - Powered by Dify`
  395. }
  396. }, [siteInfo?.title, canReplaceLogo])
  397. useAppFavicon({
  398. enable: !isInstalledApp,
  399. icon_type: siteInfo?.icon_type,
  400. icon: siteInfo?.icon,
  401. icon_background: siteInfo?.icon_background,
  402. icon_url: siteInfo?.icon_url,
  403. })
  404. const [isShowResultPanel, { setTrue: doShowResultPanel, setFalse: hideResultPanel }] = useBoolean(false)
  405. const showResultPanel = () => {
  406. // fix: useClickAway hideResSidebar will close sidebar
  407. setTimeout(() => {
  408. doShowResultPanel()
  409. }, 0)
  410. }
  411. const [resultExisted, setResultExisted] = useState(false)
  412. const renderRes = (task?: Task) => (<Res
  413. key={task?.id}
  414. isWorkflow={isWorkflow}
  415. isCallBatchAPI={isCallBatchAPI}
  416. isPC={isPC}
  417. isMobile={!isPC}
  418. isInstalledApp={isInstalledApp}
  419. installedAppInfo={installedAppInfo}
  420. isError={task?.status === TaskStatus.failed}
  421. promptConfig={promptConfig}
  422. moreLikeThisEnabled={!!moreLikeThisConfig?.enabled}
  423. inputs={isCallBatchAPI ? (task as Task).params.inputs : inputs}
  424. controlSend={controlSend}
  425. controlRetry={task?.status === TaskStatus.failed ? controlRetry : 0}
  426. controlStopResponding={controlStopResponding}
  427. onShowRes={showResultPanel}
  428. handleSaveMessage={handleSaveMessage}
  429. taskId={task?.id}
  430. onCompleted={handleCompleted}
  431. visionConfig={visionConfig}
  432. completionFiles={completionFiles}
  433. isShowTextToSpeech={!!textToSpeechConfig?.enabled}
  434. siteInfo={siteInfo}
  435. onRunStart={() => setResultExisted(true)}
  436. />)
  437. const renderBatchRes = () => {
  438. return (showTaskList.map(task => renderRes(task)))
  439. }
  440. const renderResWrap = (
  441. <div
  442. className={cn(
  443. 'relative flex h-full flex-col',
  444. !isPC && 'h-[calc(100vh_-_36px)] rounded-t-2xl shadow-lg backdrop-blur-sm',
  445. !isPC
  446. ? isShowResultPanel
  447. ? 'bg-background-default-burn'
  448. : 'border-t-[0.5px] border-divider-regular bg-components-panel-bg'
  449. : 'bg-chatbot-bg',
  450. )}
  451. >
  452. {isCallBatchAPI && (
  453. <div className={cn(
  454. 'flex shrink-0 items-center justify-between px-14 pb-2 pt-9',
  455. !isPC && 'px-4 pb-1 pt-3',
  456. )}>
  457. <div className='system-md-semibold-uppercase text-text-primary'>{t('share.generation.executions', { num: allTaskList.length })}</div>
  458. {allSuccessTaskList.length > 0 && (
  459. <ResDownload
  460. isMobile={!isPC}
  461. values={exportRes}
  462. />
  463. )}
  464. </div>
  465. )}
  466. <div className={cn(
  467. 'flex h-0 grow flex-col overflow-y-auto',
  468. isPC && 'px-14 py-8',
  469. isPC && isCallBatchAPI && 'pt-0',
  470. !isPC && 'p-0 pb-2',
  471. )}>
  472. {!isCallBatchAPI ? renderRes() : renderBatchRes()}
  473. {!noPendingTask && (
  474. <div className='mt-4'>
  475. <Loading type='area' />
  476. </div>
  477. )}
  478. </div>
  479. {isCallBatchAPI && allFailedTaskList.length > 0 && (
  480. <div className='absolute bottom-6 left-1/2 z-10 flex -translate-x-1/2 items-center gap-2 rounded-xl border border-components-panel-border bg-components-panel-bg-blur p-3 shadow-lg backdrop-blur-sm'>
  481. <RiErrorWarningFill className='h-4 w-4 text-text-destructive' />
  482. <div className='system-sm-medium text-text-secondary'>{t('share.generation.batchFailed.info', { num: allFailedTaskList.length })}</div>
  483. <div className='h-3.5 w-px bg-divider-regular'></div>
  484. <div onClick={handleRetryAllFailedTask} className='system-sm-semibold-uppercase cursor-pointer text-text-accent'>{t('share.generation.batchFailed.retry')}</div>
  485. </div>
  486. )}
  487. </div>
  488. )
  489. if (!appId || !siteInfo || !promptConfig) {
  490. return (
  491. <div className='flex h-screen items-center'>
  492. <Loading type='app' />
  493. </div>)
  494. }
  495. return (
  496. <div className={cn(
  497. 'bg-background-default-burn',
  498. isPC && 'flex',
  499. !isPC && 'flex-col',
  500. isInstalledApp ? 'h-full rounded-2xl shadow-md' : 'h-screen',
  501. )}>
  502. {/* Left */}
  503. <div className={cn(
  504. 'relative flex h-full shrink-0 flex-col',
  505. isPC ? 'w-[600px] max-w-[50%]' : resultExisted ? 'h-[calc(100%_-_64px)]' : '',
  506. isInstalledApp && 'rounded-l-2xl',
  507. )}>
  508. {/* header */}
  509. <div className={cn('shrink-0 space-y-4 border-b border-divider-subtle', isPC ? 'bg-components-panel-bg p-8 pb-0' : 'p-4 pb-0')}>
  510. <div className='flex items-center gap-3'>
  511. <AppIcon
  512. size={isPC ? 'large' : 'small'}
  513. iconType={siteInfo.icon_type}
  514. icon={siteInfo.icon}
  515. background={siteInfo.icon_background || appDefaultIconBackground}
  516. imageUrl={siteInfo.icon_url}
  517. />
  518. <div className='system-md-semibold grow truncate text-text-secondary'>{siteInfo.title}</div>
  519. <MenuDropdown data={siteInfo} />
  520. </div>
  521. {siteInfo.description && (
  522. <div className='system-xs-regular text-text-tertiary'>{siteInfo.description}</div>
  523. )}
  524. <TabHeader
  525. items={[
  526. { id: 'create', name: t('share.generation.tabs.create') },
  527. { id: 'batch', name: t('share.generation.tabs.batch') },
  528. ...(!isWorkflow
  529. ? [{
  530. id: 'saved',
  531. name: t('share.generation.tabs.saved'),
  532. isRight: true,
  533. icon: <RiBookmark3Line className='h-4 w-4' />,
  534. extra: savedMessages.length > 0
  535. ? (
  536. <Badge className='ml-1'>
  537. {savedMessages.length}
  538. </Badge>
  539. )
  540. : null,
  541. }]
  542. : []),
  543. ]}
  544. value={currentTab}
  545. onChange={setCurrentTab}
  546. />
  547. </div>
  548. {/* form */}
  549. <div className={cn(
  550. 'h-0 grow overflow-y-auto bg-components-panel-bg',
  551. isPC ? 'px-8' : 'px-4',
  552. !isPC && resultExisted && customConfig?.remove_webapp_brand && 'rounded-b-2xl border-b-[0.5px] border-divider-regular',
  553. )}>
  554. <div className={cn(currentTab === 'create' ? 'block' : 'hidden')}>
  555. <RunOnce
  556. siteInfo={siteInfo}
  557. inputs={inputs}
  558. inputsRef={inputsRef}
  559. onInputsChange={setInputs}
  560. promptConfig={promptConfig}
  561. onSend={handleSend}
  562. visionConfig={visionConfig}
  563. onVisionFilesChange={setCompletionFiles}
  564. />
  565. </div>
  566. <div className={cn(isInBatchTab ? 'block' : 'hidden')}>
  567. <RunBatch
  568. vars={promptConfig.prompt_variables}
  569. onSend={handleRunBatch}
  570. isAllFinished={allTasksRun}
  571. />
  572. </div>
  573. {currentTab === 'saved' && (
  574. <SavedItems
  575. className={cn(isPC ? 'mt-6' : 'mt-4')}
  576. isShowTextToSpeech={textToSpeechConfig?.enabled}
  577. list={savedMessages}
  578. onRemove={handleRemoveSavedMessage}
  579. onStartCreateContent={() => setCurrentTab('create')}
  580. />
  581. )}
  582. </div>
  583. {/* powered by */}
  584. {!customConfig?.remove_webapp_brand && (
  585. <div className={cn(
  586. 'flex shrink-0 items-center gap-1.5 bg-components-panel-bg py-3',
  587. isPC ? 'px-8' : 'px-4',
  588. !isPC && resultExisted && 'rounded-b-2xl border-b-[0.5px] border-divider-regular',
  589. )}>
  590. <div className='system-2xs-medium-uppercase text-text-tertiary'>{t('share.chat.poweredBy')}</div>
  591. {customConfig?.replace_webapp_logo && (
  592. <img src={customConfig?.replace_webapp_logo} alt='logo' className='block h-5 w-auto' />
  593. )}
  594. {!customConfig?.replace_webapp_logo && (
  595. <LogoSite className='!h-5' />
  596. )}
  597. </div>
  598. )}
  599. </div>
  600. {/* Result */}
  601. <div className={cn(
  602. isPC
  603. ? 'h-full w-0 grow'
  604. : isShowResultPanel
  605. ? 'fixed inset-0 z-50 bg-background-overlay backdrop-blur-sm'
  606. : resultExisted
  607. ? 'relative h-16 shrink-0 overflow-hidden bg-background-default-burn pt-2.5'
  608. : '',
  609. )}>
  610. {!isPC && (
  611. <div
  612. className={cn(
  613. isShowResultPanel
  614. ? 'flex items-center justify-center p-2 pt-6'
  615. : 'absolute left-0 top-0 z-10 flex w-full items-center justify-center px-2 pb-[57px] pt-[3px]',
  616. )}
  617. onClick={() => {
  618. if (isShowResultPanel)
  619. hideResultPanel()
  620. else
  621. showResultPanel()
  622. }}
  623. >
  624. <div className='h-1 w-8 cursor-grab rounded bg-divider-solid'/>
  625. </div>
  626. )}
  627. {renderResWrap}
  628. </div>
  629. </div>
  630. )
  631. }
  632. export default TextGeneration