update split

app.py

@@ -125,8 +125,7 @@ def push_message(role, content, images=None):
     st.session_state.messages.append(message)
     return message
 
-chat_input_disabled = False
-if prompt := st.chat_input("Type a message", key="chat_input", disabled=chat_input_disabled):
+if prompt := st.chat_input("Type a message", key="chat_input"):
     push_message("user", prompt, images)
     with st.chat_message("user"):
         st.markdown(prompt)
@@ -141,7 +140,6 @@ if prompt := st.chat_input("Type a message", key="chat_input", disabled=chat_inp
             {"role": m["role"], "content": m["content"]}
             for m in st.session_state.messages
         ]
-        chat_input_disabled = True
         stream = client.chat.completions.create(
             model="gpt-4-vision-preview",
             messages=messages,
@@ -154,5 +152,4 @@ if prompt := st.chat_input("Type a message", key="chat_input", disabled=chat_inp
 
     push_message("assistant", response)
     calculate_cost()
-    chat_input_disabled = False
     clear_uploader()

utils.py

@@ -23,7 +23,7 @@ def im_2_b64(image):
 
 def calculate_cost():
     def get_text_cost(text, unit_cost):
-        num_of_words = len(text.split(" "))
+        num_of_words = len(text.split())
         tokens = max(1000.0 * num_of_words / 750.0, 0.0)
         tokens = tokens / 1000.0
         cost = tokens * unit_cost