Update app.py

app.py

@@ -11,12 +11,8 @@ from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 from langchain_groq import ChatGroq
 from langchain_google_genai import ChatGoogleGenerativeAI
 
-# Initialize Groq Langchain chat object and conversation
-groq_chat = ChatGroq(groq_api_key=os.environ.get("GROQ_API_KEY"))
-
-# Initialize Google Langchain chat object and conversation
-os.environ["GOOGLE_API_KEY"] = os.environ.get("GOOGLE_API_KEY")   
-google_chat = ChatGoogleGenerativeAI(model="gemini-1.5-pro")
+# os.environ["GROQ_API_KEY"] = os.environ.get("GROQ_API_KEY")   
+# os.environ["GOOGLE_API_KEY"] = os.environ.get("GOOGLE_API_KEY")   
 
 # Initialize memory to manages the chat history, 
 # ensuring the AI remembers the specified number of history messages, in this case 8.
@@ -27,11 +23,25 @@ def generate_response(user_input, history, model, temperature, max_tokens, top_p
     print( "Model =", model)
 
     if model.startswith("gemini"):
-        chat = google_chat
-        # chat.model = model
+        chat = ChatGoogleGenerativeAI(
+            google_api_key = os.environ.get("GOOGLE_API_KEY"),
+            model = model,
+            temperature=temperature,
+            max_tokens=max_tokens,
+            timeout=None,
+            max_retries=2,
+            top_p = top_p
+        )        
     else:
-        chat = groq_chat
-        chat.model_name = model
+        chat = ChatGroq(
+            groq_api_key = os.environ.get("GROQ_API_KEY"),
+            model_name = model,
+            temperature=temperature,
+            max_tokens=max_tokens, 
+            request_timeout=None,
+            max_retries=2,
+            top_p = top_p            
+        )
 
     prompt = ChatPromptTemplate.from_messages(
         [