Update app.py

app.py

@@ -15,7 +15,7 @@ import plotly.express as px
 import re
 import os
 
-API_KEY = os.getenv("GROQ_TOKENS")
+GROQ_API_KEY = os.getenv("GROQ_TOKENS")
 
 st.set_page_config(layout="wide")
 # Custom CSS for Jupyter-like appearance
@@ -131,7 +131,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 = API_KEY
+    api_key: str = GROQ_API_KEY
 
     @property
     def _llm_type(self) -> str:
@@ -164,7 +164,7 @@ class GroqLLM(LLM):
 class GroqTextLLM(LLM):
     model: str = "deepseek-r1-distill-llama-70b"
     temperature: float = 0.7
-    api_key: str = API_KEY
+    api_key: str = GROQ_API_KEY
 
     @property
     def _llm_type(self) -> str: