added option to append message

app.py

@@ -89,6 +89,8 @@ with st.sidebar:
         else:
             st.write("No cost incurred yet")
 
+    append = st.checkbox("Append to previous message", value=False)
+
 
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
@@ -141,21 +143,22 @@ if prompt := st.chat_input("Type a message", key="chat_input"):
                 with cols[i]:
                     st.image(image)
 
-    with st.chat_message("assistant"):
-        messages = [
-            {"role": m["role"], "content": m["content"]}
-            for m in st.session_state.messages
-        ]
-        stream = client.chat.completions.create(
-            model="gpt-4-vision-preview",
-            messages=messages,
-            stream=True,
-            seed=random_seed,
-            temperature=temperature,
-            max_tokens=max_tokens,
-        )
-        response = st.write_stream(stream)
-
-    push_message("assistant", response)
-    calculate_cost()
-    clear_uploader()
+    if not append:
+        with st.chat_message("assistant"):
+            messages = [
+                {"role": m["role"], "content": m["content"]}
+                for m in st.session_state.messages
+            ]
+            stream = client.chat.completions.create(
+                model="gpt-4-vision-preview",
+                messages=messages,
+                stream=True,
+                seed=random_seed,
+                temperature=temperature,
+                max_tokens=max_tokens,
+            )
+            response = st.write_stream(stream)
+
+        push_message("assistant", response)
+        calculate_cost()
+    clear_uploader()

utils.py

@@ -52,9 +52,11 @@ def clear_uploader():
 
 def undo():
     if len(st.session_state.messages) > 0:
-        st.session_state.messages.pop()
-        st.session_state.messages.pop()
-        st.session_state.cost.pop()
+        st.query_params.clear()
+        msg = st.session_state.messages.pop()
+        if msg["role"] == "assistant":
+            st.session_state.cost.pop()
+        time.sleep(0.1)
         st.rerun()
 
 def restart():