Update app.py

app.py

@@ -3,11 +3,14 @@ from groq import Groq
 import os
 
 # -----------------------------
-# GROQ CLIENT
+# GROQ CLIENT SETUP
 # -----------------------------
-# Make sure you add your Groq API key as a "Secret" on HF Spaces:
-# Go to Settings -> Secrets -> Add "GROQ_API_KEY"
-client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+api_key = os.environ.get("GROQ_API_KEY")
+
+if api_key:
+    client = Groq(api_key=api_key)
+else:
+    client = None
 
 # -----------------------------
 # CHAT FUNCTION
@@ -16,35 +19,36 @@ def chat_with_ai(message, history):
     if history is None:
         history = []
 
-    # Convert history into proper format for Groq
-    messages = [
-        {"role": "system", "content": "You are a helpful AI assistant."}
-    ]
+    if client is None:
+        reply = "❌ Groq API key not found. Please add it in Settings → Secrets → GROQ_API_KEY."
+        history.append((message, reply))
+        return history, history, ""
 
+    # Prepare messages for Groq
+    messages = [{"role": "system", "content": "You are a helpful AI assistant."}]
     for user, bot in history:
         messages.append({"role": "user", "content": user})
         messages.append({"role": "assistant", "content": bot})
-
     messages.append({"role": "user", "content": message})
 
-    # Call Groq
-    response = client.chat.completions.create(
-        model="llama-3.3-70b-versatile",  # latest working model
-        messages=messages,
-        temperature=0.7,
-        max_tokens=1024
-    )
-
-    reply = response.choices[0].message.content
+    try:
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=messages,
+            temperature=0.7,
+            max_tokens=1024
+        )
+        reply = response.choices[0].message.content
+    except Exception as e:
+        reply = f"❌ Error from Groq API: {str(e)}"
 
     history.append((message, reply))
-    # Return updated history for chatbot AND empty string for textbox
     return history, history, ""
 
 # -----------------------------
 # GRADIO UI
 # -----------------------------
-with gr.Blocks(title="AI Chatbot") as app:
+with gr.Blocks(title="Simple Groq Chatbot") as app:
     gr.Markdown("# 🤖 Simple Groq Chatbot")
 
     chatbot = gr.Chatbot(height=400)
@@ -53,7 +57,7 @@ with gr.Blocks(title="AI Chatbot") as app:
 
     state = gr.State([])
 
-    # IMPORTANT: add msg as OUTPUT to clear it
+    # Submit message
     msg.submit(chat_with_ai, [msg, state], [chatbot, state, msg])
     clear.click(lambda: ([], [], ""), None, [chatbot, state, msg])