feat: playground

This commit is contained in:
jialin
2024-05-15 19:36:17 +08:00
parent 72a53424fa
commit fd9f3d8bd9
19 changed files with 380 additions and 20 deletions
+51 -8
View File
@@ -1,16 +1,59 @@
import { useState,useEffect} from 'react';
import { Button, Select,Space,Input,Popover } from 'antd';
import { ControlOutlined } from '@ant-design/icons';
import { PageContainer } from '@ant-design/pro-components';
import SealInput from '@/components/seal-form/seal-input';
import MessageInput from './components/messageInput';
import ChatContent from './components/chatContent';
const dataList = [
{value: 'llama3:latest', label: 'llama3:latest'},
{value: 'wangfuyun/AnimateLCM', label: 'wangfuyun/AnimateLCM'},
{value: 'Revanthraja/Text_to_Vision', label: 'Revanthraja/Text_to_Vision'},
]
const Playground: React.FC = () => {
const [ModelList, setModelList] = useState(dataList)
const [messageList, setMessageList] = useState<any[]>([])
const [selectedModel, setSelectedModel] = useState('llama3:latest')
const handleSelectChange = (value: string) => {
setSelectedModel(value)
}
const getMessageList = () => {
// fetch message list from server
setMessageList(['1'])
}
useEffect(() => {
getMessageList()
}, [selectedModel])
const Dashboard: React.FC = () => {
return (
<PageContainer
ghost
header={{
title: 'Dashboard',
}}
>
<div>Playground</div>
title={false}
extra={[
<Space size={20}>
<Select showSearch options={ModelList} style={{width: 300}} value={selectedModel} onChange={handleSelectChange} variant='filled'></Select>
<Popover content="ssd" title="Title" trigger="click">
<Button type="primary" shape="circle" icon={<ControlOutlined />}>
</Button>
</Popover>
</Space>
]}
footer={[
<MessageInput/>
]}
>
<div style={{display: 'flex', justifyContent: 'center'}}>
<ChatContent messageList={messageList}></ChatContent>
</div>
<SealInput.Input label="test label"></SealInput.Input>
</PageContainer>
);
};
export default Dashboard;
export default Playground;