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

feat: update api to support image comparison

Browse Source
This commit is contained in:
clean99 2023-11-21 21:14:45 +08:00
parent 4290c70548
commit 02c3178b6b
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend/main.py
View File

@ -96,7 +96,10 @@ async def stream_code_test(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})
prompt_messages = assemble_prompt(params["image"]) if params.get("resultImg") and params["resultImg"]:
prompt_messages = assemble_prompt(params["image"], params["resultImg"])
else:
prompt_messages = assemble_prompt(params["image"])
# 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 = {}