Update app.py

app.py

@@ -0,0 +1,14 @@
+import pandas as pd
+from pandasai import SmartDataframe
+from langchain_groq.chat_models import ChatGroq
+from langchain_community.llms import Ollama 
+import sqlite3
+import os
+
+llm = ChatGroq(model_name="llama3-70b-8192", api_key = os.environ["GROQ_API_KEY"])
+
+df = pd.read_excel('data.xlsx')
+df = SmartDataframe(df, config={"llm": llm})
+
+print( df.chat('Which are the 5 happiest countries?'))
+print( df.chat('What is the sum of the GDPs of the 2 happiest countries?'))