Update main.py

main.py

@@ -7,6 +7,7 @@ from flask_cors import CORS, cross_origin
 import logging
 from dotenv import load_dotenv
 from pandasai import SmartDatalake
+from pandasai import SmartDataframe
 from pandasai.responses.response_parser import ResponseParser
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain