import { createSlice } from '@reduxjs/toolkit';

const initialState = {
  isSubmitting: false,
  submission: {},
  stopStream: false,
  disabled: false,
  model: 'chatgpt',
  promptPrefix: '',
  chatGptLabel: '',
  customModel: null,
};

const currentSlice = createSlice({
  name: 'submit',
  initialState,
  reducers: {
    setSubmitState: (state, action) => {
      state.isSubmitting = action.payload;
    },
    setSubmission: (state, action) => {
      state.submission = action.payload;
      if (Object.keys(action.payload).length === 0) {
        state.isSubmitting = false;
      }
    },
    setStopStream: (state, action) => {
      state.stopStream = action.payload;
    },
    setDisabled: (state, action) => {
      state.disabled = action.payload;
    },
    setModel: (state, action) => {
      state.model = action.payload;
    },
    setCustomGpt: (state, action) => {
      state.promptPrefix = action.payload.promptPrefix;
      state.chatGptLabel = action.payload.chatGptLabel;
    },
    setCustomModel: (state, action) => {
      state.customModel = action.payload;
    }
  }
});

export const { setSubmitState, setSubmission, setStopStream, setDisabled, setModel, setCustomGpt, setCustomModel } =
  currentSlice.actions;

export default currentSlice.reducer;