feat: update api to support image comparison

2023-11-21 21:14:45 +08:00 · 2023-11-21 21:14:45 +08:00 · 02c3178b6b
commit 02c3178b6b
parent 4290c70548
1 changed files with 4 additions and 1 deletions
--- a/backend/main.py
+++ b/backend/main.py
@ -96,6 +96,9 @@ async def stream_code_test(websocket: WebSocket):
    async def process_chunk(content):
        await websocket.send_json({"type": "chunk", "value": content})

+    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