import { Form, Modal, Select } from 'antd';
import { useTranslation } from 'react-i18next';
import { useDispatch, useSelector } from 'umi';

type FieldType = {
  embd_id?: string;
  img2txt_id?: string;
  llm_id?: string;
  asr_id?: string;
};

const SsModal = () => {
  const dispatch = useDispatch();
  const settingModel = useSelector((state: any) => state.settingModel);
  const { isShowSSModal, llmInfo = {}, tenantIfo } = settingModel;
  const [form] = Form.useForm();
  const { t } = useTranslation();

  const handleCancel = () => {
    dispatch({
      type: 'settingModel/updateState',
      payload: {
        isShowSSModal: false,
      },
    });
  };

  const handleOk = async () => {
    try {
      const values = await form.validateFields();
      const retcode = await dispatch<any>({
        type: 'settingModel/set_tenant_info',
        payload: {
          ...values,
          tenant_id: tenantIfo.tenant_id,
        },
      });
      retcode === 0 &&
        dispatch({
          type: 'settingModel/updateState',
          payload: {
            isShowSSModal: false,
          },
        });
    } catch (errorInfo) {
      console.log('Failed:', errorInfo);
    }
  };

  const handleChange = () => {};

  return (
    <Modal
      title="Basic Modal"
      open={isShowSSModal}
      onOk={handleOk}
      onCancel={handleCancel}
    >
      <Form
        form={form}
        name="validateOnly"
        // labelCol={{ span: 8 }}
        // wrapperCol={{ span: 16 }}
        style={{ maxWidth: 600 }}
        autoComplete="off"
        layout="vertical"
      >
        <Form.Item<FieldType>
          label="embedding 模型"
          name="embd_id"
          rules={[{ required: true, message: 'Please input value' }]}
          initialValue={tenantIfo.embd_id}
        >
          <Select
            // style={{ width: 200 }}
            onChange={handleChange}
            // fieldNames={label:}
            options={Object.keys(llmInfo).map((t) => {
              const options = llmInfo[t]
                .filter((d: any) => d.model_type === 'embedding')
                .map((d: any) => ({ label: d.llm_name, value: d.llm_name }));
              return { label: t, options };
            })}
          />
        </Form.Item>
        <Form.Item<FieldType>
          label="chat 模型"
          name="llm_id"
          rules={[{ required: true, message: 'Please input value' }]}
          initialValue={tenantIfo.llm_id}
        >
          <Select
            // style={{ width: 200 }}
            onChange={handleChange}
            // fieldNames={label:}
            options={Object.keys(llmInfo).map((t) => {
              const options = llmInfo[t]
                .filter((d: any) => d.model_type === 'chat')
                .map((d: any) => ({ label: d.llm_name, value: d.llm_name }));
              return { label: t, options };
            })}
          />
        </Form.Item>
        <Form.Item<FieldType>
          label="image2text 模型"
          name="img2txt_id"
          rules={[{ required: true, message: 'Please input value' }]}
          initialValue={tenantIfo.img2txt_id}
        >
          <Select
            // style={{ width: 200 }}
            onChange={handleChange}
            // fieldNames={label:}
            options={Object.keys(llmInfo).map((t) => {
              const options = llmInfo[t]
                .filter((d: any) => d.model_type === 'image2text')
                .map((d: any) => ({ label: d.llm_name, value: d.llm_name }));
              return { label: t, options };
            })}
          />
        </Form.Item>
        <Form.Item<FieldType>
          label="speech2text 模型"
          name="asr_id"
          rules={[{ required: true, message: 'Please input value' }]}
          initialValue={tenantIfo.asr_id}
        >
          <Select
            // style={{ width: 200 }}
            onChange={handleChange}
            // fieldNames={label:}
            options={Object.keys(llmInfo).map((t) => {
              const options = llmInfo[t]
                .filter((d: any) => d.model_type === 'speech2text')
                .map((d: any) => ({ label: d.llm_name, value: d.llm_name }));
              return { label: t, options };
            })}
          />
        </Form.Item>
      </Form>
    </Modal>
  );
};
export default SsModal;