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

catch prompt assembly error

Browse Source
This commit is contained in:
Abi Raja 2023-11-30 16:04:03 -05:00
parent f51e70d701
commit 414d83c458
1 changed files with 16 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
backend/main.py
View File

@ -137,12 +137,23 @@ async def stream_code(websocket: WebSocket):
async def process_chunk(content): async def process_chunk(content):
await websocket.send_json({"type": "chunk", "value": content}) await websocket.send_json({"type": "chunk", "value": content})
if params.get("resultImage") and params["resultImage"]: # Assemble the prompt
prompt_messages = assemble_prompt( try:
params["image"], generated_code_config, params["resultImage"] if params.get("resultImage") and params["resultImage"]:
prompt_messages = assemble_prompt(
params["image"], generated_code_config, params["resultImage"]
)
else:
prompt_messages = assemble_prompt(params["image"], generated_code_config)
except:
await websocket.send_json(
{
"type": "error",
"value": "Error assembling prompt. Contact support at support@picoapps.xyz",
}
) )
else: await websocket.close()
prompt_messages = assemble_prompt(params["image"], generated_code_config) return
# Image cache for updates so that we don't have to regenerate images # Image cache for updates so that we don't have to regenerate images
image_cache = {} image_cache = {}