Update PCAgent/chat.py

PCAgent/chat.py

@@ -53,7 +53,7 @@ def add_response_old(role, prompt, chat_history, image=None):
             },
             {
                 "type": "image", 
-                "image": f"file://{image}"
+                "image": image
             },
         ]
     else:
@@ -76,23 +76,13 @@ def add_response(role, prompt, chat_history, image=[], use_qwen=False):
         },
     ]
     for i in range(len(image)):
-        if not use_qwen:
-            base64_image = resize_encode_image(image[i])
-            content.append(
-                {
-                    "type": "image_url", 
-                    "image_url": {
-                        "url": f"data:image/png;base64,{base64_image}"
-                    }
-                }
-            )
-        else:
-            content.append(
-                {
-                    "type": "image", 
-                    "image": image[i]
-                }
-            )
+        base64_image = resize_encode_image(image[i])
+        content.append(
+            {
+                "type": "image_", 
+                "image": image[i]
+            }
+        )
     new_chat_history.append([role, content])
     return new_chat_history
 
@@ -108,17 +98,13 @@ def add_response_two_image(role, prompt, chat_history, image):
             "text": prompt
         },
         {
-            "type": "image_url", 
-            "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_image1}"
-            }
+            "type": "image", 
+            "image": image[0]
         },
         {
-            "type": "image_url", 
-            "image_url": {
-                "url": f"data:image/jpeg;base64,{base64_image2}"
-            }
-        },
+            "type": "image", 
+            "image": image[1]
+        }
     ]
 
     new_chat_history.append([role, content])