mixpeek 0.7.2__py3-none-any.whl → 0.7.3__py3-none-any.whl

mixpeek/endpoints/tools.py

@@ -5,10 +5,41 @@ import base64
 from urllib.parse import urlparse
 from urllib.request import urlretrieve
 from tqdm import tqdm
+from PIL import Image
+import io
 
 class Tools:
     def __init__(self):
         self.video = self.Video(self)
+        self.image = self.Image(self)
+
+    class Image:
+        def __init__(self, parent):
+            pass
+
+        def process(self, image_source: str):
+            # Download image if it's a URL
+            if urlparse(image_source).scheme in ('http', 'https'):
+                with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+                    urlretrieve(image_source, temp_file.name)
+                    image_source = temp_file.name
+
+            # Open and process the image
+            with Image.open(image_source) as img:
+                # Convert image to RGB mode if it's not already
+                if img.mode != 'RGB':
+                    img = img.convert('RGB')
+
+                # Convert image to base64
+                buffered = io.BytesIO()
+                img.save(buffered, format="JPEG")
+                base64_string = base64.b64encode(buffered.getvalue()).decode('utf-8')
+
+            # Remove temporary file if it was created
+            if urlparse(image_source).scheme in ('http', 'https'):
+                os.unlink(image_source)
+
+            return base64_string
 
     class Video:
         def __init__(self, parent):
@@ -25,6 +56,7 @@ class Tools:
                 }
                 yield data
 
+
 class VideoChunker:
     def __init__(self, video_source, chunk_interval, target_resolution):
         self.video_source = video_source

{mixpeek-0.7.2.dist-info → mixpeek-0.7.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mixpeek
-Version: 0.7.2
+Version: 0.7.3
 Summary: Mixpeek Python SDK
 Home-page: https://github.com/mixpeek/mixpeek-python
 Author: Ethan Steininger
@@ -14,6 +14,7 @@ Requires-Dist: requests ==2.32.3
 Requires-Dist: pydantic ==2.7.3
 Requires-Dist: tqdm ==4.66.4
 Requires-Dist: urllib3 ==2.2.1
+Requires-Dist: pillow ==10.4.0
 
 # Mixpeek Python SDK
 

{mixpeek-0.7.2.dist-info → mixpeek-0.7.3.dist-info}/RECORD

@@ -6,8 +6,8 @@ mixpeek/endpoints/collections.py,sha256=ffOBenNHG-mVpjT7kQcOfn78-wjgjdQTVyrloyeA
 mixpeek/endpoints/embed.py,sha256=ybDqyIt0oJiwIdH1QSwuV-CPeJgd-zW0etNAcBdgZYE,2290
 mixpeek/endpoints/index.py,sha256=QgMgys1CCukepCpK351aVXRxEsxBfI2YriX7aaH1NrU,1939
 mixpeek/endpoints/search.py,sha256=u3onbss7Llx0AgcTSRtel6bGO5Ly3jvraXPcA26GzsY,2387
-mixpeek/endpoints/tools.py,sha256=QH3X0tYKr-W_9ApENgP7PeuLJi7Qr9W2jrrgQFH4F2A,3567
-mixpeek-0.7.2.dist-info/METADATA,sha256=d_YN-IkRgs4iNHw6v7dN9qXg1jT46NvpjI74VqPqwKM,10585
-mixpeek-0.7.2.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
-mixpeek-0.7.2.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
-mixpeek-0.7.2.dist-info/RECORD,,
+mixpeek/endpoints/tools.py,sha256=T8qb9zngY3V_vwtmBBrTqPvqZl1QIL8q3at3UGIj4bg,4709
+mixpeek-0.7.3.dist-info/METADATA,sha256=AO5vsb1itiebws3tEFrXnHuywAZei3TBgRXVW3Lfgy4,10616
+mixpeek-0.7.3.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
+mixpeek-0.7.3.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
+mixpeek-0.7.3.dist-info/RECORD,,