Update page1.py

page1.py

@@ -6,8 +6,10 @@ from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
 from streamlit_chat import message
+import time
 def text():
-
+    with st.chat_message("assistant"):
+        st.write("Hello 👋, how may I assist you today?")
     apiKey = "AIzaSyAXkkcrrUBjPEgj93tZ9azy7zcS1wI1jUA"
     msgs = StreamlitChatMessageHistory(key="special_app_key")
 
@@ -30,8 +32,7 @@ def text():
             st.markdown(message["content"])
 
     prompt = st.chat_input("Say something")
-    with st.chat_message("assistant"):
-        st.write("Hello 👋, how may I assist you today?")
+
     if prompt:
         with st.chat_message("user").markdown(prompt):
             st.session_state.messages.append(
@@ -40,8 +41,11 @@ def text():
                     "content": prompt
                 }
             )
-        for chunk in llm_chain.stream(prompt):
-            text_output = chunk.get("text", "")
+
+        with st.spinner('Generating...'):
+            for chunk in llm_chain.stream(prompt):
+                text_output = chunk.get("text", "")
+
 
         with st.chat_message("assistant").markdown(text_output):
             st.session_state.messages.append(
@@ -49,9 +53,4 @@ def text():
                     "role": "assistant",
                     "content": text_output
                 }
-            )
-
-
-
-
-
+            )