Update app.py

app.py

@@ -113,30 +113,32 @@ with gr.Blocks() as demo:
     gr.Markdown("# 🎧 AI Radio Imaging with Llama 3 + MusicGen (Zero GPU)")
 
     # Script Generation Section
-    with gr.Box():
-        gr.Markdown("## Step 1: Generate the Promo Script")
-        user_prompt = gr.Textbox(label="Enter your promo idea", placeholder="E.g., A 15-second hype jingle for a morning talk show.")
-        llama_model_id = gr.Textbox(label="Llama 3 Model ID", value="meta-llama/Meta-Llama-3-70B")
-        generate_script_button = gr.Button("Generate Promo Script")
-        script_output = gr.Textbox(label="Generated Script", interactive=False)
-
-        generate_script_button.click(
-            fn=radio_imaging_script,
-            inputs=[user_prompt, llama_model_id],
-            outputs=script_output
-        )
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## Step 1: Generate the Promo Script")
+            user_prompt = gr.Textbox(label="Enter your promo idea", placeholder="E.g., A 15-second hype jingle for a morning talk show.")
+            llama_model_id = gr.Textbox(label="Llama 3 Model ID", value="meta-llama/Meta-Llama-3-70B")
+            generate_script_button = gr.Button("Generate Promo Script")
+            script_output = gr.Textbox(label="Generated Script", interactive=False)
+
+            generate_script_button.click(
+                fn=radio_imaging_script,
+                inputs=[user_prompt, llama_model_id],
+                outputs=script_output
+            )
 
     # Audio Generation Section
-    with gr.Box():
-        gr.Markdown("## Step 2: Generate the Sound")
-        audio_length = gr.Slider(label="Audio Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
-        generate_audio_button = gr.Button("Generate Sound from Script")
-        audio_output = gr.Audio(label="Generated Audio", type="filepath")
-
-        generate_audio_button.click(
-            fn=radio_imaging_audio,
-            inputs=[script_output, audio_length],
-            outputs=audio_output
-        )
+    with gr.Row():
+        with gr.Column():
+            gr.Markdown("## Step 2: Generate the Sound")
+            audio_length = gr.Slider(label="Audio Length (tokens)", minimum=128, maximum=1024, step=64, value=512)
+            generate_audio_button = gr.Button("Generate Sound from Script")
+            audio_output = gr.Audio(label="Generated Audio", type="filepath")
+
+            generate_audio_button.click(
+                fn=radio_imaging_audio,
+                inputs=[script_output, audio_length],
+                outputs=audio_output
+            )
 
 demo.launch(debug=True)