Skip to content

Commit 3311162

Browse files
authored
Update inference.py
1 parent 83c51d7 commit 3311162

File tree

1 file changed

+17
-13
lines changed

1 file changed

+17
-13
lines changed

tabs/inference/inference.py

Lines changed: 17 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -257,21 +257,25 @@ def run_advanced_rvc(model_name, youtube_url, export_format, f0_method, f0_up_ke
257257

258258
final_mix = instrumental_audio.overlay(lead_audio) if lead_audio else instrumental_audio
259259
if backing_audio:
260-
output_backing = backing_audio.overlay(lead_audio)
260+
backing_mix = backing_audio.overlay(lead_audio) if lead_audio else backing_audio
261261
final_mix = final_mix.overlay(backing_audio)
262-
backing_mix = backing_audio.overlay(lead_audio) if lead_audio else instrumental_audio
263-
if backing_audio:
264-
output_bfile = backing_audio.overlay(lead_audio)
265262

266-
267-
268-
# ai Mix
269-
output_file = os.path.join(current_dir, f"aicover_{model_name}.{export_format.lower()}")
270-
final_mix.export(output_file, format=export_format.lower())
271-
output_bfile = os.path.join(current_dir, f"aicover-with_{backing}_{model_name}.{export_format.lower()}")
272-
backing_mix.export(output_bfile, format=export_format.lower())
273-
logging.info("Mixing complete. Output saved to %s", output_file)
274-
return f"Mixed file saved as: {output_file}", output_file, output_bfile, output_bfile, rvc_lead, rvc_backing
263+
output_file = os.path.join(current_dir, f"aicover_{model_name}.{export_format.lower()}")
264+
output_backing_file = os.path.join(current_dir, f"aicover-with_{backing}_{model_name}.{export_format.lower()}")
265+
266+
final_mix.export(output_file, format=export_format.lower())
267+
backing_mix.export(output_backing_file, format=export_format.lower())
268+
269+
logging.info("Mixing complete. Output saved to %s and %s", output_file, output_backing_file)
270+
271+
return f"Mixed file saved as: {output_file}", output_file, output_backing_file, rvc_lead, rvc_backing
272+
else:
273+
logging.info("No backing audio provided. Only the main mix will be exported.")
274+
275+
output_file = os.path.join(current_dir, f"aicover_{model_name}.{export_format.lower()}")
276+
final_mix.export(output_file, format=export_format.lower())
277+
278+
return f"Mixed file saved as: {output_file}", output_file, None, rvc_lead, rvc_backing
275279

276280
except Exception as e:
277281
logging.exception("Error during advanced RVC pipeline: %s", e)

0 commit comments

Comments
 (0)