Update app.py

app.py

@@ -1,25 +1,39 @@
 import gradio as gr
 import requests
+import os
 
-# Set your Groq API key
-GROQ_API_KEY = "Groq_API_Key"
+# Load API key from environment variables (Recommended for security)
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")
+
+# Groq API URL
 GROQ_API_URL = "https://api.groq.com/openai/v1/chat/completions"
 
 def chat_with_groq(user_input):
+    if not GROQ_API_KEY:
+        return "Error: API key is missing. Set the GROQ_API_KEY environment variable."
+
     headers = {
         "Authorization": f"Bearer {GROQ_API_KEY}",
         "Content-Type": "application/json"
     }
+
     data = {
         "model": "llama3-8b-8192",  # ✅ Using only LLaMA 3-8B model
         "messages": [{"role": "user", "content": user_input}]
     }
-    response = requests.post(GROQ_API_URL, json=data, headers=headers)
-    
-    if response.status_code == 200:
-        return response.json()["choices"][0]["message"]["content"]
-    else:
-        return f"Error: {response.status_code}, {response.text}"
+
+    try:
+        response = requests.post(GROQ_API_URL, json=data, headers=headers)
+        response_json = response.json()
+
+        if response.status_code == 200:
+            return response_json.get("choices", [{}])[0].get("message", {}).get("content", "No response")
+        else:
+            error_message = response_json.get("error", {}).get("message", "Unknown error")
+            return f"Error {response.status_code}: {error_message}"
+
+    except requests.exceptions.RequestException as e:
+        return f"Request failed: {e}"
 
 # Create Gradio UI
 iface = gr.Interface(
@@ -32,3 +46,4 @@ iface = gr.Interface(
 
 # Launch the chatbot
 iface.launch()
+