mariesig commited on
Commit
cdd0f38
·
1 Parent(s): fe2c42f

remove unused aic_api.py and .gitattributes files

Browse files
Files changed (2) hide show
  1. .gitattributes copy +0 -36
  2. aic_api.py +0 -104
.gitattributes copy DELETED
@@ -1,36 +0,0 @@
1
- *.7z filter=lfs diff=lfs merge=lfs -text
2
- *.arrow filter=lfs diff=lfs merge=lfs -text
3
- *.bin filter=lfs diff=lfs merge=lfs -text
4
- *.bz2 filter=lfs diff=lfs merge=lfs -text
5
- *.ckpt filter=lfs diff=lfs merge=lfs -text
6
- *.ftz filter=lfs diff=lfs merge=lfs -text
7
- *.gz filter=lfs diff=lfs merge=lfs -text
8
- *.h5 filter=lfs diff=lfs merge=lfs -text
9
- *.joblib filter=lfs diff=lfs merge=lfs -text
10
- *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
- *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
- *.model filter=lfs diff=lfs merge=lfs -text
13
- *.msgpack filter=lfs diff=lfs merge=lfs -text
14
- *.npy filter=lfs diff=lfs merge=lfs -text
15
- *.npz filter=lfs diff=lfs merge=lfs -text
16
- *.onnx filter=lfs diff=lfs merge=lfs -text
17
- *.ot filter=lfs diff=lfs merge=lfs -text
18
- *.parquet filter=lfs diff=lfs merge=lfs -text
19
- *.pb filter=lfs diff=lfs merge=lfs -text
20
- *.pickle filter=lfs diff=lfs merge=lfs -text
21
- *.pkl filter=lfs diff=lfs merge=lfs -text
22
- *.pt filter=lfs diff=lfs merge=lfs -text
23
- *.pth filter=lfs diff=lfs merge=lfs -text
24
- *.rar filter=lfs diff=lfs merge=lfs -text
25
- *.safetensors filter=lfs diff=lfs merge=lfs -text
26
- saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
- *.tar.* filter=lfs diff=lfs merge=lfs -text
28
- *.tar filter=lfs diff=lfs merge=lfs -text
29
- *.tflite filter=lfs diff=lfs merge=lfs -text
30
- *.tgz filter=lfs diff=lfs merge=lfs -text
31
- *.wasm filter=lfs diff=lfs merge=lfs -text
32
- *.xz filter=lfs diff=lfs merge=lfs -text
33
- *.zip filter=lfs diff=lfs merge=lfs -text
34
- *.zst filter=lfs diff=lfs merge=lfs -text
35
- *tfevents* filter=lfs diff=lfs merge=lfs -text
36
- *.wav filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
aic_api.py DELETED
@@ -1,104 +0,0 @@
1
- import asyncio
2
- import json
3
- from typing import Any
4
- import aiofiles
5
- from aiofiles import os as aiofiles_os
6
- import aiohttp
7
- import time
8
- from pathlib import Path
9
- from constants import API_V2_URL, CHUNK_SIZE, TIMEOUT_FACTOR_MB, BASE_TIMEOUT_SECONDS
10
- import os
11
-
12
-
13
- class ApiParamsV2:
14
- def __init__(
15
- self,
16
- enhancement_level: float = 100.0,
17
- api_key: str = "",
18
- enhancement_model: str = "LARK_V2",
19
- loudness_target: float = -14,
20
- true_peak: float = -1,
21
- transcode: str = "WAV",
22
- ):
23
- self.api_key = api_key
24
- self.enhancement_level = enhancement_level
25
- self.enhancement_model = enhancement_model
26
- self.loudness_target = loudness_target
27
- self.true_peak = true_peak
28
- self.transcode = transcode
29
-
30
-
31
- # --------------------------------------------------------------
32
-
33
-
34
- async def upload_and_enhance_v2(
35
- url: str,
36
- file_path: str,
37
- api_key: str,
38
- arguments: dict[str, Any],
39
- ) -> str | None:
40
- form_data = aiohttp.FormData()
41
- form_data.add_field("media_enhancement", json.dumps(arguments))
42
-
43
- async with aiofiles.open(file_path, "rb") as file:
44
- form_data.add_field(
45
- "file",
46
- file,
47
- content_type="application/octet-stream",
48
- filename=Path(file_path).name,
49
- )
50
-
51
- async with aiohttp.ClientSession(headers={"X-API-Key": api_key}) as session:
52
- async with session.post(url, data=form_data) as response:
53
- if response.status != 201:
54
- raise Exception(await response.text())
55
- response_json = await response.json()
56
- uid = response_json["uid"]
57
- print(f"Uploaded file's uid: {uid}")
58
- return uid
59
-
60
-
61
- async def download_enhanced_media_v2(
62
- url: str,
63
- output_file_path: str,
64
- api_key: str,
65
- ) -> int:
66
- async with aiohttp.ClientSession(headers={"X-API-Key": api_key}) as session:
67
- async with session.get(url) as response:
68
- if response.status == 200:
69
- await aiofiles_os.makedirs(Path(output_file_path).parent, exist_ok=True)
70
- async with aiofiles.open(output_file_path, "wb") as f:
71
- async for chunk in response.content.iter_chunked(CHUNK_SIZE):
72
- await f.write(chunk)
73
- print(f"Download successfully to: {output_file_path}")
74
- return response.status
75
-
76
-
77
- def process_file_v2(input_file_path: str, output_file_path: str, params: ApiParamsV2) -> None:
78
- api_key = params.api_key
79
- arguments: dict[str, float | str] = {
80
- "enhancement_level": params.enhancement_level,
81
- "enhancement_model": params.enhancement_model,
82
- "loudness_target": params.loudness_target,
83
- "true_peak": params.true_peak,
84
- "transcode": params.transcode,
85
- }
86
- url = f"{API_V2_URL}/medias"
87
-
88
- try:
89
- generated_name = asyncio.run(
90
- upload_and_enhance_v2(url, input_file_path, api_key, arguments)
91
- )
92
- except Exception as e:
93
- raise RuntimeError(f"Failed to upload file: {e}")
94
- response = 412
95
- file_size_bytes = os.path.getsize(input_file_path)
96
- file_size_mb = file_size_bytes / (1024 * 1024)
97
- timeout_seconds = int(file_size_mb * TIMEOUT_FACTOR_MB) + BASE_TIMEOUT_SECONDS
98
- start_time = time.time()
99
- while response == 412 and time.time() - start_time < timeout_seconds:
100
- time.sleep(5)
101
- url = f"{API_V2_URL}/medias/{generated_name}/file"
102
- response = asyncio.run(download_enhanced_media_v2(url, output_file_path, api_key))
103
- if response == 412:
104
- raise TimeoutError(f"Download timed out after {timeout_seconds} seconds. Please try again.")