Update app.py

app.py

@@ -790,6 +790,10 @@ if __name__ == '__main__':
                 args=(conversations_path, conversations)
             )
 
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.markdown(message["content"])
+
         def generate_conv_title(llm):
             if st.session_state.chat_id == 'New Conversation':
                 output = llm.chat(
@@ -807,19 +811,9 @@ if __name__ == '__main__':
                     packed_bytes = msgpack.packb(conversations, use_bin_type=True)
                     fp.write(packed_bytes)
 
-        chat_placeholder = st.empty()
-        
-        with chat_placeholder.container():
-            for i, msg in enumerate(st.session_state.messages):
-                nkey = int(i/2)
-                if msg["role"] == "user":
-                    message(msg["content"], is_user=True, key='chat_messages_user_'+str(nkey))
-                else:
-                    message(msg["content"], is_user=False, key='chat_messages_assistant_'+str(nkey))
-         
-            pills(" ", ["Option 1", "Option 2", "Option 3"], ["🍀", "🎈", "🌈"])
-
         with st.container():
+            pills(" ", ["Option 1", "Option 2", "Option 3"], ["🍀", "🎈", "🌈"])
+            
             if prompt := st.text_input(
                 ' ', 
                 on_change=generate_conv_title, 
@@ -828,9 +822,7 @@ if __name__ == '__main__':
                 label_visibility='collapsed',
                 args=(st.session_state.llm, )
             ):
-                nkey = int(len(st.session_state.messages)/2)
-                
-                message(prompt, is_user=True, key='chat_messages_user_'+str(nkey))
+                st.chat_message("user").markdown(prompt)
                 st.session_state.messages.append({"role": "user", "content": prompt})
 
                 ai_response = generate_answer(
@@ -844,41 +836,28 @@ if __name__ == '__main__':
                     st.session_state.documents_only
                 )
 
-                message(ai_response, key='chat_messages_assistant_'+str(nkey))
-                st.session_state.messages.append({"role": "assistant", "content": ai_response})
-                
-        if prompt := st.chat_input(
-            "Message Video Game Assistant", 
-            on_submit=generate_conv_title, 
-            key='user_input',
-            args=(st.session_state.llm, )
-        ):         
-            st.chat_message("user").markdown(prompt)
-            st.session_state.messages.append({"role": "user", "content": prompt})
-            
-            ai_response = generate_answer(
-                prompt, 
-                st.session_state.client, 
-                st.session_state.collection_name, 
-                st.session_state.model, 
-                st.session_state.dense_model, 
-                st.session_state.sparse_model, 
-                "\n".join([f'{msg["role"]}: {msg["content"]}' for msg in st.session_state.messages]), 
-                st.session_state.documents_only
-            )
-            
-            with st.chat_message("assistant"):
-                message_placeholder = st.empty()
-                full_response = ""
-                for chunk in re.split(r'(\s+)', ai_response):
-                    full_response += chunk + " "
-                    time.sleep(0.05)
-                    message_placeholder.write(full_response + '▌')
-                message_placeholder.write(re.sub('▌', '', full_response))
+                with st.chat_message("assistant"):
+                    message_placeholder = st.empty()
+                    full_response = ""
+                    for chunk in re.split(r'(\s+)', ai_response):
+                        full_response += chunk + " "
+                        time.sleep(0.05)
+                        message_placeholder.write(full_response + '▌')
+                    message_placeholder.write(re.sub('▌', '', full_response))
                                           
-            st.session_state.messages.append({"role": "assistant", "content": full_response})
+                st.session_state.messages.append({"role": "assistant", "content": full_response})
                 
-            conversations.update({st.session_state.id_chat: st.session_state.messages})
-            with open(conversations_path, 'wb') as fp:
-                packed_bytes = msgpack.packb(conversations, use_bin_type=True)
-                fp.write(packed_bytes)
+                conversations.update({st.session_state.id_chat: st.session_state.messages})
+                with open(conversations_path, 'wb') as fp:
+                    packed_bytes = msgpack.packb(conversations, use_bin_type=True)
+                    fp.write(packed_bytes)
+
+            styl = f"""
+            <style>
+                .stTextInput {{
+                  position: fixed;
+                  bottom: 3rem;
+                }}
+            </style>
+            """
+            st.markdown(styl, unsafe_allow_html=True)