chore: add default secret

app.py

@@ -11,6 +11,8 @@ import base64
 import uuid
 import os
 
+print(os.getenv("API_KEY"))
+
 
 @dataclass
 class AppState:
@@ -33,6 +35,15 @@ def create_client(api_key):
     )
 
 
+def test_api_key(client):
+    # Try making a simple request to check if the API key works
+    try:
+        # Attempt to retrieve available models as a test
+        client.models.list()
+    except Exception as e:
+        raise e
+
+
 def determine_pause(audio, sampling_rate, state):
     # Take the last 1 second of audio
     pause_length = int(sampling_rate * 1)  # 1 second
@@ -164,16 +175,50 @@ def response(state: AppState):
 
 def set_api_key(api_key, state):
     try:
-        state.client = create_client(os.getenv("API_KEY"))
-        return gr.update(value="API key set successfully!", visible=True), state
+        state.client = create_client(api_key)
+        test_api_key(state.client)  # Test the provided 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
     except Exception as e:
-        return gr.update(value="Connection error", visible=True), state
+        api_key_status = gr.update(
+            value="Invalid API key. Please try again.", visible=True
+        )
+        return api_key_status, None, None, state
+
+
+def initial_setup(state):
+    api_key = os.getenv("API_KEY")
+    if api_key:
+        try:
+            state.client = create_client(api_key)
+            test_api_key(state.client)  # Test the API key from the environment variable
+            api_key_status = gr.update(value="Use default API key", 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
+        except Exception as e:
+            # Failed to use the api_key, show input box
+            api_key_status = gr.update(
+                value="Failed to use API key from environment variable. Please enter a valid API key.",
+                visible=True,
+            )
+            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
+    else:
+        # No API key in environment variable
+        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
 
 
 with gr.Blocks() as demo:
     gr.Markdown("# Lepton AI LLM Voice Mode")
     gr.Markdown(
-        "You can find Lepton AI serverless endpoint API Key at [here](https://dashboard.lepton.ai/workspace-redirect/settings/api-tokens)"
+        "You can find Lepton AI LLM voice doc [here](https://www.lepton.ai/playground/chat/llama-3.2-3b)"
     )
     with gr.Row():
         with gr.Column(scale=3):
@@ -201,10 +246,17 @@ with gr.Blocks() as demo:
 
     state = gr.State(AppState())
 
+    # Initial setup to set API key from environment variable
+    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, state],
+        outputs=[api_key_status, api_key_input, set_key_button, state],
     )
 
     stream = input_audio.stream(