domenico/karaoke-me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixing client output option

Browse Source
This commit is contained in:
Domenico Testa
2020-04-17 22:59:23 +02:00
parent 416704acbf
commit cd7047e96d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
client/client.py
View File

@@ -80,9 +80,9 @@ def list_jobs(ctx: click.core.Context):
@cli.command('remove-vocals') @cli.command('remove-vocals')
@click.option('--file', type=click.File(mode='rb'), required=True) @click.option('--file', type=click.File(mode='rb'), required=True)
@click.option('--output-path', '-o', type=click.Path(writable=True)) @click.option('--output', '-o', type=click.Path(writable=True), required=True)
@click.pass_context @click.pass_context
def remove_vocals(ctx: click.core.Context, file: click.File, output_path: click.Path): def remove_vocals(ctx: click.core.Context, file: click.File, output: click.Path):
'''Removes the vocal part from an audio/video file''' '''Removes the vocal part from an audio/video file'''
response = api_post(ctx, '/jobs') response = api_post(ctx, '/jobs')
job = response.json() job = response.json()
@@ -116,11 +116,11 @@ def remove_vocals(ctx: click.core.Context, file: click.File, output_path: click.
print(f"File processed in {processing_time}s") print(f"File processed in {processing_time}s")
response = requests.get(job['output_url'], stream=True) response = requests.get(job['output_url'], stream=True)
with open(output_path, 'wb') as f: with open(output, 'wb') as f:
for chunk in response.raw: for chunk in response.raw:
f.write(chunk) f.write(chunk)
print(f"Saved file to: {output_path}") print(f"Saved file to: {output}")
if __name__ == '__main__': if __name__ == '__main__':