Upload 3 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+ffmpeg filter=lfs diff=lfs merge=lfs -text

ffmpeg

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51bbdd978250ff7239d213940ff2c92ea56e7d768e8db98d9cbc4079d82e42dc
+size 78999296

qili.py

@@ -0,0 +1,128 @@
+import os
+import sys
+import uuid
+import json
+import requests
+import traceback
+
+from importlib import import_module
+
+sys.path.append(os.getcwd())
+
+conf=None
+def makeConfReady():
+    global conf
+    if conf is None:
+        conf=import_module("qili-conf").conf
+
+def fetch(request, timeout=60000):
+    try:
+        makeConfReady()
+        with requests.Session() as session:
+            response = session.post(
+                conf["api"],
+                headers={
+                    'Content-Type': 'application/json',
+                    'x-application-id': conf["apiKey"],
+                    "x-session-token":conf["token"]
+                },
+                data=request if isinstance(request, str) else json.dumps(request),
+                timeout=timeout / 1000
+            )
+            response.raise_for_status()
+
+            # Parse JSON response
+            response_data = response.json()
+
+            if 'data' not in response_data:
+                raise ValueError(response_data.get('message') or response_data.get('statusText') or 'Unknown error')
+
+            if 'errors' in response_data:
+                raise ValueError(response_data['errors'])
+
+            return response_data['data']
+
+    except requests.exceptions.Timeout:
+        raise TimeoutError('Timeout from qili service')
+
+
+def upload_bytes(bytes, key=None, ext=".wav"):
+    makeConfReady()
+    try: 
+        if key is None:
+            key=f"_temp_/1/{str(uuid.uuid4())}{ext}"
+        data = fetch({
+            "query": '''
+                query($key:String!){
+                    file_upload_token(key:$key){
+                        token
+                        key
+                    }
+                }
+            ''', 
+            "variables": {
+                "key": key
+            }
+        })
+        data=data["file_upload_token"]
+        files={"file":(os.path.basename(key), bytes)}
+        response = requests.post(conf["api"], files=files, data=data)
+        if response.ok:
+            data = response.json()
+            return data.get("data", {}).get("file_create", {}).get("url")
+        else:
+            raise Exception(f"{response.status_code} - {response.reason}, {response.text}")
+    except Exception as e:
+        traceback.print_exc()
+        return str(e)
+
+
+def upload(files, root_key=None):
+    makeConfReady()
+    if not isinstance(files, list):
+        shouldReturnString=True
+        files = [files]
+    
+    if(root_key==None):
+        root_key="_temp_/1/" + str(uuid.uuid4())
+        
+    try:
+        keys, queries, variables = [], [], {}
+        
+        for i, file in enumerate(files):
+            k = f"key{i}"
+            keys.append(k)
+            queries.append(f"""
+                token{i}:file_upload_token(key:${k}){{
+                    token
+                    key
+                }}
+            """)
+            variables[k] = f"{root_key}{os.path.basename(file)}"
+        
+        keys=','.join([f'${k}:String!' for k in keys])
+        queries='\n'.join(queries)
+
+        query = f"""query({ keys }){{ { queries } }}"""
+
+        data = fetch({"query": query, "variables": variables})
+        
+        tokens = list(data.values())
+        urls = []
+        
+        for i, file in enumerate(files):
+            form = {"file": open(file, "rb")}
+            response = requests.post(conf["api"], files=form, data=tokens[i])
+            
+            if response.ok:
+                data = response.json()
+                urls.append(data.get("data", {}).get("file_create", {}).get("url"))
+            else:
+                raise Exception(f"{response.status_code} - {response.reason}, {response.text}")
+        
+        if shouldReturnString:
+            urls[0]
+        return list(filter(None, urls))
+    except Exception as e:
+        traceback.print_exc()
+        return str(e)