Update app.py

app.py

@@ -1,41 +1,19 @@
 
 import streamlit as st
-
-
 from langchain_openai import ChatOpenAI
+from langchain.schema import AIMessage, HumanMessage, SystemMessage
 
-#import os
-#os.environ["OPENAI_API_KEY"] = "sk-PLfFw23dd932dfg34446dftyvvdfgdfgmvXr2dL8hVowXdt"
-
-
-from langchain.schema import (
-    AIMessage,
-    HumanMessage,
-    SystemMessage
-)
-
-# From here down is all the StreamLit UI
+# Set up Streamlit page
 st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
 st.header("DIBYAJYOTI'S PERSONAL GPT ASSISTANT")
 
 
-
 if "sessionMessages" not in st.session_state:
-     st.session_state.sessionMessages = [
+    st.session_state.sessionMessages = [
         SystemMessage(content="You are a helpful assistant.")
     ]
 
-
-
-def load_answer(question):
-
-    st.session_state.sessionMessages.append(HumanMessage(content=question))
-
-    assistant_answer  = chat.invoke(st.session_state.sessionMessages )
-
-    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
-
-    return assistant_answer.content
+chat = ChatOpenAI(temperature=0)
 
 
 def get_text():
@@ -43,18 +21,23 @@ def get_text():
     return input_text
 
 
-chat = ChatOpenAI(temperature=0)
+def load_answer(question):
+    st.session_state.sessionMessages.append(HumanMessage(content=question))
+    
 
+    assistant_answer = chat.invoke(st.session_state.sessionMessages)
 
+    st.session_state.sessionMessages.append(AIMessage(content=assistant_answer.content))
 
+    return assistant_answer.content
 
-user_input=get_text()
-submit = st.button('CLICK HERE TO GET YOUR RESPONSE')  
+user_input = get_text()
+submit = st.button('CLICK HERE TO GET YOUR RESPONSE')
 
 if submit:
-    
-    response = load_answer(user_input)
-    st.subheader("Answer:")
-
-    st.write(response)
-
+    if user_input:
+        response = load_answer(user_input)
+        st.subheader("Answer:")
+        st.write(response)
+    else:
+        st.warning("Please enter a question.")