Update ai_agent.py

ai_agent.py

@@ -6,12 +6,15 @@ import plotly.express as px
 import plotly.graph_objects as go
 from langchain_groq import ChatGroq
 from prompts import ENHANCED_SYSTEM_PROMPT, get_chart_prompt, validate_plot_spec, INSIGHTS_SYSTEM_PROMPT, get_insights_prompt, validate_ai_response, final_plot_validation
+from dotenv import load_dotenv()
+
+
+load_dotenv()
 
-GROQ_API_KEY = "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"
 
 def initialize_llm():
     try:
-        os.environ["GROQ_API_KEY"] = GROQ_API_KEY
+        GROQ_API_KEY = os.environ["GROQ_API_KEY"] 
         llm = ChatGroq(model="llama-3.3-70b-versatile", api_key=GROQ_API_KEY, temperature=0.0)
         return llm
     except: