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

support text to code

Browse Source
This commit is contained in:
Abi Raja 2024-09-10 14:46:16 +02:00
parent 86f6781682
commit 6600035267
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
backend/prompts/__init__.py
View File

@ -43,12 +43,17 @@ async def create_prompt(
prompt_messages.append(message) prompt_messages.append(message)
else: else:
# Assemble the prompt for non-imported code # Assemble the prompt for non-imported code
if params.get("resultImage"): if input_mode == "image":
prompt_messages = assemble_prompt( if params.get("resultImage"):
params["image"], stack, params["resultImage"] prompt_messages = assemble_prompt(
) params["image"], stack, params["resultImage"]
)
else:
prompt_messages = assemble_prompt(params["image"], stack)
elif input_mode == "text":
prompt_messages = assemble_text_prompt(params["image"], stack)
else: else:
prompt_messages = assemble_prompt(params["image"], stack) raise Exception("Invalid input mode")
if params["generationType"] == "update": if params["generationType"] == "update":
# Transform the history tree into message format # Transform the history tree into message format

2
frontend/src/App.tsx
View File

@ -343,7 +343,7 @@ function App({ navbarComponent }: Props) {
doGenerateCode({ doGenerateCode({
generationType: "update", generationType: "update",
inputMode, inputMode,
image: referenceImages[0], image: inputMode === "text" ? initialPrompt : referenceImages[0],
resultImage, resultImage,
history: updatedHistory, history: updatedHistory,
isImportedFromCode, isImportedFromCode,