junyangwang0410 commited on
Commit
375bf58
·
verified ·
1 Parent(s): 181a05c

Update PCAgent/chat.py

Browse files
Files changed (1) hide show
  1. PCAgent/chat.py +7 -7
PCAgent/chat.py CHANGED
@@ -53,7 +53,7 @@ def add_response_old(role, prompt, chat_history, image=None):
53
  },
54
  {
55
  "type": "image",
56
- "image": image
57
  },
58
  ]
59
  else:
@@ -76,11 +76,11 @@ def add_response(role, prompt, chat_history, image=[], use_qwen=False):
76
  },
77
  ]
78
  for i in range(len(image)):
79
- base64_image = resize_encode_image(image[i])
80
  content.append(
81
  {
82
  "type": "image_",
83
- "image": image[i]
84
  }
85
  )
86
  new_chat_history.append([role, content])
@@ -90,8 +90,8 @@ def add_response(role, prompt, chat_history, image=[], use_qwen=False):
90
  def add_response_two_image(role, prompt, chat_history, image):
91
  new_chat_history = copy.deepcopy(chat_history)
92
 
93
- base64_image1 = resize_encode_image(image[0])
94
- base64_image2 = resize_encode_image(image[1])
95
  content = [
96
  {
97
  "type": "text",
@@ -99,11 +99,11 @@ def add_response_two_image(role, prompt, chat_history, image):
99
  },
100
  {
101
  "type": "image",
102
- "image": image[0]
103
  },
104
  {
105
  "type": "image",
106
- "image": image[1]
107
  }
108
  ]
109
 
 
53
  },
54
  {
55
  "type": "image",
56
+ "image": f"file://{image}"
57
  },
58
  ]
59
  else:
 
76
  },
77
  ]
78
  for i in range(len(image)):
79
+ # base64_image = resize_encode_image(image[i])
80
  content.append(
81
  {
82
  "type": "image_",
83
+ "image": f"file://{image[i]}"
84
  }
85
  )
86
  new_chat_history.append([role, content])
 
90
  def add_response_two_image(role, prompt, chat_history, image):
91
  new_chat_history = copy.deepcopy(chat_history)
92
 
93
+ # base64_image1 = resize_encode_image(image[0])
94
+ # base64_image2 = resize_encode_image(image[1])
95
  content = [
96
  {
97
  "type": "text",
 
99
  },
100
  {
101
  "type": "image",
102
+ "image": f"file://{image[0]}"
103
  },
104
  {
105
  "type": "image",
106
+ "image": f"file://{image[1]}"
107
  }
108
  ]
109