also use appHistory

This commit is contained in:
Abi Raja 2024-07-25 12:37:06 -04:00
parent 83f6f00b10
commit dcef298dba
2 changed files with 14 additions and 6 deletions

View File

@ -48,16 +48,21 @@ function App() {
const [appState, setAppState] = useState<AppState>(AppState.INITIAL); const [appState, setAppState] = useState<AppState>(AppState.INITIAL);
const { const {
// Inputs
inputMode, inputMode,
setInputMode, setInputMode,
isImportedFromCode, isImportedFromCode,
setIsImportedFromCode, setIsImportedFromCode,
referenceImages, referenceImages,
setReferenceImages, setReferenceImages,
// Outputs
generatedCode, generatedCode,
setGeneratedCode, setGeneratedCode,
currentVersion, currentVersion,
setCurrentVersion, setCurrentVersion,
appHistory,
setAppHistory,
} = useProjectStore(); } = useProjectStore();
const [executionConsole, setExecutionConsole] = useState<string[]>([]); const [executionConsole, setExecutionConsole] = useState<string[]>([]);
@ -88,9 +93,6 @@ function App() {
const selectedCodeGenerationModel = const selectedCodeGenerationModel =
settings.codeGenerationModel || CodeGenerationModel.GPT_4_VISION; settings.codeGenerationModel || CodeGenerationModel.GPT_4_VISION;
// App history
const [appHistory, setAppHistory] = useState<History>([]);
const [shouldIncludeResultImage, setShouldIncludeResultImage] = const [shouldIncludeResultImage, setShouldIncludeResultImage] =
useState<boolean>(false); useState<boolean>(false);

View File

@ -22,8 +22,10 @@ interface ProjectStore {
currentVersion: number | null; currentVersion: number | null;
setCurrentVersion: (version: number | null) => void; setCurrentVersion: (version: number | null) => void;
appHistory: History[]; appHistory: History;
setAppHistory: (history: History[]) => void; setAppHistory: (
updater: History | ((currentHistory: History) => History)
) => void;
} }
export const useProjectStore = create<ProjectStore>((set) => ({ export const useProjectStore = create<ProjectStore>((set) => ({
@ -45,5 +47,9 @@ export const useProjectStore = create<ProjectStore>((set) => ({
currentVersion: null, currentVersion: null,
setCurrentVersion: (version) => set({ currentVersion: version }), setCurrentVersion: (version) => set({ currentVersion: version }),
appHistory: [], appHistory: [],
setAppHistory: (history) => set({ appHistory: history }), setAppHistory: (updater) =>
set((state) => ({
appHistory:
typeof updater === "function" ? updater(state.appHistory) : updater,
})),
})); }));