Update app.py

app.py

@@ -3,10 +3,10 @@ import gradio as gr
 from groq import Groq
 
 # Read API key from Hugging Face Space secret
-api_key = userdata.get('GROQ_API')
+api_key = os.environ.get("GROQ_API")
 
 if not api_key:
-  raise ValueError(" No Groq API key found in Colab Secrets. Please add it first.")
+    raise ValueError("No Groq API key found. Please add it as a secret.")
 
 # Initialize Groq client
 client = Groq(api_key=api_key)
@@ -15,44 +15,45 @@ models = client.models.list()
 
 print("=== AVAILABLE MODELS ===\n")
 for m in models.data:
-  print("-", m.id)
+    print("-", m.id)
 
 # You can change this later if needed
 MODEL_NAME = "openai/gpt-oss-safeguard-20b"
 
-
 def chat_groq(message, history):
-  """
-  message: latest user message (string)
-  history: list of [user,bot] pairs from Gradio
-  """
-  try:
-    system_prompt = "You are a helpful AI assistant."
-
-    messages = [{"role": "system", "content": system_prompt}]
-
-    for user_msg, bot_msg in history:
-      messages.append({"role": "user", "content": user_msg})
-      if bot_msg:
-        messages.append({"role": "user", "content": message})
+    """
+    message: latest user message (string)
+    history: list of [user,bot] pairs from Gradio
+    """
+    try:
+        system_prompt = "You are a helpful AI assistant."
+
+        messages = [{"role": "system", "content": system_prompt}]
 
-    completion =client.chat.completions.create(
-        model= MODEL_NAME,
-        messages=messages,
-        max_tokens=1024,
-        temperature=0.7
-    )
+        for user_msg, bot_msg in history:
+            messages.append({"role": "user", "content": user_msg})
+            if bot_msg:
+                messages.append({"role": "assistant", "content": bot_msg})
+
+        # Add latest user message
+        messages.append({"role": "user", "content": message})
 
-    reply = completion.choices[0].message.content
-    return reply
+        completion = client.chat.completions.create(
+            model=MODEL_NAME,
+            messages=messages,
+            max_tokens=1024,
+            temperature=0.7
+        )
 
-  except Exception as e:
-    return f"Error: {e}"
+        reply = completion.choices[0].message["content"]
+        return reply
 
+    except Exception as e:
+        return f"Error: {e}"
 
 ui = gr.ChatInterface(
     fn=chat_groq,
-    title="Affan's GenAI App(Groq)"
+    title="Affan's GenAI App (Groq)"
 )
 
-ui.launch()
+ui.launch()