90 lines
2.2 KiB
TypeScript
90 lines
2.2 KiB
TypeScript
import React from 'react'
|
|
import type { UseChatHelpers } from 'ai/react'
|
|
|
|
import { cn } from '@/lib/utils'
|
|
import { Button } from '@/components/ui/button'
|
|
import { IconRefresh, IconStop } from '@/components/ui/icons'
|
|
import { ButtonScrollToBottom } from '@/components/button-scroll-to-bottom'
|
|
import { FooterText } from '@/components/footer'
|
|
import { PromptForm, PromptFormRef } from '@/components/prompt-form'
|
|
|
|
export interface ChatPanelProps
|
|
extends Pick<
|
|
UseChatHelpers,
|
|
| 'append'
|
|
| 'isLoading'
|
|
| 'reload'
|
|
| 'messages'
|
|
| 'stop'
|
|
| 'input'
|
|
| 'setInput'
|
|
> {
|
|
id?: string
|
|
className?: string
|
|
onSubmit: (content: string) => Promise<void>
|
|
}
|
|
|
|
export function ChatPanel({
|
|
id,
|
|
isLoading,
|
|
stop,
|
|
append,
|
|
reload,
|
|
input,
|
|
setInput,
|
|
messages,
|
|
className,
|
|
onSubmit
|
|
}: ChatPanelProps) {
|
|
const promptFormRef = React.useRef<PromptFormRef>(null)
|
|
React.useEffect(() => {
|
|
promptFormRef?.current?.focus()
|
|
}, [id])
|
|
|
|
return (
|
|
<div
|
|
className={cn(
|
|
'bg-gradient-to-b from-muted/10 from-10% to-muted/30 to-50%',
|
|
className
|
|
)}
|
|
>
|
|
<ButtonScrollToBottom />
|
|
<div className="mx-auto sm:max-w-2xl sm:px-4">
|
|
<div className="flex h-10 items-center justify-center">
|
|
{isLoading ? (
|
|
<Button
|
|
variant="outline"
|
|
onClick={() => stop()}
|
|
className="bg-background"
|
|
>
|
|
<IconStop className="mr-2" />
|
|
Stop generating
|
|
</Button>
|
|
) : (
|
|
messages?.length > 0 && (
|
|
<Button
|
|
variant="outline"
|
|
onClick={() => reload()}
|
|
className="bg-background"
|
|
>
|
|
<IconRefresh className="mr-2" />
|
|
Regenerate response
|
|
</Button>
|
|
)
|
|
)}
|
|
</div>
|
|
<div className="space-y-4 border-t bg-background px-4 py-2 shadow-lg sm:rounded-t-xl sm:border md:py-4">
|
|
<PromptForm
|
|
ref={promptFormRef}
|
|
onSubmit={onSubmit}
|
|
input={input}
|
|
setInput={setInput}
|
|
isLoading={isLoading}
|
|
/>
|
|
<FooterText className="hidden sm:block" />
|
|
</div>
|
|
</div>
|
|
</div>
|
|
)
|
|
}
|