Update main.py

main.py

@@ -9,12 +9,13 @@ from langchain_huggingface import HuggingFaceEndpoint
 from langchain.schema.runnable.config import RunnableConfig
 from langchain.schema import StrOutputParser
 from langchain_anthropic import AnthropicLLM, ChatAnthropic
+from langchain_core.prompts import ChatPromptTemplate
 
 from langchain.agents import AgentExecutor
 from langchain.agents.agent_types import AgentType
 from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent, create_csv_agent
 
-
+from pandasai import PandasAI
 import chainlit as cl
 from chainlit.input_widget import TextInput, Select, Switch, Slider