Update app.py

app.py

@@ -1,6 +1,6 @@
 # Q&A Chatbot
 from langchain_openai import OpenAI
-
+from langchain_google_genai import GoogleGenerativeAI
 from dotenv import load_dotenv
 
 load_dotenv()  # take environment variables from .env.
@@ -16,7 +16,8 @@ from langchain.chains import SequentialChain
 ## Function to load OpenAI model and get respones
 
 def get_openai_response(country_name):
-    llm=OpenAI(openai_api_key=os.environ["OPEN_API_KEY"],temperature=0.5)
+    llm = GoogleGenerativeAI(model="models/text-bison-001", google_api_key=os.environ["GOOGLE_API_KEY"])
+    #llm=OpenAI(openai_api_key=os.environ["OPEN_API_KEY"],temperature=0.5)
     capital_template=PromptTemplate(input_variables=['country'],
     template="Please tell me the capital of the {country}")
     capital_chain=LLMChain(llm=llm,prompt=capital_template,output_key="capital")