Increase readability

app.py

@@ -2,16 +2,14 @@ import gradio as gr
 import numpy as np
 import io
 from pydub import AudioSegment
-import tempfile
 import requests
-import time
 from dataclasses import dataclass, field
 from threading import Lock
 import base64
 import uuid
-import os
 import json
 import sseclient
+import os
 
 @dataclass
 class AppState:
@@ -142,49 +140,63 @@ def response(state: AppState):
 
     state.stream = None
 
-def set_api_key(api_key, state):
-    state.api_key = api_key
-    api_key_status = gr.update(value="API key set successfully!", visible=True)
-    api_key_input = gr.update(visible=False)
-    set_key_button = gr.update(visible=False)
-    return api_key_status, api_key_input, set_key_button, state
-
 def initial_setup(state):
-    if state.api_key:
-        api_key_status = gr.update(value="Using default API key", visible=True)
-        api_key_input = gr.update(visible=False)
-        set_key_button = gr.update(visible=False)
-    else:
-        api_key_status = gr.update(visible=False)
-        api_key_input = gr.update(visible=True)
-        set_key_button = gr.update(visible=True)
-    return api_key_status, api_key_input, set_key_button, state
+    if not state.api_key:
+        raise gr.Error("API key not found in environment variables. Please set the API_KEY environment variable.")
+    return gr.update(value="The API key used is supported by Herm studio", visible=True)
 
 with gr.Blocks() as demo:
     gr.Markdown("# LLM Voice Mode")
-    with gr.Row():
-        with gr.Column(scale=3):
-            api_key_input = gr.Textbox(type="password", placeholder="Enter your API Key", show_label=False, container=False)
-        with gr.Column(scale=1):
-            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)
+    
+    api_key_status = gr.Textbox(
+        show_label=False,
+        container=False,
+        interactive=False,
+        visible=True
+    )
 
     with gr.Blocks():
         with gr.Row():
-            input_audio = gr.Audio(label="Input Audio", sources="microphone", type="numpy")
-            output_audio = gr.Audio(label="Output Audio", autoplay=True, streaming=True)
-        chatbot = gr.Chatbot(label="Conversation", type="messages")
+            input_audio = gr.Audio(
+                label="Input Audio",
+                sources="microphone",
+                type="numpy"
+            )
+            output_audio = gr.Audio(
+                label="Output Audio",
+                autoplay=True,
+                streaming=True
+            )
+        chatbot = gr.Chatbot(
+            label="Conversation",
+            type="messages"
+        )
 
     state = gr.State(AppState())
 
-    demo.load(initial_setup, inputs=state, outputs=[api_key_status, api_key_input, set_key_button, state])
-
-    set_key_button.click(set_api_key, inputs=[api_key_input, state], outputs=[api_key_status, api_key_input, set_key_button, state])
+    demo.load(
+        fn=initial_setup,
+        inputs=state,
+        outputs=api_key_status
+    )
 
-    stream = input_audio.stream(process_audio, [input_audio, state], [state], stream_every=0.25, time_limit=60)
+    input_audio.stream(
+        fn=process_audio,
+        inputs=[input_audio, state],
+        outputs=[state],
+        stream_every=0.25,
+        time_limit=60
+    )
 
-    respond = input_audio.stop_recording(response, [state], [chatbot, output_audio, state])
-    respond.then(lambda s: s.conversation, [state], [chatbot])
+    respond = input_audio.stop_recording(
+        fn=response,
+        inputs=[state],
+        outputs=[chatbot, output_audio, state]
+    )
+    respond.then(
+        fn=lambda s: s.conversation,
+        inputs=[state],
+        outputs=[chatbot]
+    )
 
 demo.launch()