Spaces:
Running
Running
| # 从日志队列获取日志 | |
| import time | |
| from pathlib import Path | |
| from PySide6.QtCore import QThread, Signal as pyqtSignal | |
| from videotrans.configure.config import ROOT_DIR, tr, settings, logger | |
| from videotrans.configure import config | |
| from videotrans.process.prepare_audio import vocal_bgm | |
| from videotrans.process.signelobj import GlobalProcessManager | |
| from videotrans.util import tools | |
| class SeparateWorker(QThread): | |
| finish_event = pyqtSignal(str) | |
| def __init__(self, *, file=None, out=None, parent=None, uuid=None): | |
| super().__init__(parent=parent) | |
| self.file = file | |
| self.uuid = uuid | |
| self.out = out | |
| self.vocal=None | |
| self.error=None | |
| def _process_callback(self,data): | |
| if isinstance(data,str): | |
| return self.finish_event.emit(f'logs:{data}') | |
| if not isinstance(data,dict): | |
| return | |
| msg_type = data.get("type") | |
| percent = data.get("percent") | |
| filename = data.get("filename") | |
| if msg_type == "file": | |
| self.finish_event.emit(f"logs:{tr('Downloading please wait')} {filename} {percent:.2f}%") | |
| else: | |
| current_file_idx = data.get("current") | |
| total_files = data.get("total") | |
| self.finish_event.emit(f"logs:{tr('Downloading please wait')} {current_file_idx}/{total_files} files") | |
| def run(self): | |
| try: | |
| uvr_models=settings.get('uvr_models') | |
| if uvr_models.startswith('spleeter'): | |
| tools.down_file_from_ms(f'{ROOT_DIR}/models/onnx', [ | |
| f"https://www.modelscope.cn/models/himyworld/videotrans/resolve/master/onnx/vocals.fp16.onnx", | |
| f"https://www.modelscope.cn/models/himyworld/videotrans/resolve/master/onnx/accompaniment.fp16.onnx" | |
| ], callback=self._process_callback) | |
| else: | |
| tools.down_file_from_ms(f'{ROOT_DIR}/models/onnx', [ | |
| f"https://www.modelscope.cn/models/himyworld/videotrans/resolve/master/onnx/{uvr_models}.onnx" | |
| ], callback=self._process_callback) | |
| p=Path(self.file) | |
| self.vocal=f"{self.out}/vocal-{p.stem}.wav" | |
| self.finish_event.emit('logs:Separating vocals from the background ...') | |
| # 如果不是wav，需要先转为wav | |
| if p.suffix.lower()!= '.wav': | |
| newfile = config.TEMP_DIR + f'/sep-{time.time()}.wav' | |
| cmd = [ | |
| "-y", | |
| "-i", | |
| self.file, | |
| "-vn", | |
| "-ac", | |
| "2", | |
| "-ar", | |
| "44100", | |
| newfile | |
| ] | |
| tools.runffmpeg(cmd) | |
| self.file = newfile | |
| tools.set_process(text='start separate...',uuid=self.uuid) | |
| kw={"input_file":self.file,"vocal_file":self.vocal,"instr_file":f"{self.out}/instrument-{p.stem}.wav","uvr_models":uvr_models} | |
| future=GlobalProcessManager.submit_task_cpu( | |
| vocal_bgm, | |
| **kw | |
| ) | |
| rs,err=future.result(timeout=3600) | |
| if rs is False: | |
| self.finish_event.emit(err) | |
| except Exception as e: | |
| logger.exception(f'分离人声背景声失败{e}',exc_info=True) | |
| msg = f"error:separate vocal and background music:{str(e)}" | |
| self.error=msg | |
| self.finish_event.emit(msg) | |
| else: | |
| self.finish_event.emit('succeed') | |