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

mixpeek/endpoints/index.py

@@ -5,31 +5,29 @@ class Index:
         self.base_url = base_url
         self.headers = headers
 
-    def upload(self, file, collection_id, metadata=None, settings=None):
-        try:
-            url = f"{self.base_url}index/upload"
-            files = {"file": file}
-            data = {
-                "collection_id": collection_id,
-                "metadata": metadata,
-                "settings": settings
-            }
-            response = requests.post(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 _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 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, url, collection_id, metadata=None, settings=None):
+    def url(self, target_url, collection_id, metadata=None, settings=None):
         try:
-            url = f"{self.base_url}index/url"
-            data = {
-                "url": url,
-                "collection_id": collection_id,
-                "metadata": metadata,
-                "settings": settings
-            }
-            response = requests.post(url, json=data, headers=self.headers)
+            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()
         except requests.RequestException as e:
@@ -37,14 +35,9 @@ class Index:
 
     def youtube(self, youtube_video_id, collection_id, metadata=None, settings=None):
         try:
-            url = f"{self.base_url}index/youtube"
-            data = {
-                "youtube_video_id": youtube_video_id,
-                "collection_id": collection_id,
-                "metadata": metadata,
-                "settings": settings
-            }
-            response = requests.post(url, json=data, headers=self.headers)
+            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)
             response.raise_for_status()
             return response.json()
         except requests.RequestException as e:

mixpeek/endpoints/search.py

@@ -1,4 +1,6 @@
 import requests
+import os
+import json
 
 class Search:
     def __init__(self, base_url, headers):
@@ -25,27 +27,32 @@ class Search:
         except requests.RequestException as e:
             return {"error": str(e)}
 
-    def upload(self, file, filters=None, page=1, page_size=10):
-        try:
-            url = f"{self.base_url}search/upload"
-            files = {"file": file}
-            data = {
-                "filters": filters or "{}",
-                "page": page,
-                "page_size": page_size
-            }
-            response = requests.post(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 upload(self, file_path, filters=None, page=1, page_size=10):
+    #     try:
+    #         url = f"{self.base_url}search/upload"
+            
+    #         filename = os.path.basename(file_path)
+    #         files = {
+    #             'file': (filename, open(file_path, 'rb'), 'application/octet-stream')
+    #         }
+            
+    #         payload = {
+    #             'filters': json.dumps(filters or {}),
+    #             'page': str(page),
+    #             'page_size': str(page_size)
+    #         }
+            
+    #         response = requests.post(url, headers=self.headers, data=payload, files=files)
+    #         response.raise_for_status()
+    #         return response.json()
+    #     except requests.RequestException as e:
+    #         return {"error": str(e)}
 
-    def url(self, url, input_type="file", filters=None, modality="text", page=1, page_size=10):
+    def url(self, target_url, filters=None, modality="text", page=1, page_size=10):
         try:
             url = f"{self.base_url}search/url"
             data = {
-                "url": url,
-                "input_type": input_type,
+                "url": target_url,
                 "filters": filters or {},
                 "modality": modality,
                 "pagination": {

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.1.dist-info → mixpeek-0.7.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mixpeek
-Version: 0.7.1
+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.1.dist-info → mixpeek-0.7.3.dist-info}/RECORD

@@ -4,10 +4,10 @@ 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=2WcjrRHCMqqj0ddWEkl1f5U81Ni9xKySC1_sdkrhnhM,1857
-mixpeek/endpoints/search.py,sha256=fp78miF5bSWTiQn32NMoth9mlLBsj7NfjNzS6HrEz3Q,2170
-mixpeek/endpoints/tools.py,sha256=QH3X0tYKr-W_9ApENgP7PeuLJi7Qr9W2jrrgQFH4F2A,3567
-mixpeek-0.7.1.dist-info/METADATA,sha256=mbjStU-fvEpahmG3iCgAlOSD5Id6IL8KOEh3sMyUwiA,10585
-mixpeek-0.7.1.dist-info/WHEEL,sha256=Mdi9PDNwEZptOjTlUcAth7XJDFtKrHYaQMPulZeBCiQ,91
-mixpeek-0.7.1.dist-info/top_level.txt,sha256=EJ8Jc4IhqyUwnUlBwKbs498Ju4O9a-IDh2kXc_lo6Vg,8
-mixpeek-0.7.1.dist-info/RECORD,,
+mixpeek/endpoints/index.py,sha256=QgMgys1CCukepCpK351aVXRxEsxBfI2YriX7aaH1NrU,1939
+mixpeek/endpoints/search.py,sha256=u3onbss7Llx0AgcTSRtel6bGO5Ly3jvraXPcA26GzsY,2387
+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,,