257 lines
8.9 KiB
TypeScript
257 lines
8.9 KiB
TypeScript
import { useForm } from "@mantine/form"
|
|
import { useMutation, useQueryClient } from "@tanstack/react-query"
|
|
import React from "react"
|
|
import useDynamicTextareaSize from "~hooks/useDynamicTextareaSize"
|
|
import PhotoIcon from "@heroicons/react/24/outline/PhotoIcon"
|
|
import XMarkIcon from "@heroicons/react/24/outline/XMarkIcon"
|
|
import { toBase64 } from "~libs/to-base64"
|
|
import { useMessageOption } from "~hooks/useMessageOption"
|
|
import { Tooltip } from "antd"
|
|
import { MicIcon, StopCircleIcon } from "lucide-react"
|
|
import { Image } from "antd"
|
|
import { useSpeechRecognition } from "~hooks/useSpeechRecognition"
|
|
|
|
type Props = {
|
|
dropedFile: File | undefined
|
|
}
|
|
|
|
export const PlaygroundForm = ({ dropedFile }: Props) => {
|
|
const textareaRef = React.useRef<HTMLTextAreaElement>(null)
|
|
const inputRef = React.useRef<HTMLInputElement>(null)
|
|
|
|
const resetHeight = () => {
|
|
const textarea = textareaRef.current
|
|
if (textarea) {
|
|
textarea.style.height = "auto"
|
|
}
|
|
}
|
|
const form = useForm({
|
|
initialValues: {
|
|
message: "",
|
|
image: ""
|
|
}
|
|
})
|
|
|
|
React.useEffect(() => {
|
|
if (textareaRef.current) {
|
|
textareaRef.current.focus()
|
|
}
|
|
}, [])
|
|
|
|
const onInputChange = async (
|
|
e: React.ChangeEvent<HTMLInputElement> | File
|
|
) => {
|
|
if (e instanceof File) {
|
|
const base64 = await toBase64(e)
|
|
form.setFieldValue("image", base64)
|
|
} else {
|
|
if (e.target.files) {
|
|
const base64 = await toBase64(e.target.files[0])
|
|
form.setFieldValue("image", base64)
|
|
}
|
|
}
|
|
}
|
|
|
|
React.useEffect(() => {
|
|
if (dropedFile) {
|
|
onInputChange(dropedFile)
|
|
}
|
|
}, [dropedFile])
|
|
|
|
useDynamicTextareaSize(textareaRef, form.values.message, 300)
|
|
|
|
const {
|
|
onSubmit,
|
|
selectedModel,
|
|
chatMode,
|
|
speechToTextLanguage,
|
|
stopStreamingRequest
|
|
} = useMessageOption()
|
|
|
|
const { isListening, start, stop, transcript } = useSpeechRecognition()
|
|
|
|
React.useEffect(() => {
|
|
if (isListening) {
|
|
form.setFieldValue("message", transcript)
|
|
}
|
|
}, [transcript])
|
|
|
|
const queryClient = useQueryClient()
|
|
|
|
const { mutateAsync: sendMessage, isPending: isSending } = useMutation({
|
|
mutationFn: onSubmit,
|
|
onSuccess: () => {
|
|
queryClient.invalidateQueries({
|
|
queryKey: ["fetchChatHistory"]
|
|
})
|
|
}
|
|
})
|
|
|
|
return (
|
|
<div className="px-3 pt-3 md:px-6 md:pt-6 md:bg-white dark:bg-[#262626] border rounded-t-xl border-black/10 dark:border-gray-600">
|
|
<div
|
|
className={`h-full rounded-md shadow relative ${
|
|
form.values.image.length === 0 ? "hidden" : "block"
|
|
}`}>
|
|
<div className="relative">
|
|
<Image
|
|
src={form.values.image}
|
|
alt="Uploaded Image"
|
|
width={180}
|
|
preview={false}
|
|
className="rounded-md"
|
|
/>
|
|
<button
|
|
onClick={() => {
|
|
form.setFieldValue("image", "")
|
|
}}
|
|
className="flex items-center justify-center absolute top-0 m-2 bg-white dark:bg-[#262626] p-1 rounded-full hover:bg-gray-100 dark:hover:bg-gray-600 text-black dark:text-gray-100">
|
|
<XMarkIcon className="h-5 w-5" />
|
|
</button>
|
|
</div>
|
|
</div>
|
|
{/* <div className="flex gap-3 justify-end">
|
|
<Tooltip title="New Chat">
|
|
<button
|
|
onClick={clearChat}
|
|
className="text-gray-500 dark:text-gray-100 mr-3">
|
|
<ArrowPathIcon className="h-5 w-5" />
|
|
</button>
|
|
</Tooltip>
|
|
</div> */}
|
|
<div>
|
|
<div className="flex">
|
|
<form
|
|
onSubmit={form.onSubmit(async (value) => {
|
|
if (!selectedModel || selectedModel.length === 0) {
|
|
form.setFieldError("message", "Please select a model")
|
|
return
|
|
}
|
|
form.reset()
|
|
resetHeight()
|
|
await sendMessage({
|
|
image: value.image,
|
|
message: value.message.trim()
|
|
})
|
|
})}
|
|
className="shrink-0 flex-grow flex flex-col items-center ">
|
|
<input
|
|
id="file-upload"
|
|
name="file-upload"
|
|
type="file"
|
|
className="sr-only"
|
|
ref={inputRef}
|
|
accept="image/*"
|
|
multiple={false}
|
|
onChange={onInputChange}
|
|
/>
|
|
<div className="w-full border-x border-t flex flex-col dark:border-gray-600 rounded-t-xl p-2">
|
|
<textarea
|
|
onKeyDown={(e) => {
|
|
if (e.key === "Enter" && !e.shiftKey && !isSending) {
|
|
e.preventDefault()
|
|
form.onSubmit(async (value) => {
|
|
if (value.message.trim().length === 0) {
|
|
return
|
|
}
|
|
if (!selectedModel || selectedModel.length === 0) {
|
|
form.setFieldError("message", "Please select a model")
|
|
return
|
|
}
|
|
form.reset()
|
|
resetHeight()
|
|
await sendMessage({
|
|
image: value.image,
|
|
message: value.message.trim()
|
|
})
|
|
})()
|
|
}
|
|
}}
|
|
ref={textareaRef}
|
|
className="px-2 py-2 w-full resize-none bg-transparent focus-within:outline-none sm:text-sm focus:ring-0 focus-visible:ring-0 ring-0 dark:ring-0 border-0 dark:text-gray-100"
|
|
required
|
|
rows={1}
|
|
style={{ minHeight: "60px" }}
|
|
tabIndex={0}
|
|
placeholder="Type a message..."
|
|
{...form.getInputProps("message")}
|
|
/>
|
|
<div className="flex mt-4 justify-end gap-3">
|
|
<Tooltip title="Voice Message">
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
if (isListening) {
|
|
stop()
|
|
} else {
|
|
start({
|
|
lang: speechToTextLanguage,
|
|
continuous: true
|
|
})
|
|
}
|
|
}}
|
|
className={`flex items-center justify-center dark:text-gray-300`}>
|
|
{!isListening ? (
|
|
<MicIcon className="h-5 w-5" />
|
|
) : (
|
|
<div className="relative">
|
|
<span className="animate-ping absolute inline-flex h-3 w-3 rounded-full bg-red-400 opacity-75"></span>
|
|
<MicIcon className="h-5 w-5" />
|
|
</div>
|
|
)}
|
|
</button>
|
|
</Tooltip>
|
|
<Tooltip title="Upload Image">
|
|
<button
|
|
type="button"
|
|
onClick={() => {
|
|
inputRef.current?.click()
|
|
}}
|
|
className={`flex items-center justify-center dark:text-gray-300 ${
|
|
chatMode === "rag" ? "hidden" : "block"
|
|
}`}>
|
|
<PhotoIcon className="h-5 w-5" />
|
|
</button>
|
|
</Tooltip>
|
|
{!isSending ? (
|
|
<button
|
|
disabled={isSending || form.values.message.length === 0}
|
|
className="inline-flex items-center rounded-md border border-transparent bg-black px-2 py-2 text-sm font-medium leading-4 text-white shadow-sm hover:bg-gray-800 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2 dark:bg-white dark:text-gray-800 dark:hover:bg-gray-100 dark:focus:ring-gray-500 dark:focus:ring-offset-gray-100 disabled:opacity-50 ">
|
|
<svg
|
|
xmlns="http://www.w3.org/2000/svg"
|
|
fill="none"
|
|
stroke="currentColor"
|
|
strokeLinecap="round"
|
|
strokeLinejoin="round"
|
|
strokeWidth="2"
|
|
className="h-4 w-4 mr-2"
|
|
viewBox="0 0 24 24">
|
|
<path d="M9 10L4 15 9 20"></path>
|
|
<path d="M20 4v7a4 4 0 01-4 4H4"></path>
|
|
</svg>
|
|
Send
|
|
</button>
|
|
) : (
|
|
<Tooltip title="Stop Streaming">
|
|
<button
|
|
type="button"
|
|
onClick={stopStreamingRequest}
|
|
className="text-gray-800 dark:text-gray-300">
|
|
<StopCircleIcon className="h-6 w-6" />
|
|
</button>
|
|
</Tooltip>
|
|
)}
|
|
</div>
|
|
</div>
|
|
</form>
|
|
</div>
|
|
{form.errors.message && (
|
|
<div className="text-red-500 text-center text-sm mt-1">
|
|
{form.errors.message}
|
|
</div>
|
|
)}
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|