Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from langchain_core import AIMessage, HumanMessage
+from langchain_core.messages import AIMessage, HumanMessage
 
 
 def get_response(user_input):
@@ -23,22 +23,27 @@ with st.sidebar:
     website_url = st.text_input("Website URL")
     openai_apikey = st.text_input("Enter your OpenAI API key")
 
-#user_input
-user_query = st.chat_input("Type your message here...")
-if user_query is not None and user_query !="":
-    response = get_response(user_query)
-    st.session_state.chat_history .append(HumanMessage(content=user_query))
-    st.session_state.chat_history .append(AIMessageMessage(content=response))
+if (website_url is None or website_url == "") or (openai_apikey is None or openai_apikey == ""):
+    st.info("Please ensure if website URL and Open AI api key are entered")
     
-            
-#conversation
-for message in st.session_state.chat_history:
-    if isinstance(message, AIMessage): # checking if the messsage is the instance of an AI message
-        with st.chat_message("AI"):
-            st.write(message.content)
-    elif isinstance(message, HumanMessage): # checking if the messsage is the instance of a Human
-        with st.chat_message("Human"):
-            st.write(message.content)
+
+else:
+    #user_input
+    user_query = st.chat_input("Type your message here...")
+    if user_query is not None and user_query !="":
+        response = get_response(user_query)
+        st.session_state.chat_history .append(HumanMessage(content=user_query))
+        st.session_state.chat_history .append(AIMessageMessage(content=response))
+        
+                
+    #conversation
+    for message in st.session_state.chat_history:
+        if isinstance(message, AIMessage): # checking if the messsage is the instance of an AI message
+            with st.chat_message("AI"):
+                st.write(message.content)
+        elif isinstance(message, HumanMessage): # checking if the messsage is the instance of a Human
+            with st.chat_message("Human"):
+                st.write(message.content)