feat: update prompt to support image comparison

backend/prompts.py

@@ -2,7 +2,7 @@ SYSTEM_PROMPT = """
 You are an expert Tailwind developer
 You take screenshots of a reference web page from the user, and then build single page apps 
 using Tailwind, HTML and JS.
-You might also be given a screenshot of a web page that you have already built, and asked to
+You might also be given a screenshot(The second image) of a web page that you have already built, and asked to
 update it to look more like the reference image.
 
 - Make sure the app looks exactly like the screenshot.
@@ -27,13 +27,8 @@ USER_PROMPT = """
 Generate code for a web page that looks exactly like this.
 """
 
-
-def assemble_prompt(image_data_url):
-    return [
-        {"role": "system", "content": SYSTEM_PROMPT},
-        {
-            "role": "user",
-            "content": [
+def assemble_prompt(image_data_url, result_image_data_url=None):
+    content = [
         {
             "type": "image_url",
             "image_url": {"url": image_data_url, "detail": "high"},
@@ -42,6 +37,16 @@ def assemble_prompt(image_data_url):
             "type": "text",
             "text": USER_PROMPT,
         },
-            ],
+    ]
+    if result_image_data_url:
+        content.insert(1, {
+            "type": "image_url",
+            "image_url": {"url": result_image_data_url, "detail": "high"},
+        })
+    return [
+        {"role": "system", "content": SYSTEM_PROMPT},
+        {
+            "role": "user",
+            "content": content,
         },
     ]