Github/screenshot-to-code
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: update do generate code

Browse Source
This commit is contained in:
clean99 2023-11-26 14:51:39 +08:00
parent 3ae5cbfe19
commit ac47e9d622
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frontend/src/App.tsx
View File

@ -96,7 +96,7 @@ function App() {
setAppState(AppState.CODE_READY);
};
function doGenerateCode(params: CodeGenerationParams) {
function doGenerateCode(params: CodeGenerationParams, setCode: (value: React.SetStateAction<string>) => void = setGeneratedCode) {
setExecutionConsole([]);
setAppState(AppState.CODING);
@ -106,8 +106,8 @@ function App() {
generateCode(
wsRef,
updatedParams,
(token) => setGeneratedCode((prev) => prev + token),
(code) => setGeneratedCode(code),
(token) => setCode((prev) => prev + token),
(code) => setCode(code),
(line) => setExecutionConsole((prev) => [...prev, line]),
() => setAppState(AppState.CODE_READY)
);

4
frontend/src/components/ReactCodeEditor.tsx
View File

@ -19,7 +19,7 @@ import { Textarea } from "./ui/textarea";
interface IProps {
doGenerateCode: (params: CodeGenerationParams) => void;
doGenerateCode: (params: CodeGenerationParams, setCode: (value: React.SetStateAction<string>) => void) => void;
referenceImage: string;
}
@ -44,7 +44,7 @@ export const ReactCodeEditor: React.FC<IProps> = ({ doGenerateCode, referenceIma
codeType: CodeType.REACT,
image: referenceImage,
history: updatedHistory,
});
}, setGeneratedReactCode);
setReactHistory(updatedHistory);
setGeneratedReactCode("");

2
frontend/src/generateCode.ts
View File

@ -8,7 +8,7 @@ const ERROR_MESSAGE =
const STOP_MESSAGE = "Code generation stopped";
export enum CodeType {
HTML = "HTML",
HTML = "html",
REACT = "react",
}