migrate to gemini pro api

app.py

@@ -1,13 +1,16 @@
 import streamlit as st
-import os
+# import os
 from langchain.prompts import PromptTemplate
-from langchain_community.chat_models import ChatOpenAI
+# from langchain_community.chat_models import ChatOpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
 from dotenv import load_dotenv
 load_dotenv()
 
 def getBlogContent(topic, number_of_words, field, points):
-  llm = ChatOpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"),
-               model_name=os.getenv("MODEL_NAME"), temperature=0)
+  # llm = ChatOpenAI(openai_api_key=os.getenv("OPENAI_API_KEY"),
+  #              model_name=os.getenv("MODEL_NAME"), temperature=0)
+
+  gemini_llm = ChatGoogleGenerativeAI(model="gemini-pro",temperature=0.3)
 
   template = """
   You are an expert in "{field}". Write a blog to express your opinion on the following topic "{topic}".
@@ -26,7 +29,7 @@ def getBlogContent(topic, number_of_words, field, points):
   )
 
   final_prompt = prompt.format(field=field, topic=topic, number_of_words=number_of_words, points=points)
-  response = llm.invoke(final_prompt)
+  response = gemini_llm.invoke(final_prompt)
   print(response)
   return response.content
 

requirements.txt

@@ -3,6 +3,7 @@ uvicorn
 langchain
 langchain-openai
 langchain-community
+langchain-google-genai
 huggingface_hub
 python-box
 streamlit