Update app.py

app.py

@@ -5,21 +5,6 @@ from groq import Groq
 # Read API key from Hugging Face Space secret
 api_key = os.environ.get("GROQ_API")
 
-if not api_key:
-    raise ValueError("No Groq API key found. Please add it as a secret.")
-
-# Initialize Groq client
-client = Groq(api_key=api_key)
-
-models = client.models.list()
-
-print("=== AVAILABLE MODELS ===\n")
-for m in models.data:
-    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)
@@ -35,20 +20,18 @@ def chat_groq(message, history):
       if bot_msg:
         messages.append({"role": "user", "content": message})
 
-        completion = client.chat.completions.create(
-            model=MODEL_NAME,
-            messages=messages,
-            max_tokens=1024,
-            temperature=0.7
-)
-
-# Access the content attribute directly
-        reply = completion.choices[0].message.content
-        return reply
+    completion =client.chat.completions.create(
+        model= MODEL_NAME,
+        messages=messages,
+        max_tokens=1024,
+        temperature=0.7
+    )
 
+    reply = completion.choices[0].message.content
+    return reply
 
-    except Exception as e:
-        return f"Error: {e}"
+  except Exception as e:
+    return f"Error: {e}"
 
 ui = gr.ChatInterface(
     fn=chat_groq,