Update app.py

app.py

@@ -135,62 +135,66 @@ with gr.Blocks() as demo:
         Generate and mix radio promos effortlessly with AI tools!
     """)
 
+    with gr.Tabs():
+    # Step 1: Generate Script
     with gr.Tab("Step 1: Generate Script"):
-    with gr.Row():
-        user_prompt = gr.Textbox(label="Promo Idea", placeholder="E.g., A 30-second promo for a morning show.")
-        llama_model_id = gr.Textbox(label="Llama Model ID", value="meta-llama/Meta-Llama-3-8B-Instruct")
-        duration = gr.Slider(label="Duration (seconds)", minimum=15, maximum=60, step=15, value=30)
-
-    generate_script_button = gr.Button("Generate Script")
-    script_output = gr.Textbox(label="Generated Script", lines=5)
-    sound_design_output = gr.Textbox(label="Sound Design", lines=3)
-    music_suggestion_output = gr.Textbox(label="Music Suggestions", lines=3)
-
-    generate_script_button.click(
-        fn=lambda user_prompt, model_id, duration: generate_script(user_prompt, model_id, hf_token, duration),
-        inputs=[user_prompt, llama_model_id, duration],  # Gradio components only
-        outputs=[script_output, sound_design_output, music_suggestion_output],
-    )
-
-
-        with gr.Tab("Step 2: Generate Voice"):
-            with gr.Row():
-                speaker = gr.Textbox(label="Voice Style (optional)", placeholder="E.g., male, female, or neutral.")
-
-            generate_voice_button = gr.Button("Generate Voice")
-            voice_output = gr.Audio(label="Generated Voice", type="filepath")
-
-            generate_voice_button.click(
-                fn=generate_voice,
-                inputs=[script_output, speaker],
-                outputs=[voice_output],
-            )
-
-        with gr.Tab("Step 3: Generate Music"):
-            with gr.Row():
-                audio_length = gr.Slider(label="Music Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
-
-            generate_music_button = gr.Button("Generate Music")
-            music_output = gr.Audio(label="Generated Music", type="filepath")
-
-            generate_music_button.click(
-                fn=generate_music,
-                inputs=[music_suggestion_output, audio_length],
-                outputs=[music_output],
-            )
-
-        with gr.Tab("Step 4: Blend Audio"):
-            with gr.Row():
-                ducking = gr.Checkbox(label="Enable Ducking", value=True)
-
-            blend_button = gr.Button("Blend Audio")
-            final_output = gr.Audio(label="Final Promo Audio", type="filepath")
-
-            blend_button.click(
-                fn=blend_audio,
-                inputs=[voice_output, music_output, ducking],
-                outputs=[final_output],
-            )
+        with gr.Row():
+            user_prompt = gr.Textbox(label="Promo Idea", placeholder="E.g., A 30-second promo for a morning show.")
+            llama_model_id = gr.Textbox(label="Llama Model ID", value="meta-llama/Meta-Llama-3-8B-Instruct")
+            duration = gr.Slider(label="Duration (seconds)", minimum=15, maximum=60, step=15, value=30)
+
+        generate_script_button = gr.Button("Generate Script")
+        script_output = gr.Textbox(label="Generated Script", lines=5)
+        sound_design_output = gr.Textbox(label="Sound Design", lines=3)
+        music_suggestion_output = gr.Textbox(label="Music Suggestions", lines=3)
+
+        generate_script_button.click(
+            fn=lambda user_prompt, model_id, duration: generate_script(user_prompt, model_id, hf_token, duration),
+            inputs=[user_prompt, llama_model_id, duration],
+            outputs=[script_output, sound_design_output, music_suggestion_output],
+        )
+
+    # Step 2: Generate Voice
+    with gr.Tab("Step 2: Generate Voice"):
+        with gr.Row():
+            speaker = gr.Textbox(label="Voice Style (optional)", placeholder="E.g., male, female, or neutral.")
+
+        generate_voice_button = gr.Button("Generate Voice")
+        voice_output = gr.Audio(label="Generated Voice", type="filepath")
+
+        generate_voice_button.click(
+            fn=generate_voice,
+            inputs=[script_output, speaker],
+            outputs=[voice_output],
+        )
+
+    # Step 3: Generate Music
+    with gr.Tab("Step 3: Generate Music"):
+        with gr.Row():
+            audio_length = gr.Slider(label="Music Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
+
+        generate_music_button = gr.Button("Generate Music")
+        music_output = gr.Audio(label="Generated Music", type="filepath")
+
+        generate_music_button.click(
+            fn=generate_music,
+            inputs=[music_suggestion_output, audio_length],
+            outputs=[music_output],
+        )
+
+    # Step 4: Blend Audio
+    with gr.Tab("Step 4: Blend Audio"):
+        with gr.Row():
+            ducking = gr.Checkbox(label="Enable Ducking", value=True)
+
+        blend_button = gr.Button("Blend Audio")
+        final_output = gr.Audio(label="Final Promo Audio", type="filepath")
+
+        blend_button.click(
+            fn=blend_audio,
+            inputs=[voice_output, music_output, ducking],
+            outputs=[final_output],
+        )
 
     gr.Markdown("""
         <hr>