Update core/refiner/simple_refiner.py

core/refiner/simple_refiner.py

@@ -13,11 +13,12 @@ class SimpleRefiner(BaseRefiner):
             memory = []
         else:
             memory = memory.messages[1:]
+        
+        user_context = [{"role": "user", "content": [
+                        {"type": "text", "text": f"{message}"},]}]
         if image_paths:
             if not isinstance(image_paths, list):
                 image_paths = [image_paths]
-            user_context = [{"role": "user", "content": [
-                            {"type": "text", "text": f"{message}"},]}]
             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": [