Update pages/2_Non-Interactive Chat Bot.py

pages/2_Non-Interactive Chat Bot.py

@@ -1,30 +1,33 @@
-import streamlit as st
-import os
-from langchain_google_genai import ChatGoogleGenerativeAI
-from langchain_core.prompts import ChatPromptTemplate
-
-# Set GOOGLE_API_KEY 
-os.environ['GOOGLE_API_KEY'] = 'AIzaSyBHPIIk4-BOgXvnQ2_o6c2wTGpY2ByRIDs'
-
-st.title("Non-Interactive Chat Bot")
-
-model_gemini = ChatGoogleGenerativeAI(
-    model='gemini-pro',
-    temperature=0,
-    max_output_tokens=500,
-    convert_system_message_to_human=True
-)
-
-# Assuming model_gemini is defined correctly elsewhere
-prompt = ChatPromptTemplate.from_template("{content}")
-chain = prompt | model_gemini
-
-input_text = st.text_area('Ask your Question :', height=100)
-
-# Add a button to trigger the chatbot response
-if st.button('Submit'):
-    # Invoke the chain with input text
-    response = chain.invoke({'content': input_text})
-    
-    # Display response on the Streamlit web interface
-    st.write(response.content)
+import streamlit as st
+import os
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_core.prompts import ChatPromptTemplate
+
+# Set your Gemini API key from environment variables
+gemini_api_key = os.getenv("GEMINI_API_KEY")
+
+# Initialize the Gemini API-powered chatbot model
+model_gemini = ChatGoogleGenerativeAI(
+    model='gemini-pro',  
+    temperature=0,       
+    max_output_tokens=500,
+    convert_system_message_to_human=True
+)
+
+# Create a chat prompt template
+prompt = ChatPromptTemplate.from_template("{content}")
+chain = prompt | model_gemini  
+
+# Streamlit app title
+st.title("Non-Interactive Chat Bot")
+
+# Text area for user input
+input_text = st.text_area('Ask your Question :', height=100)
+
+# Button to submit user query
+if st.button('Submit'):
+    # Invoke the chain with user input text
+    response = chain.invoke({'content': input_text})
+    
+    # Display the chatbot response
+    st.write(response.content)