Update generate_code.py

PiyushMishra318 · PiyushMishra318 · commit 7883e25e1dc1 · 2023-12-12T10:04:28.000+05:30
diff --git a/backend/routes/generate_code.py b/backend/routes/generate_code.py
@@ -150,11 +150,11 @@ async def process_chunk(content: str):
         try:
             if params.get("resultImage") and params["resultImage"]:
                 prompt_messages = assemble_prompt(
-                    params["image"], generated_code_config, params["resultImage"]
+                    params["images"], generated_code_config, params["resultImage"]
                 )
             else:
                 prompt_messages = assemble_prompt(
-                    params["image"], generated_code_config
+                    params["images"], generated_code_config
                 )
         except:
             await websocket.send_json(

Original file line number	Diff line number	Diff line change
`@@ -150,11 +150,11 @@ async def process_chunk(content: str):`
`150`	`150`	`try:`
`151`	`151`	`if params.get("resultImage") and params["resultImage"]:`
`152`	`152`	`prompt_messages = assemble_prompt(`
`153`		`- params["image"], generated_code_config, params["resultImage"]`
	`153`	`+ params["images"], generated_code_config, params["resultImage"]`
`154`	`154`	`)`
`155`	`155`	`else:`
`156`	`156`	`prompt_messages = assemble_prompt(`
`157`		`- params["image"], generated_code_config`
	`157`	`+ params["images"], generated_code_config`
`158`	`158`	`)`
`159`	`159`	`except:`
`160`	`160`	`await websocket.send_json(`