|
1 | | -import { FormProvider, useForm } from "react-hook-form"; |
| 1 | +import { FormProvider } from "react-hook-form"; |
2 | 2 | import Input from "../../components/internal-ui/Input"; |
3 | | -import { BasicGenFields, ENDPOINTS, UserType } from "../constants"; |
| 3 | +import { basicGenDefaultValues, BasicGenFields, ENDPOINTS } from "../constants"; |
4 | 4 | import { Button, HStack, Stack } from "@chakra-ui/react"; |
5 | 5 |
|
6 | 6 | import useGptStore from "../../utils/store"; |
7 | 7 | import ChatBox from "../../components/internal-ui/ChatBox"; |
8 | 8 |
|
9 | 9 | import { ApiRequestBasicGen, BasicGenFormData } from "../model"; |
10 | | -import { useEffect, useState } from "react"; |
| 10 | + |
11 | 11 | import { IoSendOutline } from "react-icons/io5"; |
12 | | -import { v4 as uuid } from "uuid"; |
| 12 | + |
| 13 | +import useFormHandler from "../../hooks/useFormHandler"; |
| 14 | +import { basicGenMapper } from "../../utils/mapperFunctions"; |
13 | 15 |
|
14 | 16 | const BasicGen = () => { |
15 | | - const formMethods = useForm<BasicGenFormData>({ |
16 | | - defaultValues: { |
17 | | - minToken: 100, |
18 | | - maxToken: 500, |
19 | | - service: "terraform", |
20 | | - input: undefined, |
21 | | - }, |
22 | | - mode: "onSubmit", |
23 | | - }); |
| 17 | + const { formMethods, handleSubmit, onSubmit, request } = useFormHandler< |
| 18 | + BasicGenFormData, |
| 19 | + ApiRequestBasicGen |
| 20 | + >(basicGenDefaultValues); |
24 | 21 |
|
25 | | - const { handleSubmit } = formMethods; |
| 22 | + const handleFormSubmit = handleSubmit((data) => { |
| 23 | + onSubmit(basicGenMapper(data), data.input); |
| 24 | + }); |
26 | 25 |
|
27 | 26 | const messages = useGptStore((s) => s.messages); |
28 | | - const addMessage = useGptStore((s) => s.addMessage); |
29 | | - const [req, setReq] = useState<ApiRequestBasicGen | null>(null); |
30 | | - |
31 | | - const onSubmit = (data: BasicGenFormData) => { |
32 | | - addMessage(UserType.USER, data.input, uuid()); |
33 | | - const request: ApiRequestBasicGen = { |
34 | | - min_token: data.minToken, |
35 | | - max_token: data.maxToken, |
36 | | - service: data.service, |
37 | | - input: data.input, |
38 | | - requestId: uuid(), |
39 | | - }; |
40 | | - if (data.input) setReq(request); |
41 | | - }; |
42 | | - |
43 | | - useEffect(() => { |
44 | | - return () => setReq(null); |
45 | | - }, []); |
46 | 27 |
|
47 | 28 | return ( |
48 | | - <div> |
49 | | - <FormProvider {...formMethods}> |
50 | | - <form onSubmit={(e) => void handleSubmit(onSubmit)(e)}> |
51 | | - <Stack gap="3" justifyContent="center" alignItems="center"> |
52 | | - <div className="flex gap-2"> |
53 | | - <Input fieldName={BasicGenFields.MIN_TOKEN} label="Min token" /> |
54 | | - <Input fieldName={BasicGenFields.MAX_TOKEN} label="Max token" /> |
55 | | - <Input fieldName={BasicGenFields.SERVICE} label="Service" /> |
56 | | - </div> |
57 | | - <ChatBox |
58 | | - endpoint={ENDPOINTS.postBasic} |
59 | | - request={req} |
60 | | - messageData={messages} |
61 | | - id={req?.requestId ?? ""} |
62 | | - /> |
63 | | - <HStack |
64 | | - mt="3" |
65 | | - alignItems="center" |
66 | | - alignContent="center" |
67 | | - justifyContent="center" |
68 | | - bottom="5" |
| 29 | + <FormProvider {...formMethods}> |
| 30 | + <form onSubmit={handleFormSubmit}> |
| 31 | + <Stack gap="3" justifyContent="center" alignItems="center"> |
| 32 | + <div className="flex gap-2"> |
| 33 | + <Input fieldName={BasicGenFields.MIN_TOKEN} label="Min token" /> |
| 34 | + <Input fieldName={BasicGenFields.MAX_TOKEN} label="Max token" /> |
| 35 | + <Input fieldName={BasicGenFields.SERVICE} label="Service" /> |
| 36 | + </div> |
| 37 | + <ChatBox |
| 38 | + endpoint={ENDPOINTS.postBasic} |
| 39 | + request={request} |
| 40 | + messageData={messages} |
| 41 | + id={request?.requestId ?? ""} |
| 42 | + /> |
| 43 | + <HStack |
| 44 | + mt="3" |
| 45 | + alignItems="center" |
| 46 | + alignContent="center" |
| 47 | + justifyContent="center" |
| 48 | + bottom="5" |
| 49 | + > |
| 50 | + <Input placeholder="Text" fieldName={BasicGenFields.INPUT} /> |
| 51 | + <Button |
| 52 | + type="submit" |
| 53 | + bg="orange.800" |
| 54 | + disabled={formMethods.getFieldState(BasicGenFields.INPUT).invalid} |
69 | 55 | > |
70 | | - <Input placeholder="Text" fieldName={BasicGenFields.INPUT} /> |
71 | | - <Button |
72 | | - type="submit" |
73 | | - bg="orange.800" |
74 | | - disabled={ |
75 | | - formMethods.getFieldState(BasicGenFields.INPUT).invalid |
76 | | - } |
77 | | - > |
78 | | - <IoSendOutline /> |
79 | | - </Button> |
80 | | - </HStack> |
81 | | - </Stack> |
82 | | - </form> |
83 | | - </FormProvider> |
84 | | - </div> |
| 56 | + <IoSendOutline /> |
| 57 | + </Button> |
| 58 | + </HStack> |
| 59 | + </Stack> |
| 60 | + </form> |
| 61 | + </FormProvider> |
85 | 62 | ); |
86 | 63 | }; |
87 | 64 |
|
|
0 commit comments