Update app.py

app.py

@@ -15,6 +15,9 @@ from io import StringIO
 from typing import Optional, List
 import plotly.express as px
 import re
+import os
+
+groq_api = os.getenv("API_KEY")
 
 st.set_page_config(layout="wide")
 # Custom CSS for Jupyter-like appearance
@@ -130,7 +133,7 @@ if 'df' not in st.session_state:
 class GroqLLM(LLM):
     model: str = "qwen-2.5-coder-32b"
     temperature: float = 0.3
-    api_key: str = GROQ_TOKENS
+    api_key: str = groq_api
 
     @property
     def _llm_type(self) -> str:
@@ -163,7 +166,7 @@ class GroqLLM(LLM):
 class GroqTextLLM(LLM):
     model: str = "deepseek-r1-distill-llama-70b"
     temperature: float = 0.7
-    api_key: str = GROQ_TOKENS
+    api_key: str = groq_api
 
     @property
     def _llm_type(self) -> str: