Fix UI: Replace Gallery with 4 separate Audio widgets

app.py

@@ -44,10 +44,10 @@ def generate_audio(prompt, seconds_total=11):
     global model, model_config, device
     
     if model is None:
-        return [], "Model not loaded. Please wait..."
+        return None, None, None, None, "Model not loaded. Please wait..."
     
     if not prompt or not prompt.strip():
-        return [], "Please enter a text prompt."
+        return None, None, None, None, "Please enter a text prompt."
     
     # Set up text and timing conditioning (repeat for batch_size)
     conditioning = [{
@@ -68,8 +68,6 @@ def generate_audio(prompt, seconds_total=11):
             batch_size=4  # Generate 4 variations
         )
         
-        # Rearrange audio batch: [batch, channels, samples] -> [channels, batch*samples]
-        # Then split back into individual files
         sample_rate = model_config["sample_rate"]
         audio_files = []
         
@@ -96,13 +94,14 @@ def generate_audio(prompt, seconds_total=11):
             sf.write(filename, audio, sample_rate)
             audio_files.append(filename)
         
-        return audio_files, f"Generated 4 variations for: '{prompt}'"
+        # Return 4 separate audio files and status message
+        return audio_files[0], audio_files[1], audio_files[2], audio_files[3], f"Generated 4 variations for: '{prompt}'"
     
     except Exception as e:
         import traceback
         error_msg = f"Error generating audio: {str(e)}\n{traceback.format_exc()}"
         print(error_msg)
-        return [], error_msg
+        return None, None, None, None, error_msg
 
 # Load model on startup
 print("Initializing model...")
@@ -141,19 +140,16 @@ with gr.Blocks(title="Stable Audio Open Small - 4 Variations") as demo:
         
         with gr.Column():
             status_output = gr.Textbox(label="Status", interactive=False)
-            audio_gallery = gr.Gallery(
-                label="Generated Audio Variations",
-                show_label=True,
-                elem_id="gallery",
-                columns=2,
-                rows=2,
-                height="auto"
-            )
+            gr.Markdown("### Generated Audio Variations")
+            audio_output_1 = gr.Audio(label="Variation 1", interactive=False)
+            audio_output_2 = gr.Audio(label="Variation 2", interactive=False)
+            audio_output_3 = gr.Audio(label="Variation 3", interactive=False)
+            audio_output_4 = gr.Audio(label="Variation 4", interactive=False)
     
     generate_btn.click(
         fn=generate_audio,
         inputs=[prompt_input, seconds_input],
-        outputs=[audio_gallery, status_output]
+        outputs=[audio_output_1, audio_output_2, audio_output_3, audio_output_4, status_output]
     )
     
     gr.Markdown("""