most-client 1.0.15__py3-none-any.whl → 1.0.17__py3-none-any.whl

most/api.py

@@ -1,9 +1,24 @@
-from typing import List, Dict
+import io
+import os
+import uuid
+from pathlib import Path
+from typing import Dict, List, Optional
+
 import json5
 import requests
 from adaptix import Retort
-from most.types import Audio, Result, Script, JobStatus, Text, StoredAudioData, is_valid_id, DialogResult
-from pathlib import Path
+from pydub import AudioSegment
+
+from most.types import (
+    Audio,
+    DialogResult,
+    JobStatus,
+    Result,
+    Script,
+    StoredAudioData,
+    Text,
+    is_valid_id,
+)
 
 
 class MostClient(object):
@@ -131,6 +146,17 @@ class MostClient(object):
                              files={"audio_file": f})
         return self.retort.load(resp.json(), Audio)
 
+    def upload_audio_segment(self, audio: AudioSegment,
+                             audio_name: Optional[str] = None) -> Audio:
+        f = io.BytesIO()
+        audio.export(f, format="mp3")
+        f.seek(0)
+        if audio_name is None:
+            audio_name = uuid.uuid4().hex + ".mp3"
+        resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
+                         files={"audio_file": (audio_name, f, 'audio/mp3')})
+        return self.retort.load(resp.json(), Audio)
+
     def upload_audio_url(self, audio_url) -> Audio:
         resp = self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_url",
                          json={"audio_url": audio_url})
@@ -248,3 +274,18 @@ class MostClient(object):
 
     def __repr__(self):
         return "<MostClient(model_id='%s')>" % (self.model_id, )
+
+    def get_audio_segment_by_url(self, audio_url,
+                                 format=None):
+        if format is None:
+            format = os.path.splitext(audio_url)[1]
+            format = format.strip().lower()
+
+        resp = self.session.get(audio_url,
+                                timeout=None)
+        if resp.status_code >= 400:
+            raise RuntimeError("Audio url is not accessable")
+
+        audio = AudioSegment.from_file(io.BytesIO(resp.content),
+                                       format=format)
+        return audio

most/async_api.py

@@ -1,9 +1,24 @@
-from typing import List, Dict
-import json5
-from adaptix import Retort
-from most.types import Audio, Result, Script, JobStatus, Text, StoredAudioData, is_valid_id, DialogResult
+import io
+import os
+import uuid
 from pathlib import Path
+from typing import Dict, List, Optional
+
 import httpx
+import json5
+from adaptix import Retort
+from pydub import AudioSegment
+
+from most.types import (
+    Audio,
+    DialogResult,
+    JobStatus,
+    Result,
+    Script,
+    StoredAudioData,
+    Text,
+    is_valid_id,
+)
 
 
 class AsyncMostClient(object):
@@ -135,6 +150,17 @@ class AsyncMostClient(object):
                                    files={"audio_file": f})
         return self.retort.load(resp.json(), Audio)
 
+    async def upload_audio_segment(self, audio: AudioSegment,
+                                   audio_name: Optional[str] = None) -> Audio:
+        f = io.BytesIO()
+        audio.export(f, format="mp3")
+        f.seek(0)
+        if audio_name is None:
+            audio_name = uuid.uuid4().hex + ".mp3"
+        resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload",
+                               files={"audio_file": (audio_name, f, 'audio/mp3')})
+        return self.retort.load(resp.json(), Audio)
+
     async def upload_text(self, text: str) -> Text:
         resp = await self.post(f"https://api.the-most.ai/api/external/{self.client_id}/upload_text",
                                json={"text": text})
@@ -146,8 +172,8 @@ class AsyncMostClient(object):
         return self.retort.load(resp.json(), Audio)
 
     async def list_audios(self,
-                    offset: int = 0,
-                    limit: int = 10) -> List[Audio]:
+                          offset: int = 0,
+                          limit: int = 10) -> List[Audio]:
         resp = await self.get(f"https://api.the-most.ai/api/external/{self.client_id}/list?offset={offset}&limit={limit}")
         audio_list = resp.json()
         return self.retort.load(audio_list, List[Audio])
@@ -253,3 +279,18 @@ class AsyncMostClient(object):
 
     def __repr__(self):
         return "<AsyncMostClient(model_id='%s')>" % (self.model_id, )
+
+    async def get_audio_segment_by_url(self, audio_url,
+                                       format=None):
+        if format is None:
+            format = os.path.splitext(audio_url)[1]
+            format = format.strip().lower()
+
+        resp = await self.session.get(audio_url,
+                                      timeout=None)
+        if resp.status_code >= 400:
+            raise RuntimeError("Audio url is not accessable")
+
+        audio = AudioSegment.from_file(io.BytesIO(resp.content),
+                                       format=format)
+        return audio

most/types.py

@@ -1,7 +1,8 @@
 import re
 from dataclasses import dataclass
-from dataclasses_json import dataclass_json, DataClassJsonMixin
-from typing import Optional, List, Literal, Dict
+from typing import Dict, List, Literal, Optional
+
+from dataclasses_json import DataClassJsonMixin, dataclass_json
 
 
 @dataclass_json

{most_client-1.0.15.dist-info → most_client-1.0.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: most-client
-Version: 1.0.15
+Version: 1.0.17
 Summary: Most AI API for https://the-most.ai
 Home-page: https://github.com/the-most-ai/most-client
 Author: George Kasparyants
@@ -26,6 +26,7 @@ Requires-Dist: pytest
 Requires-Dist: tox
 Requires-Dist: twine
 Requires-Dist: httpx
+Requires-Dist: pydub
 Dynamic: author
 Dynamic: author-email
 Dynamic: classifier

most_client-1.0.17.dist-info/RECORD

@@ -0,0 +1,9 @@
+most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
+most/api.py,sha256=yRGL7lZ5qBWSdaclz-t7fZAJ_k1zu0YbHptUX9uaVdM,11772
+most/async_api.py,sha256=xRZPsb64i1d_U0PWIJKC_7fPMjZRPfamvvtSMyFlA5E,12505
+most/types.py,sha256=apUz6FHFAHWJNVvRh57wVQk5DFw2XKnLLhkAyWgtvfw,2825
+most_client-1.0.17.dist-info/METADATA,sha256=51ED0zyrdWP5OqTlY31mCcXRpBmk39y1BXNWJYx2BMY,1027
+most_client-1.0.17.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+most_client-1.0.17.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
+most_client-1.0.17.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
+most_client-1.0.17.dist-info/RECORD,,

{most_client-1.0.15.dist-info → most_client-1.0.17.dist-info}/WHEEL

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

most_client-1.0.15.dist-info/RECORD

@@ -1,9 +0,0 @@
-most/__init__.py,sha256=62uFFeM_1VVR83K3bTYWK3PEoqnmFCy9aWYerQ6U4Ds,67
-most/api.py,sha256=xFuvjL45gNHDJyPd9_QG3u7uv75CMi3EWX5m47odd7E,10598
-most/async_api.py,sha256=eB7bF5a_4hDBCn64VXqXIcu0GIlzYdEIkqihh287qEo,11271
-most/types.py,sha256=tZFuRKx6hxnOvpLw5pPe7p3H1mW_cxFkTc2fJTKxHiI,2824
-most_client-1.0.15.dist-info/METADATA,sha256=-PTbL9VVxEatBuS1VB_icOQjvEZjIjg1XAWxSo8leFw,1006
-most_client-1.0.15.dist-info/WHEEL,sha256=nn6H5-ilmfVryoAQl3ZQ2l8SH5imPWFpm1A5FgEuFV4,91
-most_client-1.0.15.dist-info/top_level.txt,sha256=2g5fk02LKkM1hV3pVVti_LQ60TToLBcR2zQ3JEKGVk8,5
-most_client-1.0.15.dist-info/zip-safe,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
-most_client-1.0.15.dist-info/RECORD,,