Update app.py

app.py

@@ -1,77 +1,64 @@
-import os
 import streamlit as st
 from streamlit_chat import message
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains import ConversationChain
-from langchain.chains.conversation.memory import ConversationSummaryMemory
+from langchain.chains.conversation.memory import ConversationBufferMemory, ConversationSummaryMemory
 
-# Securely fetch API key from environment variables
-API_KEY = os.getenv("GOOGLE_API_KEY")
-if not API_KEY:
-    st.error("API Key not found. Please set the GOOGLE_API_KEY environment variable.")
-    st.stop()
-
-# Initialize session state variables
+# 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'] = ''
+
+# 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)
 
-# Streamlit UI
-st.set_page_config(page_title="ChatGPT Clone with Gemini", page_icon=":robot_face:")
-st.markdown("<h1 style='text-align: center;'>ChatGPT Clone with Google Gemini Flash 1.5</h1>", unsafe_allow_html=True)
+# Sidebar input for the API key
+st.sidebar.title(":D")
+st.session_state['API_Key'] = st.sidebar.text_input("Enter your Google API key:", type="password")
 
-# Sidebar API Key input for manual override (if needed)
-st.sidebar.title("Settings")
-st.sidebar.write("Ensure GOOGLE_API_KEY is set.")
-summarise_button = st.sidebar.button("Summarize Conversation")
+# 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)
 
-# Function to get response from the model
-def getresponse(user_input):
+# Step 5: Define the getresponse function
+def getresponse(userInput, api_key):
     if st.session_state['conversation'] is None:
-        llm = ChatGoogleGenerativeAI(
-            model="gemini-flash-1.5",
-            temperature=0,
-            api_key=API_KEY
-        )
+        # Initialize the Google Generative AI model
+        chat_model = ChatGoogleGenerativeAI(api_key=api_key, model="google/gemini-flash-1.5")
         st.session_state['conversation'] = ConversationChain(
-            llm=llm,
+            llm=chat_model,
             verbose=True,
-            memory=ConversationSummaryMemory(llm=llm)
+            memory=ConversationSummaryMemory(llm=chat_model)
         )
-
-    response = st.session_state['conversation'].predict(input=user_input)
+    response = st.session_state['conversation'].predict(input=userInput)
     return response
 
-# Display chat messages from history
+# Step 6: Create the Chat UI
 response_container = st.container()
 container = st.container()
+
 with container:
-    with st.form(key='input_form', clear_on_submit=True):
-        user_input = st.text_area("Your question:", key='input', height=100)
+    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 and user_input:
-            # Append user input to messages
+        if submit_button:
             st.session_state['messages'].append(user_input)
-
-            # Get model response
-            model_response = getresponse(user_input)
+            model_response = getresponse(user_input, st.session_state['API_Key'])
             st.session_state['messages'].append(model_response)
 
-# Display the chat messages
 with response_container:
-    for i, msg in enumerate(st.session_state['messages']):
-        if i % 2 == 0:
-            message(msg, is_user=True, key=f"user_{i}")
+    for i in range(len(st.session_state['messages'])):
+        if (i % 2) == 0:
+            message(st.session_state['messages'][i], is_user=True, key=str(i) + '_user')
         else:
-            message(msg, key=f"bot_{i}")
+            message(st.session_state['messages'][i], key=str(i) + '_AI')
 
-# Handle summarization
-if summarise_button:
-    if st.session_state['conversation']:
-        summary = st.session_state['conversation'].memory.buffer
-        st.sidebar.write("**Conversation Summary:**")
-        st.sidebar.write(summary)
-    else:
-        st.sidebar.error("No conversation history available for summarization.")
+# Step 7: Handle API Key and State
+if 'conversation' not in st.session_state:
+    st.session_state['conversation'] = None