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

mixpeek/client.py

@@ -5,6 +5,7 @@ from .endpoints.collections import Collections
 from .endpoints.index import Index
 from .endpoints.search import Search
 from .endpoints.tools import Tools
+from .endpoints.register import Register
 
 
 class Mixpeek:
@@ -20,6 +21,7 @@ class Mixpeek:
         self.collections = Collections(self.base_url, self.headers)
         self.index = Index(self.base_url, self.headers)
         self.search = Search(self.base_url, self.headers)
+        self.register = Register(self.base_url, self.headers)
 
         # tools is all local
-        self.tools = Tools()
+        self.tools = Tools()

mixpeek/endpoints/index.py

@@ -1,4 +1,5 @@
 import requests
+from .tasks import Task
 
 class Index:
     def __init__(self, base_url, headers):
@@ -12,33 +13,37 @@ class Index:
             base_data["settings"] = settings
         return base_data
 
-    # def upload(self, file, collection_id, metadata=None, settings=None):
-    #     try:
-    #         api_url = f"{self.base_url}index/upload"
-    #         files = {"file": file}
-    #         data = self._prepare_data({"collection_id": collection_id}, metadata, settings)
-    #         response = requests.post(api_url, files=files, data=data, headers=self.headers)
-    #         response.raise_for_status()
-    #         return response.json()
-    #     except requests.RequestException as e:
-    #         return {"error": str(e)}
-
     def url(self, target_url, collection_id, metadata=None, settings=None):
         try:
             endpoint = f"{self.base_url}index/url"
             data = self._prepare_data({"url": target_url, "collection_id": collection_id}, metadata, settings)
+
             response = requests.post(endpoint, json=data, headers=self.headers)
             response.raise_for_status()
-            return response.json()
+            task_id = response.json().get("task_id")
+            if task_id:
+                return Task(self.base_url, self.headers, task_id)
+            else:
+                return response.json()
         except requests.RequestException as e:
             return {"error": str(e)}
-
-    def youtube(self, youtube_video_id, collection_id, metadata=None, settings=None):
+        
+    def upload(self, file_path, collection_id, metadata=None, settings=None):
         try:
-            api_url = f"{self.base_url}index/youtube"
-            data = self._prepare_data({"youtube_video_id": youtube_video_id, "collection_id": collection_id}, metadata, settings)
-            response = requests.post(api_url, json=data, headers=self.headers)
+            endpoint = f"{self.base_url}index/upload"
+            data = self._prepare_data({"collection_id": collection_id}, metadata, settings)
+            
+            with open(file_path, 'rb') as file:
+                files = [('file', (file.name, file, 'application/octet-stream'))]
+                response = requests.post(endpoint, headers=self.headers, data=data, files=files)
+            
             response.raise_for_status()
-            return response.json()
+            task_id = response.json().get("task_id")
+            if task_id:
+                return Task(self.base_url, self.headers, task_id)
+            else:
+                return response.json()
         except requests.RequestException as e:
-            return {"error": str(e)}
+            return {"error": str(e)}
+        except IOError as e:
+            return {"error": f"File error: {str(e)}"}

mixpeek/endpoints/register.py

@@ -0,0 +1,34 @@
+import requests
+from .tasks import Task
+
+class Register:
+    def __init__(self, base_url, headers):
+        self.base_url = base_url
+        self.headers = headers
+
+    def _prepare_data(self, base_data, metadata=None, settings=None):
+        if metadata is not None:
+            base_data["metadata"] = metadata
+        if settings is not None:
+            base_data["settings"] = settings
+        return base_data
+
+    def faces(self, file_path, collection_id, metadata=None, settings=None):
+        try:
+            endpoint = f"{self.base_url}register/faces"
+            data = self._prepare_data({"collection_id": collection_id}, metadata, settings)
+            
+            with open(file_path, 'rb') as file:
+                files = [('file', (file.name, file, 'application/octet-stream'))]
+                response = requests.post(endpoint, headers=self.headers, data=data, files=files)
+            
+            response.raise_for_status()
+            task_id = response.json().get("task_id")
+            if task_id:
+                return Task(self.base_url, self.headers, task_id)
+            else:
+                return response.json()
+        except requests.RequestException as e:
+            return {"error": str(e)}
+        except IOError as e:
+            return {"error": f"File error: {str(e)}"}

mixpeek/endpoints/tasks.py

@@ -0,0 +1,26 @@
+import requests
+import time
+
+class Task:
+    def __init__(self, base_url, headers, task_id):
+        self.base_url = base_url
+        self.headers = headers
+        self.task_id = task_id
+
+    def get_task_status(self, task_id):
+        try:
+            endpoint = f"{self.base_url}tasks/{task_id}"
+            response = requests.get(endpoint, headers=self.headers)
+            response.raise_for_status()
+            return response.json()
+        except requests.RequestException as e:
+            return {"error": str(e)}
+
+    def wait_for_done(self, sleep_interval=2, callback=None):
+        while True:
+            status = self.get_task_status(self.task_id)
+            if callback:
+                callback(status)
+            if status.get("status") == "DONE":
+                return status
+            time.sleep(sleep_interval)

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.8.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mixpeek
-Version: 0.7.2
+Version: 0.8.0
 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.8.0.dist-info/RECORD

@@ -0,0 +1,15 @@
+mixpeek/__init__.py,sha256=XDdcK7wTEOEcF1cp-GeWmgPJ21Ny1R9pB0PPNrdDTMo,28
+mixpeek/client.py,sha256=FcSjc73WMMKPHzZKye2_XicwJP1bY_iaMWdIv-ft1Rw,906
+mixpeek/exceptions.py,sha256=Orhdo5UFLn3fcWVJtlgkznW8Iy5ndL96h0qTY8zOlDA,235
+mixpeek/endpoints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+mixpeek/endpoints/collections.py,sha256=ffOBenNHG-mVpjT7kQcOfn78-wjgjdQTVyrloyeAy90,2890
+mixpeek/endpoints/embed.py,sha256=ybDqyIt0oJiwIdH1QSwuV-CPeJgd-zW0etNAcBdgZYE,2290
+mixpeek/endpoints/index.py,sha256=kG5bf0ycIWvooicA5xowXrWH4x6nbzP6XdyK-XQHqU0,1976
+mixpeek/endpoints/register.py,sha256=ZVes4BPxQsPRNk70tGeVe_CYr4eVQKL-CTVmzpJAHM4,1314
+mixpeek/endpoints/search.py,sha256=u3onbss7Llx0AgcTSRtel6bGO5Ly3jvraXPcA26GzsY,2387
+mixpeek/endpoints/tasks.py,sha256=GBkscCYVv46JkJ4MVOtM6tL__rymnM3bmt4PrdRK04Y,839
+mixpeek/endpoints/tools.py,sha256=T8qb9zngY3V_vwtmBBrTqPvqZl1QIL8q3at3UGIj4bg,4709
+mixpeek-0.8.0.dist-info/METADATA,sha256=xrC5EB9t4qThzh2l8mafI_AAI7Fkk58c6SQhFPFVHWY,10616
+mixpeek-0.8.0.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+mixpeek-0.8.0.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
+mixpeek-0.8.0.dist-info/RECORD,,

{mixpeek-0.7.2.dist-info → mixpeek-0.8.0.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (73.0.1)
+Generator: setuptools (75.1.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

mixpeek-0.7.2.dist-info/RECORD

@@ -1,13 +0,0 @@
-mixpeek/__init__.py,sha256=XDdcK7wTEOEcF1cp-GeWmgPJ21Ny1R9pB0PPNrdDTMo,28
-mixpeek/client.py,sha256=JFQO-hLFe7MESdkNoweuoaGrc4Vnqtl55_JRvgDCspg,804
-mixpeek/exceptions.py,sha256=Orhdo5UFLn3fcWVJtlgkznW8Iy5ndL96h0qTY8zOlDA,235
-mixpeek/endpoints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-mixpeek/endpoints/collections.py,sha256=ffOBenNHG-mVpjT7kQcOfn78-wjgjdQTVyrloyeAy90,2890
-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,,