Update function.py

function.py

@@ -3,13 +3,15 @@ from langchain.chains import LLMChain, SimpleSequentialChain
 from langchain.memory import SimpleMemory
 from langchain.prompts import PromptTemplate
 from langchain.chat_models import ChatOpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 
 
 
 
 def generate_event_name(topic, audience, benefit):
     # Initialize the LLM with GPT-4 chat model
-    llm = ChatOpenAI(model="gpt-4", openai_api_key=st.secrets["OPENAI_API_KEY"])
+    # llm = ChatOpenAI(model="gpt-4", openai_api_key=st.secrets["OPENAI_API_KEY"])
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
 
     # Define the template for generating the event name
     event_name_template = (
@@ -40,10 +42,11 @@ def ai_generated_content(topic, audience, benefit, date_time, event_name):
     
     # Initialize the LLM and memory
     # llm = OpenAI(model="gpt-4",openai_api_key=st.secrets["OPENAI_API_KEY"])
-    llm = ChatOpenAI(
-        model="gpt-4", 
-        openai_api_key=st.secrets["OPENAI_API_KEY"]  # Fetch API key from Streamlit secrets
-    )
+    # llm = ChatOpenAI(
+    #     model="gpt-4", 
+    #     openai_api_key=st.secrets["OPENAI_API_KEY"]  # Fetch API key from Streamlit secrets
+    # )
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
 
 
     # hero-section headline
@@ -180,7 +183,8 @@ def ai_generated_content(topic, audience, benefit, date_time, event_name):
 
 def generate_first_person_benefits(benefits):
     # Initialize the LLM with GPT-4 chat model
-    llm = ChatOpenAI(model="gpt-4", openai_api_key=st.secrets["OPENAI_API_KEY"])
+    # llm = ChatOpenAI(model="gpt-4", openai_api_key=st.secrets["OPENAI_API_KEY"])
+    llm = ChatGoogleGenerativeAI(model="gemini-pro", google_api_key=st.secrets["GOOGLE_API_KEY"])
 
     # Define the prompt template
     prompt_template = """