Update conversation.py

conversation.py

@@ -109,14 +109,17 @@ class Conversation:
             assert len(text_list) == len(single_turn['images']) + 1, print(text_list, len(single_turn['images']))
             message = ''
             for image_idx in range(len(single_turn['images'])):
-                # image = single_turn['images'][image_idx]
-                # image_base64 = encode_image(image)
-                # image_str = f'<img src="data:image/png;base64,{image_base64}" alt="user upload image" />'
                 image_path = single_turn['images'][image_idx]
-                if image_path == '':
-                    message += text_list[image_idx] + '<corrupt_image>'
-                else:
-                    message += text_list[image_idx] + f'![](file={image_path})'
+                image = Image.open(image_path)
+                image_base64 = encode_image(image)
+                image_str = f'<img src="data:image/png;base64,{image_base64}" alt="user upload image" />'
+                message += text_list[image_idx] + image_str
+                
+                # image_path = single_turn['images'][image_idx]
+                # if image_path == '':
+                #     message += text_list[image_idx] + '<corrupt_image>'
+                # else:
+                #     message += text_list[image_idx] + f'![](file={image_path})'
             message += text_list[-1]
 
             if i % 2 == 0: