chatgpt-web/src/views/Chat/index.vue

<script setup lang='ts'>
import { nextTick, onMounted, ref } from 'vue'
import { NButton, NInput, NPopover, useMessage } from 'naive-ui'
import { Message } from './components'
import { clearChatContext, fetchChatAPI } from './request'
import { Icon } from '@/components'

interface ListProps {
  dateTime: string
  message: string
  reversal?: boolean
}

const scrollRef = ref<HTMLDivElement>()

const ms = useMessage()

const value = ref('')

const loading = ref(false)

const list = ref<ListProps[]>([])

onMounted(initChat)

function initChat() {
  addMessage('Hi, I am ChatGPT, a chatbot based on GPT-3.', false)
}

async function handleClear() {
  try {
    const { message } = await clearChatContext()
    ms.success(message)
    list.value = []
    setTimeout(initChat, 100)
  }
  catch (error) {
    ms.error('Clear failed, please try again later.')
  }
}

function handleEnter(event: KeyboardEvent) {
  if (event.key === 'Enter')
    handleSubmit()
}

async function handleSubmit() {
  if (!value.value) {
    ms.warning('Please enter a message')
    return
  }

  addMessage(value.value, true)

  try {
    loading.value = true
    const { text } = await fetchChatAPI(value.value)
    value.value = ''
    addMessage(text, false)
  }
  catch (error: any) {
    addMessage(error.message ?? 'Request failed, please try again later.', false)
  }
  finally {
    loading.value = false
  }
}

function addMessage(message: string, reversal = false) {
  list.value.push({ dateTime: new Date().toLocaleString(), message, reversal })
  nextTick(() => scrollRef.value && (scrollRef.value.scrollTop = scrollRef.value.scrollHeight))
}
</script>

<template>
  <div class="flex flex-col h-full overflow-hidden border rounded-md shadow-md">
    <header class="flex items-center justify-between p-4">
      <h1 class="text-xl font-bold">
        ChatGPT Web
      </h1>
      <div>
        <NPopover>
          <template #trigger>
            <button
              class="w-[40px] h-[40px] rounded-full hover:bg-neutral-100 transition flex justify-center items-center"
              @click="handleClear"
            >
              <Icon icon="ri:delete-bin-6-line" />
            </button>
          </template>
          <span>Clear Context</span>
        </NPopover>
      </div>
    </header>
    <main class="flex-1 overflow-hidden border-y">
      <div ref="scrollRef" class="h-full p-4 overflow-hidden overflow-y-auto">
        <div>
          <Message
            v-for="(item, index) of list" :key="index" :date-time="item.dateTime" :message="item.message"
            :reversal="item.reversal"
          />
        </div>
      </div>
    </main>
    <footer class="p-4">
      <div class="flex items-center justify-between space-x-2">
        <NInput v-model:value="value" :disabled="loading" placeholder="Type a message..." @keyup="handleEnter" />
        <NButton type="primary" :loading="loading" @click="handleSubmit">
          <template #icon>
            <Icon icon="ri:send-plane-fill" />
          </template>
        </NButton>
      </div>
    </footer>
  </div>
</template>