update file structure

app.py

@@ -1,6 +1,6 @@
 from openai import OpenAI
 import streamlit as st
-from utils import im_2_b64, calculate_cost
+from utils import im_2_b64, calculate_cost, clear_uploader, undo, restart
 import pickle
 from upload import upload_file, get_file
 
@@ -34,16 +34,6 @@ if len(st.session_state.messages) == 0 and "id" in st.query_params:
         for k, v in obj.items():
             st.session_state[k] = v
 
-def clear_uploader():
-    st.session_state["uploader_key"] += 1
-    st.rerun()
-
-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.rerun()
 
 def share():
     obj = {}
@@ -63,8 +53,7 @@ with st.sidebar:
     cols = st.columns(2)
     with cols[0]:
         if st.button("Restart", type="primary", use_container_width=True):
-            st.session_state.messages = []
-            clear_uploader()
+            restart()
     
     with cols[1]:
         if st.button("Undo", use_container_width=True):

utils.py

@@ -43,3 +43,21 @@ def calculate_cost():
                 total_cost += get_text_cost(content["text"], UNIT_COST[role])
     
     st.session_state.cost.append(total_cost)
+
+
+def clear_uploader():
+    st.session_state["uploader_key"] += 1
+    st.rerun()
+
+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.rerun()
+
+def restart():
+    st.session_state.messages = []
+    st.session_state.cost = []
+    st.query_params.clear()
+    clear_uploader()