Update page2.py

page2.py

@@ -9,9 +9,11 @@ from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
+import os
 
 # Streamlit app
 def image():
+    key = os.environ.get("api_key")
     st.markdown("""
                         <style>
                             .anim-typewriter {
@@ -58,7 +60,7 @@ def image():
         image_url = f"data:image/jpeg;base64,{image_base64}"
 
         return image_url
-    apiKey = "AIzaSyAXkkcrrUBjPEgj93tZ9azy7zcS1wI1jUA"
+    apiKey = key
 
     llm = ChatGoogleGenerativeAI(model="gemini-pro-vision", google_api_key=apiKey)