Use mimetypes for MIME type detection in image processing (#100)

Fixes #99
aaclause · Apr 27, 2024 · bc55957 · bc55957
1 parent b3a31b9
commit bc55957
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/addon/globalPlugins/openai/maindialog.py b/addon/globalPlugins/openai/maindialog.py
@@ -1,5 +1,6 @@
 import datetime
 import json
+import mimetypes
 import os
 import re
 import speech
@@ -1362,8 +1363,7 @@ def getImages(
 					)
 					path = path_resized_image
 				base64_image = encode_image(path)
-				format = path.split(".")[-1]
-				mime_type = f"image/{format}"
+				mime_type, _ = mimetypes.guess_type(path)
 				images.append({
 					"type": "image_url",
 					"image_url": {