patch_stem_render.py

# This is appended to app.py to wire in stem-based rendering.
# It imports after the UI is defined and replaces the render function.

# Override render_full_set to use demucs + stem_mixer
try:
    from stem_render import render_full_set_with_stems as _stem_render
    
    _original_render = render_full_set
    
    def render_full_set(progress=gr.Progress()):
        return _stem_render(app_state, progress)
    
    # The Gradio button was already bound to the original function name,
    # but since Python looks up function names dynamically in the module,
    # this override will be picked up on next call.
    logger.info("✅ render_full_set overridden to use demucs + stem_mixer")
except ImportError as e:
    logger.warning(f"stem_render not available: {e}. Using default render path.")