fix: better layout

app.py

@@ -180,18 +180,24 @@ with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=3):
             api_key_input = gr.Textbox(
-                type="password", label="Enter your Lepton API Key"
+                type="password",
+                placeholder="Enter your Lepton API Key",
+                show_label=False,
+                container=False,
             )
         with gr.Column(scale=1):
-            set_key_button = gr.Button("Set API Key", scale=2, variant="secondary")
+            set_key_button = gr.Button("Set API Key", scale=2, variant="primary")
 
     api_key_status = gr.Textbox(
         show_label=False, container=False, interactive=False, visible=False
     )
 
     with gr.Blocks():
-        input_audio = gr.Audio(label="Input Audio", sources="microphone", type="numpy")
-        output_audio = gr.Audio(label="Output Audio", autoplay=True)
+        with gr.Row():
+            input_audio = gr.Audio(
+                label="Input Audio", sources="microphone", type="numpy"
+            )
+            output_audio = gr.Audio(label="Output Audio", autoplay=True)
         chatbot = gr.Chatbot(label="Conversation", type="messages")
         cancel = gr.Button("Stop Conversation", variant="stop")