Update app.py

app.py

@@ -1,20 +1,30 @@
 import gradio as gr
 from groq import Groq
+import os
 
-def chatbot(message, history, api_key):
-    # Initialize Groq client with the provided API key
-    client = Groq(api_key=api_key)
+# Default API key for examples (replace with a dummy value or leave empty)
+DEFAULT_API_KEY = os.environ.get("GROQ_API_KEY", "")
 
-    # Prepare the messages including the conversation history
-    messages = [
-        {"role": "system", "content": "You are a helpful assistant."}
-    ]
-    for human, assistant in history:
-        messages.append({"role": "user", "content": human})
-        messages.append({"role": "assistant", "content": assistant})
-    messages.append({"role": "user", "content": message})
+def chatbot(message, history, api_key):
+    # Use the provided API key, or fall back to the default for examples
+    api_key = api_key or DEFAULT_API_KEY
+    
+    if not api_key:
+        return "Please enter a valid Groq API key to use the chatbot."
 
     try:
+        # Initialize Groq client with the API key
+        client = Groq(api_key=api_key)
+
+        # Prepare the messages including the conversation history
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant."}
+        ]
+        for human, assistant in history:
+            messages.append({"role": "user", "content": human})
+            messages.append({"role": "assistant", "content": assistant})
+        messages.append({"role": "user", "content": message})
+
         # Create the chat completion
         completion = client.chat.completions.create(
             model="llama-3.2-90b-text-preview",
@@ -44,7 +54,8 @@ with gr.Blocks(theme="soft") as iface:
         api_key_input = gr.Textbox(
             label="Enter your Groq API Key",
             placeholder="sk-...",
-            type="password"
+            type="password",
+            value=DEFAULT_API_KEY  # Pre-fill with default key if available
         )
 
     chatbot = gr.ChatInterface(