index.tsx 5.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141
  1. import type { FC, FormEvent } from 'react'
  2. import React from 'react'
  3. import { useTranslation } from 'react-i18next'
  4. import {
  5. PlayIcon,
  6. } from '@heroicons/react/24/solid'
  7. import Select from '@/app/components/base/select'
  8. import type { SiteInfo } from '@/models/share'
  9. import type { PromptConfig } from '@/models/debug'
  10. import Button from '@/app/components/base/button'
  11. import { DEFAULT_VALUE_MAX_LEN } from '@/config'
  12. import TextGenerationImageUploader from '@/app/components/base/image-uploader/text-generation-image-uploader'
  13. import type { VisionFile, VisionSettings } from '@/types/app'
  14. export type IRunOnceProps = {
  15. siteInfo: SiteInfo
  16. promptConfig: PromptConfig
  17. inputs: Record<string, any>
  18. onInputsChange: (inputs: Record<string, any>) => void
  19. onSend: () => void
  20. visionConfig: VisionSettings
  21. onVisionFilesChange: (files: VisionFile[]) => void
  22. }
  23. const RunOnce: FC<IRunOnceProps> = ({
  24. promptConfig,
  25. inputs,
  26. onInputsChange,
  27. onSend,
  28. visionConfig,
  29. onVisionFilesChange,
  30. }) => {
  31. const { t } = useTranslation()
  32. const onClear = () => {
  33. const newInputs: Record<string, any> = {}
  34. promptConfig.prompt_variables.forEach((item) => {
  35. newInputs[item.key] = ''
  36. })
  37. onInputsChange(newInputs)
  38. }
  39. const onSubmit = (e: FormEvent<HTMLFormElement>) => {
  40. e.preventDefault()
  41. onSend()
  42. }
  43. return (
  44. <div className="">
  45. <section>
  46. {/* input form */}
  47. <form onSubmit={onSubmit}>
  48. {promptConfig.prompt_variables.map(item => (
  49. <div className='w-full mt-4' key={item.key}>
  50. <label className='text-gray-900 text-sm font-medium'>{item.name}</label>
  51. <div className='mt-2'>
  52. {item.type === 'select' && (
  53. <Select
  54. className='w-full'
  55. defaultValue={inputs[item.key]}
  56. onSelect={(i) => { onInputsChange({ ...inputs, [item.key]: i.value }) }}
  57. items={(item.options || []).map(i => ({ name: i, value: i }))}
  58. allowSearch={false}
  59. bgClassName='bg-gray-50'
  60. />
  61. )}
  62. {item.type === 'string' && (
  63. <input
  64. type="text"
  65. className="block w-full p-2 text-gray-900 border border-gray-300 rounded-lg bg-gray-50 sm:text-xs focus:ring-blue-500 focus:border-blue-500 "
  66. placeholder={`${item.name}${!item.required ? `(${t('appDebug.variableTable.optional')})` : ''}`}
  67. value={inputs[item.key]}
  68. onChange={(e) => { onInputsChange({ ...inputs, [item.key]: e.target.value }) }}
  69. maxLength={item.max_length || DEFAULT_VALUE_MAX_LEN}
  70. />
  71. )}
  72. {item.type === 'paragraph' && (
  73. <textarea
  74. className="block w-full h-[104px] p-2 text-gray-900 border border-gray-300 rounded-lg bg-gray-50 sm:text-xs focus:ring-blue-500 focus:border-blue-500 "
  75. placeholder={`${item.name}${!item.required ? `(${t('appDebug.variableTable.optional')})` : ''}`}
  76. value={inputs[item.key]}
  77. onChange={(e) => { onInputsChange({ ...inputs, [item.key]: e.target.value }) }}
  78. />
  79. )}
  80. {item.type === 'number' && (
  81. <input
  82. type="number"
  83. className="block w-full p-2 text-gray-900 border border-gray-300 rounded-lg bg-gray-50 sm:text-xs focus:ring-blue-500 focus:border-blue-500 "
  84. placeholder={`${item.name}${!item.required ? `(${t('appDebug.variableTable.optional')})` : ''}`}
  85. value={inputs[item.key]}
  86. onChange={(e) => { onInputsChange({ ...inputs, [item.key]: e.target.value }) }}
  87. />
  88. )}
  89. </div>
  90. </div>
  91. ))}
  92. {
  93. visionConfig?.enabled && (
  94. <div className="w-full mt-4">
  95. <div className="text-gray-900 text-sm font-medium">{t('common.imageUploader.imageUpload')}</div>
  96. <div className='mt-2'>
  97. <TextGenerationImageUploader
  98. settings={visionConfig}
  99. onFilesChange={files => onVisionFilesChange(files.filter(file => file.progress !== -1).map(fileItem => ({
  100. type: 'image',
  101. transfer_method: fileItem.type,
  102. url: fileItem.url,
  103. upload_file_id: fileItem.fileId,
  104. })))}
  105. />
  106. </div>
  107. </div>
  108. )
  109. }
  110. {promptConfig.prompt_variables.length > 0 && (
  111. <div className='mt-4 h-[1px] bg-gray-100'></div>
  112. )}
  113. <div className='w-full mt-4'>
  114. <div className="flex items-center justify-between">
  115. <Button
  116. onClick={onClear}
  117. disabled={false}
  118. >
  119. <span className='text-[13px]'>{t('common.operation.clear')}</span>
  120. </Button>
  121. <Button
  122. type='submit'
  123. variant="primary"
  124. disabled={false}
  125. >
  126. <PlayIcon className="shrink-0 w-4 h-4 mr-1" aria-hidden="true" />
  127. <span className='text-[13px]'>{t('share.generation.run')}</span>
  128. </Button>
  129. </div>
  130. </div>
  131. </form>
  132. </section>
  133. </div>
  134. )
  135. }
  136. export default React.memo(RunOnce)