valknar/audiocraft-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix audio AbortError race condition in all generation tabs

Browse Source 
Use gr.skip() instead of gr.update() for audio/waveform/duration/seed
outputs during the initial "Generating..." status yield. This prevents
Gradio's StaticAudio component from starting a fetch that gets aborted
when the actual audio path arrives in the second yield.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
Sebastian Krüger
2025-11-27 17:04:55 +01:00
parent 1935bae594
commit f5030245e5
5 changed files with 66 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/ui/tabs/audiogen_tab.py
View File

@@ -163,21 +163,21 @@ def create_audiogen_tab(
if not prompt:
yield (
gr.update(value="Please enter a prompt"),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
return
yield (
gr.update(value="🔄 Generating..."),
gr.update(visible=True, value=0),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(), # Don't update audio yet
gr.skip(), # Don't update waveform
gr.skip(), # Don't update duration
gr.skip(), # Don't update seed
)
try:
@@ -206,10 +206,10 @@ def create_audiogen_tab(
yield (
gr.update(value=f"❌ Error: {str(e)}"),
gr.update(visible=False),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
generate_btn.click(

26
src/ui/tabs/jasco_tab.py
View File

@@ -219,21 +219,21 @@ def create_jasco_tab(
if not chords:
yield (
gr.update(value="Please enter a chord progression"),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
return
yield (
gr.update(value="🔄 Generating..."),
gr.update(visible=True, value=0),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(), # Don't update audio yet
gr.skip(), # Don't update waveform
gr.skip(), # Don't update duration
gr.skip(), # Don't update seed
)
try:
@@ -270,10 +270,10 @@ def create_jasco_tab(
yield (
gr.update(value=f"❌ Error: {str(e)}"),
gr.update(visible=False),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
generate_btn.click(

26
src/ui/tabs/magnet_tab.py
View File

@@ -186,21 +186,21 @@ def create_magnet_tab(
if not prompt:
yield (
gr.update(value="Please enter a prompt"),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
return
yield (
gr.update(value="🔄 Generating..."),
gr.update(visible=True, value=0),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(), # Don't update audio yet
gr.skip(), # Don't update waveform
gr.skip(), # Don't update duration
gr.skip(), # Don't update seed
)
try:
@@ -231,10 +231,10 @@ def create_magnet_tab(
yield (
gr.update(value=f"❌ Error: {str(e)}"),
gr.update(visible=False),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
generate_btn.click(

28
src/ui/tabs/musicgen_tab.py
View File

@@ -224,11 +224,11 @@ def create_musicgen_tab(
if not prompt:
yield (
gr.update(value="Please enter a prompt"),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
return
@@ -237,14 +237,14 @@ def create_musicgen_tab(
if long_form and duration > 30:
status_msg = f"🔄 Long-form generation ({duration}s, may take several minutes)..."
# Update status
# Update status - use gr.skip() for audio to prevent race condition
yield (
gr.update(value=status_msg),
gr.update(visible=True, value=0),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(), # Don't update audio yet
gr.skip(), # Don't update waveform
gr.skip(), # Don't update duration
gr.skip(), # Don't update seed
)
try:
@@ -279,10 +279,10 @@ def create_musicgen_tab(
yield (
gr.update(value=f"❌ Error: {str(e)}"),
gr.update(visible=False),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(), # Don't update audio
gr.skip(), # Don't update waveform
gr.skip(), # Don't update duration
gr.skip(), # Don't update seed
)
generate_btn.click(

26
src/ui/tabs/style_tab.py
View File

@@ -163,21 +163,21 @@ def create_style_tab(
if not style_audio:
yield (
gr.update(value="Please upload a style reference audio"),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
return
yield (
gr.update(value="🔄 Generating..."),
gr.update(visible=True, value=0),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(), # Don't update audio yet
gr.skip(), # Don't update waveform
gr.skip(), # Don't update duration
gr.skip(), # Don't update seed
)
try:
@@ -209,10 +209,10 @@ def create_style_tab(
yield (
gr.update(value=f"❌ Error: {str(e)}"),
gr.update(visible=False),
gr.update(),
gr.update(),
gr.update(),
gr.update(),
gr.skip(),
gr.skip(),
gr.skip(),
gr.skip(),
)
generate_btn.click(