Update app.py

app.py

@@ -2,12 +2,14 @@ import os
 import gradio as gr
 from groq import Groq
 
-# Get GROQ API key from environment variable
-GROQ_API_KEY = os.getenv("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")  # Ensure the correct environment variable name is used
+# ========== Option 1: Directly set your API key (quick testing) ==========
+GROQ_API_KEY = "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"
 
-# Check if the API key is found
-if not GROQ_API_KEY:
-    raise ValueError("API key is missing. Please set the GROQ_API_KEY environment variable.")
+# ========== Option 2: Read from environment variable ==========
+# Uncomment below two lines if you prefer reading from environment
+# GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+# if not GROQ_API_KEY:
+#     raise ValueError("API key is missing. Please set the GROQ_API_KEY environment variable.")
 
 # Initialize the Groq client
 client = Groq(api_key=GROQ_API_KEY)
@@ -27,7 +29,7 @@ def chat_with_groq(message, history):
 
         # Call Groq API
         response = client.chat.completions.create(
-            model="llama3-70b-8192",  # ✅ Updated to active supported model
+            model="llama3-70b-8192",  # ✅ Active supported model
             messages=messages,
             temperature=0.7,
             max_tokens=500,
@@ -62,4 +64,3 @@ with gr.Blocks() as demo:
 # Launch the Gradio app
 demo.launch()
 
-