Update core_agent.py

core_agent.py

@@ -21,9 +21,8 @@ from plotly.subplots import make_subplots
 from dotenv import load_dotenv
 
 from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain.prompts import PromptTemplate
-from langchain.chains import LLMChain
-from langchain.schema import HumanMessage, SystemMessage
+from langchain_core.prompts import PromptTemplate
+from langchain_core.messages import HumanMessage, SystemMessage
 
 warnings.filterwarnings("ignore")
 load_dotenv()
@@ -37,7 +36,7 @@ CARD_BG  = "#1A1A2E"
 # ─── LLM Setup ───────────────────────────────────────────────────────────────
 def get_llm(api_key: str):
     return ChatGoogleGenerativeAI(
-        model="gemini-2.5-flash",
+        model="gemini-1.5-flash",
         google_api_key=api_key,
         temperature=0.3,
         convert_system_message_to_human=True,
@@ -315,4 +314,4 @@ Respond ONLY in valid JSON like:
                 text = text[4:]
         return json.loads(text.strip())
     except Exception:
-        return {"chart_type": "distribution_plots", "x_col": None, "y_col": None, "reason": "Default chart"}
+        return {"chart_type": "distribution_plots", "x_col": None, "y_col": None, "reason": "Default chart"}