Update app.py

app.py

@@ -8,6 +8,7 @@ from datetime import datetime
 from src import preprocess_and_load_df, load_agent, ask_agent, decorate_with_code, show_response, get_from_user, load_smart_df, ask_question
 from dotenv import load_dotenv
 from langchain_groq.chat_models import ChatGroq
+from langchain_google_genai import GoogleGenerativeAI
 from streamlit_feedback import streamlit_feedback
 from huggingface_hub import HfApi
 st.set_page_config(layout="wide")
@@ -16,7 +17,7 @@ st.set_page_config(layout="wide")
 load_dotenv()
 Groq_Token = os.environ["GROQ_API_KEY"]
 hf_token = os.environ["HF_TOKEN"]
-models = {"llama3":"llama3-70b-8192","mixtral": "mixtral-8x7b-32768", "llama2": "llama2-70b-4096", "gemma": "gemma-7b-it"}
+models = {"llama3":"llama3-70b-8192","mixtral": "mixtral-8x7b-32768", "llama2": "llama2-70b-4096", "gemma": "gemma-7b-it", "gemini-pro": "gemini-pro"}
 
 self_path = os.path.dirname(os.path.abspath(__file__))