Update app.py

app.py

@@ -3,64 +3,57 @@ import gradio as gr
 from groq import Groq
 
 # Read API key from Hugging Face Space secret
-GROQ_API_KEY = os.environ.get("GROQ_API")
+api_key = userdata.get('GROQ_API')
 
-if not GROQ_API_KEY:
-    raise ValueError(
-        "GROQ_API_KEY is not set. "
-        "Go to your Space Settings -> Variables and secrets -> add it as a secret."
-    )
+if not api_key:
+  raise ValueError(" No Groq API key found in Colab Secrets. Please add it first.")
 
 # Initialize Groq client
-client = Groq(api_key=GROQ_API_KEY)
+client = Groq(api_key=api_key)
 
-# You can change this later if needed
-DEFAULT_MODEL = os.environ.get("GROQ_MODEL", "openai/gpt-oss-safeguard-20b")
+models = client.models.list()
 
+print("=== AVAILABLE MODELS ===\n")
+for m in models.data:
+  print("-", m.id)
 
-def chat_with_groq(message, history):
-    """
-    Gradio passes:
-      - message: latest user input (str)
-      - history: list of [user, bot] pairs
-    """
-    try:
-        system_prompt = "You are a helpful AI assistant for Sajid. Answer clearly and simply."
+# You can change this later if needed
+MODEL_NAME = "openai/gpt-oss-safeguard-20b"
 
-        messages = [{"role": "system", "content": system_prompt}]
 
-        # Add previous messages
-        for user_msg, bot_msg in history:
-            messages.append({"role": "user", "content": user_msg})
-            if bot_msg:
-                messages.append({"role": "assistant", "content": bot_msg})
+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."
 
-        # Add latest user message
+    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})
 
-        response = client.chat.completions.create(
-            model=DEFAULT_MODEL,
-            messages=messages,
-            max_tokens=512,
-            temperature=0.7,
-        )
+    completion =client.chat.completions.create(
+        model= MODEL_NAME,
+        messages=messages,
+        max_tokens=1024,
+        temperature=0.7
+    )
 
-        reply = response.choices[0].message.content
-        return reply
+    reply = completion.choices[0].message.content
+    return reply
 
-    except Exception as e:
-        # Show error inside the UI, useful if model name is wrong, etc.
-        return f"⚠ Error from Groq API: {e}"
+  except Exception as e:
+    return f"Error: {e}"
 
 
-demo = gr.ChatInterface(
-    fn=chat_with_groq,
-    title="Sajid's GenAI Chat (Groq + Hugging Face Space)",
-    description=(
-        "A simple LLM chat app using Groq's API, deployed on Hugging Face Spaces. "
-        "Powered by Gradio."
-    ),
+ui = gr.ChatInterface(
+    fn=chat_groq,
+    title="Affan's GenAI App(Groq)"
 )
 
-if _name_ == "_main_":
-    demo.launch()
+ui.launch()
+)