Update app.py

app.py

@@ -4,41 +4,48 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains import ConversationChain
 from langchain.chains.conversation.memory import ConversationBufferMemory, ConversationSummaryMemory
 
-# Step 2: Initialize session state variables
+# Step 1: Set up Google API key
+# Retrieve API key from secrets (ensure you set this up in Streamlit secrets)
+google_api_key = st.secrets["google_api_key"]
+
+# Step 2: Initialize Session State Variables
 if 'conversation' not in st.session_state:
     st.session_state['conversation'] = None
 if 'messages' not in st.session_state:
     st.session_state['messages'] = []
 if 'API_Key' not in st.session_state:
-    st.session_state['API_Key'] = ''
+    st.session_state['API_Key'] = google_api_key  # Use the Google API key from secrets
 
 # Step 3: Build the Streamlit UI
 st.set_page_config(page_title="Chat GPT Clone", page_icon=":robot_face:")
 st.markdown("<h1 style='text-align: center;'>How can I assist you? </h1>", unsafe_allow_html=True)
 
-# Sidebar input for the API key
+# Sidebar for API key input
 st.sidebar.title(":D")
-st.session_state['API_Key'] = st.sidebar.text_input("Enter your Google API key:", type="password")
+st.session_state['API_Key'] = st.sidebar.text_input("Enter your Google API key", type="password")
 
 # Summarization button
 summarise_button = st.sidebar.button("Summarise the conversation", key="summarise")
-if summarise_button and st.session_state['conversation'] is not None:
-    summarise_placeholder = st.sidebar.write("Conversation Summary:\n\n" + st.session_state['conversation'].memory.buffer)
+if summarise_button:
+    summarise_placeholder = st.sidebar.write("Nice chatting with you my friend <3️:\n\n" + str(st.session_state['conversation'].memory.buffer))
 
-# Step 5: Define the getresponse function
+# Step 4: Define the getresponse function using Google's Gemini
 def getresponse(userInput, api_key):
     if st.session_state['conversation'] is None:
-        # Initialize the Google Generative AI model
-        chat_model = ChatGoogleGenerativeAI(api_key=api_key, model="google/gemini-flash-1.5")
+        # Initialize the Google generative model
+        chat = ChatGoogleGenerativeAI(
+            model="gemini-1.5-flash",
+            google_api_key=api_key
+        )
         st.session_state['conversation'] = ConversationChain(
-            llm=chat_model,
+            llm=chat,
             verbose=True,
-            memory=ConversationSummaryMemory(llm=chat_model)
+            memory=ConversationSummaryMemory(llm=chat)
         )
     response = st.session_state['conversation'].predict(input=userInput)
     return response
 
-# Step 6: Create the Chat UI
+# Step 5: Creating the Chat UI
 response_container = st.container()
 container = st.container()
 
@@ -46,7 +53,6 @@ with container:
     with st.form(key='my_form', clear_on_submit=True):
         user_input = st.text_area("Your question goes here:", key='input', height=100)
         submit_button = st.form_submit_button(label='Send')
-
         if submit_button:
             st.session_state['messages'].append(user_input)
             model_response = getresponse(user_input, st.session_state['API_Key'])
@@ -59,6 +65,9 @@ with response_container:
         else:
             message(st.session_state['messages'][i], key=str(i) + '_AI')
 
-# Step 7: Handle API Key and State
-if 'conversation' not in st.session_state:
-    st.session_state['conversation'] = None
+# Step 6: Handling State and API Key
+st.session_state['API_Key'] = st.sidebar.text_input("Enter your Google API key", type="password")
+
+# Step 7: Running the App
+if __name__ == "__main__":
+    st.run()