Update app.py

app.py

@@ -2,7 +2,7 @@ import pandas as pd
 import streamlit as st
 from langchain.agents import AgentType
 from langchain_experimental.agents import create_pandas_dataframe_agent
-from langchain_ollama import ChatOllama
+from langchain_groq import ChatGroq
 
 
 # streamlit web app configuration
@@ -12,7 +12,8 @@ st.set_page_config(
     layout="centered"
 )
 
-
+with st.sidebar:
+    groq_api_key=st.text_input("Groq API Key",value="",type="password")
 def read_data(file):
     if file.name.endswith(".csv"):
         return pd.read_csv(file)
@@ -54,9 +55,9 @@ if user_prompt:
     # add user's message to chat history and display it
     st.chat_message("user").markdown(user_prompt)
     st.session_state.chat_history.append({"role":"user","content": user_prompt})
-
+    
     # loading the LLM
-    llm = ChatOllama(model="gemma:2b", temperature=0)
+    llm =ChatGroq(model="Gemma-7b-It", groq_api_key=groq_api_key)
 
     pandas_df_agent = create_pandas_dataframe_agent(
         llm,