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

Update generate_code.py

Browse Source
This commit is contained in:
Piyush Mishra 2023-12-12 10:04:28 +05:30
parent 7eab735af4
commit 8ff1bf1976
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/routes/generate_code.py
View File

@ -174,10 +174,12 @@ async def stream_code(websocket: WebSocket):
try: try:
if params.get("resultImage") and params["resultImage"]: if params.get("resultImage") and params["resultImage"]:
prompt_messages = assemble_prompt( prompt_messages = assemble_prompt(
params["image"], valid_stack, params["resultImage"] params["images"], generated_code_config, params["resultImage"]
) )
else: else:
prompt_messages = assemble_prompt(params["image"], valid_stack) prompt_messages = assemble_prompt(
params["images"], generated_code_config
)
except: except:
await websocket.send_json( await websocket.send_json(
{ {