Update app.py

app.py

@@ -8,11 +8,10 @@ from langchain_groq import ChatGroq
 from langsmith import traceable
 from typing import TypedDict
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-from google.colab import userdata  # Only needed in Google Colab
 
 # ✅ Load API keys from Google Colab secrets
-GROQ_API_KEY = userdata.get('GROQ_API_KEY')  # Ensure this is set in Colab
-LANGSMITH_API_KEY = userdata.get('LANGSMITH_API_KEY')
+GROQ_API_KEY = os.getenv("GROQ_API_KEY")  # Get from Hugging Face secrets
+LANGSMITH_API_KEY = os.getenv("LANGSMITH_API_KEY")
 
 # ✅ Set environment variables
 os.environ["GROQ_API_KEY"] = GROQ_API_KEY