Update main.py

main.py

@@ -68,6 +68,8 @@ def home():
 
 
 llm = GoogleGemini(api_key=gemini_api_key)
+
+llm2 = ChatGoogleGenerativeAI(model='gemini-pro', temperature=0.6, api_key=gemini_api_key)
         
 # Initialize Firebase app
 if not firebase_admin._apps:
@@ -138,7 +140,7 @@ def marketing_rec():
     prompt = PromptTemplate.from_template('You are a business analyst. Write some marketing tips suitable for a small mobile dashboard based on this transactions dataframe {data_frame}')
 
 # Create a chain that utilizes both the LLM and the prompt template
-    chain = LLMChain(llm=llm, prompt=prompt, verbose=True)
+    chain = LLMChain(llm=llm2, prompt=prompt, verbose=True)
     data_frame = transactions_df
     response = chain.invoke(input=data_frame)
     print(response)