Update core/refiner/simple_refiner.py

core/refiner/simple_refiner.py

@@ -42,12 +42,9 @@ class SimpleRefiner(BaseRefiner):
             if not isinstance(image_paths, list):
                 image_paths = [image_paths]
             for image_path in image_paths:
-                user_context.append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image_path.name)}"}})
-            context = [{"role": "system", "content": self.sys_prompt}] + memory + [{"role": "user", "content": [
-                            {"type": "text", "text": f"{message}"},
-                        ]}]
-        else:
-            context = [{"role": "system", "content": self.sys_prompt}] + memory + user_context
+                user_context[0]["content"].append({"type": "image_url", "image_url": {"url": f"data:image/jpeg;base64,{encode_image(image_path.name)}"}})
+        context = [{"role": "system", "content": self.sys_prompt}] + memory + user_context
+        
         response = self.model.respond(context)
 
         return response