Update app.py

app.py

@@ -35,7 +35,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 openai.api_key = os.environ['openai_key']
 os.environ["OPENAI_API_KEY"] = os.environ['openai_key']
 
-gemini.api_key = os.environ['gemini_key']
+
 os.environ["GEMINI_API_KEY"] = os.environ['gemini_key']
 
 
@@ -129,14 +129,14 @@ def submit_message(prompt, prompt_template, temperature, max_tokens, context_len
     
     docs = new_docsearch.similarity_search(query)
 
-    gen_ai = GenerativeAI.get_client(model="gemini-1.0-pro")
-    response = gen_ai.start_chat(
+    gen_ai = ChatGoogleGenerativeAI(model="gemini-pro")
+    response = gen_ai.invoke(
         messages=query,  # Pass both history and current prompt
         max_tokens=max_tokens,
         temperature=temperature  # Adjust temperature as needed
     )
 
-    completion = response.messages[-1]  # Extract the completion message
+    completion = response.content  # Extract the completion message
 
     
     get_empty_state()